画像の修復は、破損した画像の欠落している領域をもっともらしい内容で埋めることを目的としたタスクです。最近の修復方法では、修復ジェネレータの学習をガイドするための補助損失として、知覚損失とスタイル損失が導入されています。知覚およびスタイルの損失は、生成された領域の深い特徴を監視することにより、修復された結果の知覚品質を向上させるのに役立ちます。ただし、補助損失の使用には2つの課題があります。(i)適切に実行するために知覚損失とスタイル損失の重みを決定するために時間のかかるグリッド検索が必要であり、(ii)異なる補助能力を持つ損失項は次のように等しく重み付けされます。知覚とスタイルの喪失。これらの2つの課題に対応するために、時間のかかるグリッド検索を行わずに、補助損失項に独立して重みを付け、単一のトレーニングプロセス内でそれらの重みを適応的に調整する新しいフレームワークを提案します。具体的には、知覚損失とスタイル損失の補助的な可能性を解放するために、さまざまな調整可能な重みを使用してさまざまな損失条件の寄与を考慮することにより、調整可能な知覚損失(TPL)と調整可能なスタイル損失(TSL)の2つの補助損失を提案します。 TPLとTSLは、知覚とスタイルの喪失のスーパーセットであり、標準的な知覚とスタイルの喪失の補助的な可能性を解放します。さらに、単一のトレーニングプロセスでTPLとTSLを効率的に再重み付けするAuxiliary Weights Adaptation(AWA)アルゴリズムを提案します。 AWAは、最高の補助ウェイトが修復性能の最大の向上につながるという原則に基づいています。公開されているデータセットで実験を行ったところ、現在のSOTAメソッドがより良い結果を達成するのにフレームワークが役立つことがわかりました。
Image Inpainting is a task that aims to fill in missing regions of corrupted images with plausible contents. Recent inpainting methods have introduced perceptual and style losses as auxiliary losses to guide the learning of inpainting generators. Perceptual and style losses help improve the perceptual quality of inpainted results by supervising deep features of generated regions. However, two challenges have emerged with the usage of auxiliary losses: (i) the time-consuming grid search is required to decide weights for perceptual and style losses to properly perform, and (ii) loss terms with different auxiliary abilities are equally weighted by perceptual and style losses. To meet these two challenges, we propose a novel framework that independently weights auxiliary loss terms and adaptively adjusts their weights within a single training process, without a time-consuming grid search. Specifically, to release the auxiliary potential of perceptual and style losses, we propose two auxiliary losses, Tunable Perceptual Loss (TPL) and Tunable Style Loss (TSL) by using different tunable weights to consider the contributions of different loss terms. TPL and TSL are supersets of perceptual and style losses and release the auxiliary potential of standard perceptual and style losses. We further propose the Auxiliary Weights Adaptation (AWA) algorithm, which efficiently reweights TPL and TSL in a single training process. AWA is based on the principle that the best auxiliary weights would lead to the most improvement in inpainting performance. We conduct experiments on publically available datasets and find that our framework helps current SOTA methods achieve better results.