arXiv reaDer
画像復元のためのより良い損失関数のトレーニング
Training a Better Loss Function for Image Restoration
単一画像の超解像などの画像復元問題におけるニューラルネットワークのアプリケーションの中心は、自然で知覚的に満足のいく結果を促進する損失関数の選択です。損失関数の一般的な選択肢は、VGGやLPIPSなどの事前トレーニング済みネットワークです。これは、復元された画像と参照画像の違いを計算するための特徴抽出器として使用されます。ただし、このようなアプローチには複数の欠点があります。計算コストが高く、正則化とハイパーパラメータの調整が必要であり、無関係なタスクでトレーニングされた大規模なネットワークが必要です。この作業では、画像復元タスクに適した損失関数を作成する理由について説明します。まず、単一の自然画像で、単一画像の超解像、ノイズ除去、およびJPEGアーティファクト除去において、最先端の損失関数よりも優れた軽量の特徴抽出器をトレーニングするのに十分であることがわかります。ジェネレータによって導入されたエラーにペナルティを課すように訓練された、一連の弁別子を含む新しいマルチスケール弁別機能(MDF)損失を提案します。次に、効果的な損失関数は、知覚される画質の優れた予測子である必要はなく、代わりに、特定の復元方法の歪みを特定することに特化する必要があることを示します。
Central to the application of neural networks in image restoration problems, such as single image super resolution, is the choice of a loss function that encourages natural and perceptually pleasing results. A popular choice for a loss function is a pre-trained network, such as VGG and LPIPS, 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. In this work, we explore the question of what makes a good loss function for an image restoration task. First, we observe that a single natural image is sufficient to train a lightweight feature extractor that outperforms state-of-the-art loss functions in single image super resolution, denoising, and JPEG artefact removal. We propose a novel Multi-Scale Discriminative Feature (MDF) loss comprising a series of discriminators, trained to penalize errors introduced by a generator. Second, we show 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: Fri Mar 26 2021 17:29:57 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アソシエイト