GAN、正規化フロー、拡散モデルなどの深い生成モデルは、逆問題の強力な正則化器です。それらは、不適切な姿勢を減らし、高品質の結果を達成するのに役立つ大きな可能性を示しています.ただし、このような深い生成モデルの潜在的なテンソルは、特にデータ ノイズや不正確なフォワード モデルが存在する場合、反転中に目的の高次元標準ガウス分布から外れ、忠実度の低いソリューションにつながる可能性があります。この問題に対処するために、最適化問題を解決することによってカスタム演算子が定義される、新しい微分可能なデータ依存レイヤーを使用して、潜在テンソルを再パラメーター化およびガウス化することを提案します。これらの提案されたレイヤーは、逆問題を制約して、忠実度の高い分布内ソリューションを取得します。 StyleGAN2 と Glow の 2 つの代表的な深層生成モデルを使用して、圧縮センシング MRI、画像ブレ除去、アイコナル トモグラフィー (非線形 PDE 制約付き逆問題) の 3 つの反転タスクで手法を検証します。当社のアプローチは、精度と一貫性の点で最先端のパフォーマンスを実現します。
Deep generative models such as GANs, normalizing flows, and diffusion models are powerful regularizers for inverse problems. They exhibit great potential for helping reduce ill-posedness and attain high-quality results. However, the latent tensors of such deep generative models can fall out of the desired high-dimensional standard Gaussian distribution during inversion, particularly in the presence of data noise and inaccurate forward models, leading to low-fidelity solutions. To address this issue, we propose to reparameterize and Gaussianize the latent tensors using novel differentiable data-dependent layers wherein custom operators are defined by solving optimization problems. These proposed layers constrain inverse problems to obtain high-fidelity in-distribution solutions. We validate our technique on three inversion tasks: compressive-sensing MRI, image deblurring, and eikonal tomography (a nonlinear PDE-constrained inverse problem) using two representative deep generative models: StyleGAN2 and Glow. Our approach achieves state-of-the-art performance in terms of accuracy and consistency.