ディープ ニューラル ネットワークを介したイメージングの逆問題を解決するための最近の取り組みでは、最適化方法の固定回数の反復に触発されたアーキテクチャが使用されています。より多くの反復に対応するネットワークのトレーニングが困難なため、通常、反復の数は非常に少なくなります。結果として得られるソルバーは、重大なエラーを発生させずにテスト時に繰り返し実行することはできません。このペーパーでは、無限の反復に対応する代替アプローチについて説明し、最新の代替手段を上回る再構成精度の一貫した改善をもたらし、テスト時に計算予算を選択して、コンテキスト依存のトレードオフを最適化することができます。精度と計算。提案されたアプローチは、既知のフォワード モデルと古典的な最適化ベースの再構成方法からの洞察を組み込むように固定小数点の反復が構築される深層平衡モデルのアイデアを活用しています。
Recent efforts on solving inverse problems in imaging via deep neural networks use architectures inspired by a fixed number of iterations of an optimization method. The number of iterations is typically quite small due to difficulties in training networks corresponding to more iterations; the resulting solvers cannot be run for more iterations at test time without incurring significant errors. This paper describes an alternative approach corresponding to an infinite number of iterations, yielding a consistent improvement in reconstruction accuracy above state-of-the-art alternatives and where the computational budget can be selected at test time to optimize context-dependent trade-offs between accuracy and computation. The proposed approach leverages ideas from Deep Equilibrium Models, where the fixed-point iteration is constructed to incorporate a known forward model and insights from classical optimization-based reconstruction methods.