arXiv reaDer
グラウンドトゥルースデータを使用しないディープラーニングベースのデノイザーのトレーニング
Training Deep Learning Based Denoisers without Ground Truth Data
最近開発された深層学習ベースのデノイザーは、BM3Dなどの最先端の従来のデノイザーよりも優れていることがよくあります。これらは通常、ディープニューラルネットワーク(DNN)の出力画像とグラウンドトゥルース画像の間の平均二乗誤差(MSE)を最小化するようにトレーニングされています。したがって、深層学習ベースのデノイザーでは、高品質のノイズのないグラウンドトゥルースデータを使用して高性能を実現することが重要です。ただし、一部のアプリケーションでは、ノイズのない画像を取得することが困難な場合や実行不可能な場合があります。ここでは、ガウスノイズを含むトレーニングデータのノイズの多い画像の使用のみに基づいてDNNデノイザーをトレーニングするためのスタインのバイアスのないリスク推定器(SURE)に基づく方法を提案します。グラウンドトゥルースデータを使用せずに、SUREベースの方法で、DNNデノイザーをトレーニングして、グレースケール画像とカラー画像の両方でグラウンドトゥルースでトレーニングされたネットワークに近いパフォーマンスを実現できることを示します。また、パフォーマンスをさらに向上させるために、ノイズの多いテスト画像を使用したSUREベースの精製方法を提案します。私たちの迅速な精製方法は、従来のBM3D、以前の深い画像、そして多くの場合、グラウンドトゥルースでトレーニングされたネットワークを上回りました。ポアソンノイズモデルへのSUREベースの方法の潜在的な拡張も調査されました。
Recently developed deep-learning-based denoisers often outperform state-of-the-art conventional denoisers such as the BM3D. They are typically trained to minimize the mean squared error (MSE) between the output image of a deep neural network (DNN) and a ground truth image. Thus, it is important for deep-learning-based denoisers to use high quality noiseless ground truth data for high performance. However, it is often challenging or even infeasible to obtain noiseless images in some applications. Here, we propose a method based on Stein's unbiased risk estimator (SURE) for training DNN denoisers based only on the use of noisy images in the training data with Gaussian noise. We demonstrate that our SURE-based method, without the use of ground truth data, is able to train DNN denoisers to yield performances close to those networks trained with ground truth for both grayscale and color images. We also propose a SURE-based refining method with a noisy test image for further performance improvement. Our quick refining method outperformed conventional BM3D, deep image prior, and often the networks trained with ground truth. Potential extension of our SURE-based methods to Poisson noise model was also investigated.
updated: Thu Apr 22 2021 02:48:50 GMT+0000 (UTC)
published: Sun Mar 04 2018 07:55:40 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アソシエイト