arXiv reaDer
画像再構成のためのエネルギーベースモデルの共有事前学習
Shared Prior Learning of Energy-Based Models for Image Reconstruction
エネルギーベースの学習、パッチベースのワッサースタイン損失関数、共有の事前学習の3つの主要な構成要素を持つ、グラウンドトゥルースデータなしのトレーニング用に特別に設計された画像再構成のための新しい学習ベースのフレームワークを提案します。エネルギーベースの学習では、学習されたデータ忠実度項とデータ駆動型正規化子で構成されるエネルギー汎関数のパラメーターが、平均場最適制御問題で計算されます。グラウンドトゥルースデータがない場合は、損失関数をパッチベースのWasserstein関数に変更します。この関数では、出力画像のローカル統計が破損していない参照パッチと比較されます。最後に、共有事前学習では、前述の両方の最適制御問題が、正則化の共有学習パラメーターと同時に最適化され、教師なし画像再構成がさらに強化されます。勾配流のいくつかの時間離散化スキームを導き出し、Mosco収束の観点からそれらの一貫性を検証します。多数の数値実験で、提案された方法が、トレーニングに利用できるグラウンドトゥルース画像がない場合でも、さまざまな画像再構成アプリケーションに対して最先端の結果を生成することを示しています。
We propose a novel learning-based framework for image reconstruction particularly designed for training without ground truth data, which has three major building blocks: energy-based learning, a patch-based Wasserstein loss functional, and shared prior learning. In energy-based learning, the parameters of an energy functional composed of a learned data fidelity term and a data-driven regularizer are computed in a mean-field optimal control problem. In the absence of ground truth data, we change the loss functional to a patch-based Wasserstein functional, in which local statistics of the output images are compared to uncorrupted reference patches. Finally, in shared prior learning, both aforementioned optimal control problems are optimized simultaneously with shared learned parameters of the regularizer to further enhance unsupervised image reconstruction. We derive several time discretization schemes of the gradient flow and verify their consistency in terms of Mosco convergence. In numerous numerical experiments, we demonstrate that the proposed method generates state-of-the-art results for various image reconstruction applications--even if no ground truth images are available for training.
updated: Thu Nov 12 2020 17:56:05 GMT+0000 (UTC)
published: Thu Nov 12 2020 17:56:05 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アソシエイト