arXiv reaDer
Loop Unrolled Shallow Equilibrium Regularizer (LUSER) -- メモリ効率の良い逆問題ソルバー
Loop Unrolled Shallow Equilibrium Regularizer (LUSER) -- A Memory-Efficient Inverse Problem Solver
逆問題では、破損している可能性があり、多くの場合不適切な設定の測定値から、対象となる潜在的な信号を再構築することを目指しています。古典的な最適化ベースの手法は、データ整合性メトリックを正則化と共に最適化することによって進められます。現在の最先端の機械学習アプローチは、最適化ベースのソルバーの反復更新を展開し、データから正則化を学習することにより、このような手法からインスピレーションを得ています。このループ アンローリング (LU) メソッドは大きな成功を収めていますが、最高のパフォーマンスを得るには深いモデルが必要になることが多く、トレーニング中のメモリ コストが高くなります。したがって、計算コストとネットワークの表現力のバランスに対処するために、浅い均衡正則化 (LUSER) を使用した LU アルゴリズムを提案します。これらの暗黙的なモデルは、より深い畳み込みネットワークと同じくらい表現力がありますが、トレーニング中のメモリ効率ははるかに高くなります。提案された方法は、画像のブレ除去、コンピューター断層撮影 (CT)、およびシングルコイル磁気共鳴画像 (MRI) タスクで評価され、トレーニング中に必要な計算リソースが最大 8 分の 1 であるにもかかわらず、同等またはそれ以上のパフォーマンスを示します。フィードフォワード畳み込み正則化を備えたより典型的な LU アーキテクチャに対して。
In inverse problems we aim to reconstruct some underlying signal of interest from potentially corrupted and often ill-posed measurements. Classical optimization-based techniques proceed by optimizing a data consistency metric together with a regularizer. Current state-of-the-art machine learning approaches draw inspiration from such techniques by unrolling the iterative updates for an optimization-based solver and then learning a regularizer from data. This loop unrolling (LU) method has shown tremendous success, but often requires a deep model for the best performance leading to high memory costs during training. Thus, to address the balance between computation cost and network expressiveness, we propose an LU algorithm with shallow equilibrium regularizers (LUSER). These implicit models are as expressive as deeper convolutional networks, but far more memory efficient during training. The proposed method is evaluated on image deblurring, computed tomography (CT), as well as single-coil Magnetic Resonance Imaging (MRI) tasks and shows similar, or even better, performance while requiring up to 8 times less computational resources during training when compared against a more typical LU architecture with feedforward convolutional regularizers.
updated: Fri Oct 14 2022 02:08:18 GMT+0000 (UTC)
published: Mon Oct 10 2022 19:50:37 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト