ラベルのないノイズの多いデータからの学習は、機械学習の大きな課題の1つです。そのため、絶え間なく提案された新しいアイデアを伴う研究が急増しています。この研究では、スタインの不偏リスク推定量(SURE)という古典的な考え方を再考します。画像の復元のコンテキストでは、SUREとその一般化を使用して、グラウンドトゥルースデータのないさまざまな画像のノイズ除去と復元の問題に対して畳み込みニューラルネットワーク(CNN)をトレーニングできることを示します。特に、私たちの目標は、画像のノイズの多い線形変換(測定)から画像xを再構成することです。追加のデータが利用できない場合と、xと同じノイズの多い分布から描画された他の画像の測定値があるが、クリーンな画像にアクセスできない場合の2つのシナリオを検討します。たとえば、医用画像処理、顕微鏡検査、天文学の場合、ノイズのないグラウンドトゥルースデータはほとんど利用できません。この状況では、SUREを使用して、xの推定値に関連する平均二乗誤差損失を推定できることを示します。この損失の推定値を使用して、ノイズ除去と圧縮センシングの回復を実行するようにネットワークをトレーニングします。さらに、SUREフレームワークを使用して、Ulyanovらによって提示された興味深い結果を部分的に説明および改善しています。 「Deep Image Prior」:ランダムな重みで初期化され、単一のノイズの多い画像に適合するネットワークは、その画像のノイズを効果的に取り除くことができます。このペーパーで説明されているネットワークとメソッドのパブリック実装は、https://github.com/ricedsp/D-AMP_Toolboxにあります。
Learning from unlabeled and noisy data is one of the grand challenges of machine learning. As such, it has seen a flurry of research with new ideas proposed continuously. In this work, we revisit a classical idea: Stein's Unbiased Risk Estimator (SURE). We show that, in the context of image recovery, SURE and its generalizations can be used to train convolutional neural networks (CNNs) for a range of image denoising and recovery problems without any ground truth data. Specifically, our goal is to reconstruct an image x from a noisy linear transformation (measurement) of the image. We consider two scenarios: one where no additional data is available and one where we have measurements of other images that are drawn from the same noisy distribution as x, but have no access to the clean images. Such is the case, for instance, in the context of medical imaging, microscopy, and astronomy, where noise-less ground truth data is rarely available. We show that in this situation, SURE can be used to estimate the mean-squared-error loss associated with an estimate of x. Using this estimate of the loss, we train networks to perform denoising and compressed sensing recovery. In addition, we also use the SURE framework to partially explain and improve upon an intriguing results presented by Ulyanov et al. in "Deep Image Prior": that a network initialized with random weights and fit to a single noisy image can effectively denoise that image. Public implementations of the networks and methods described in this paper can be found at https://github.com/ricedsp/D-AMP_Toolbox.