核心概念
本研究では、深層学習ネットワークの性能を向上させるために、再帰的モーメンタム加速(RMA)フレームワークを提案する。RMAモジュールは長短期記憶型再帰ニューラルネットワーク(LSTM-RNN)を使用して、モーメンタム加速プロセスをシミュレートする。RMAモジュールは、LSTM-RNNの能力を活用して、過去のグラジエントから知識を学習し保持することができる。
要約
本研究では、深層学習ネットワークの性能を向上させるために、再帰的モーメンタム加速(RMA)フレームワークを提案している。
RMAモジュールは、長短期記憶型再帰ニューラルネットワーク(LSTM-RNN)を使用して、モーメンタム加速プロセスをシミュレートする。
RMAモジュールは、LSTM-RNNの能力を活用して、過去のグラジエントから知識を学習し保持することができる。
RMAを2つの一般的な深層学習ネットワーク手法(LPGD、LPD)に適用し、LPGD-RMAとLPD-RMAを提案している。
非線形逆問題の2つの例(非線形畳み込み問題、電気インピーダンストモグラフィー問題)で実験を行い、RMAが深層学習ネットワークの性能を大幅に向上させることを示している。
特に、問題の非線形性が高い場合、RMAの効果が顕著に現れることが分かった。
また、RMAは少ないデータ量でも安定した性能を発揮し、インクルージョンの数が増えても頑健であることが確認された。
統計
非線形畳み込み問題では、RMAを適用したLPD-RMAが、LPDに比べて、a=1、a=2、a=4の場合でそれぞれ8.0%、12.0%、16.0%のMSE改善を示した。
電気インピーダンストモグラフィー問題では、RMAを適用したLPGD-RMA、LPGDSW-RMA、LPD-RMAが、従来手法のGN法やPDIPM-TVに比べて、より正確な再構成結果を得ることができた。