反復ニューラルネットワーク(INN)は、イメージング、画像処理、コンピュータービジョンにおける逆問題を解決するために急速に注目を集めています。 INNは、回帰NNと反復モデルベースの画像再構成(MBIR)アルゴリズムを組み合わせており、既存のMBIR最適化モデルよりも優れた汎化機能と再構成品質の両方に優れています。このホワイトペーパーでは、モメンタムとマジョライザーを回帰NNで使用するブロック単位のMBIRアルゴリズムを一般化することにより、最初の高速で収束型のINNアーキテクチャであるMomentum-Netを提案します。高速MBIRの場合、Momentum-Netは外挿モジュールのモメンタム項を使用し、マジョライザーを使用して各反復で非反復MBIRモジュールを使用します。Momentum-Netの各反復は、画像調整、外挿、およびMBIRの3つのコアモジュールで構成されます。 Momentum-Netは、2つの無症候条件下で、一般的な微分可能(非)凸MBIR関数(またはデータフィット項)と凸型実行可能セットの固定小数点への収束を保証します。トレーニングサンプルとテストサンプル間のデータフィットの変動を考慮するために、マジョライゼーションマトリックスの「スペクトルスプレッド」に基づく正則化パラメーター選択スキームも提案しています。フォーカルスタックとスパースビュー計算機トモグラフィーを使用したライトフィールド写真の数値実験は、同一の回帰NNアーキテクチャーが与えられた場合、Momentum-Netがいくつかの既存のINNに比べてMBIR速度と精度を大幅に向上させることを示しています。各アプリケーションで最新のMBIR法と比較して、再構成の品質が大幅に向上します。
Iterative neural networks (INN) are rapidly gaining attention for solving inverse problems in imaging, image processing, and computer vision. INNs combine regression NNs and an iterative model-based image reconstruction (MBIR) algorithm, often leading to both good generalization capability and outperforming reconstruction quality over existing MBIR optimization models. This paper proposes the first fast and convergent INN architecture, Momentum-Net, by generalizing a block-wise MBIR algorithm that uses momentum and majorizers with regression NNs. For fast MBIR, Momentum-Net uses momentum terms in extrapolation modules, and noniterative MBIR modules at each iteration by using majorizers, where each iteration of Momentum-Net consists of three core modules: image refining, extrapolation, and MBIR. Momentum-Net guarantees convergence to a fixed-point for general differentiable (non)convex MBIR functions (or data-fit terms) and convex feasible sets, under two asymptomatic conditions. To consider data-fit variations across training and testing samples, we also propose a regularization parameter selection scheme based on the "spectral spread" of majorization matrices. Numerical experiments for light-field photography using a focal stack and sparse-view computational tomography demonstrate that, given identical regression NN architectures, Momentum-Net significantly improves MBIR speed and accuracy over several existing INNs; it significantly improves reconstruction quality compared to a state-of-the-art MBIR method in each application.