arXiv reaDer
タスク固有の画像再構成損失のトレーニング
Training a Task-Specific Image Reconstruction Loss
損失関数の選択は、単一画像の超解像などの画像復元の問題についてニューラルネットワークをトレーニングする際の重要な要素です。損失関数は、自然で知覚的に満足のいく結果を促進するはずです。損失の一般的な選択肢は、復元された画像と参照画像の違いを計算するための特徴抽出器として使用されるVGGなどの事前トレーニング済みネットワークです。ただし、このようなアプローチには複数の欠点があります。計算コストが高く、正則化とハイパーパラメータの調整が必要であり、無関係なタスクでトレーニングされた大規模なネットワークが必要です。さらに、すべてのアプリケーション間および異なるデータセット間で最適に機能する単一の損失関数がないことが確認されています。この作業では、代わりに、アプリケーション固有の性質を持つ一連の損失関数をトレーニングすることを提案します。私たちの損失関数は、アプリケーション固有のアーティファクトの存在を検出してペナルティを課すように訓練された一連の弁別器で構成されています。単一の自然画像とそれに対応する歪みが、単一画像の超解像、ノイズ除去、JPEGアーティファクト除去などのアプリケーションで最先端の損失関数よりも優れた特徴抽出器をトレーニングするのに十分であることを示します。最後に、有効な損失関数は、知覚される画質の優れた予測子である必要はなく、代わりに、特定の復元方法の歪みを特定することに特化する必要があると結論付けます。
The choice of a loss function is an important factor when training neural networks for image restoration problems, such as single image super resolution. The loss function should encourage natural and perceptually pleasing results. A popular choice for a loss is a pre-trained network, such as VGG, which is used as a feature extractor for computing the difference between restored and reference images. However, such an approach has multiple drawbacks: it is computationally expensive, requires regularization and hyper-parameter tuning, and involves a large network trained on an unrelated task. Furthermore, it has been observed that there is no single loss function that works best across all applications and across different datasets. In this work, we instead propose to train a set of loss functions that are application specific in nature. Our loss function comprises a series of discriminators that are trained to detect and penalize the presence of application-specific artifacts. We show that a single natural image and corresponding distortions are sufficient to train our feature extractor that outperforms state-of-the-art loss functions in applications like single image super resolution, denoising, and JPEG artifact removal. Finally, we conclude that an effective loss function does not have to be a good predictor of perceived image quality, but instead needs to be specialized in identifying the distortions for a given restoration method.
updated: Sun Oct 17 2021 08:14:08 GMT+0000 (UTC)
published: Fri Mar 26 2021 17:29:57 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アソシエイト