arXiv reaDer
TAPE: 画像復元のためのタスクに依存しない事前埋め込み
TAPE: Task-Agnostic Prior Embedding for Image Restoration
自然なイメージを復元するための一般化された事前確率を学習することは、重要でありながら困難な作業です。初期の方法には、主に、正規化されたスパース性、l_0 勾配、ダーク チャネルの事前分布などを含む手作りの事前分布が含まれていました。最近では、さまざまな画像の事前分布を学習するためにディープ ニューラル ネットワークが使用されていますが、一般化を保証するものではありません。この論文では、タスクに依存しない事前確率を変換器に埋め込む新しいアプローチを提案します。 Task-Agnostic Prior Embedding (TAPE) と名付けられた私たちのアプローチは、タスクに依存しない事前トレーニングとタスク固有の微調整の 2 つの段階で構成され、最初の段階では自然画像に関する事前の知識がトランスフォーマーに埋め込まれ、2 番目の段階では自然画像に関する事前の知識が埋め込まれます。ステージは知識を抽出して、下流の画像復元を支援します。さまざまな種類の劣化に関する実験により、TAPE の有効性が検証されています。 PSNR に関する画像復元性能は 1.45dB も向上し、タスク固有のアルゴリズムよりも優れています。さらに重要なことに、TAPE は、劣化した画像から一般化された画像のプリアを解きほぐす能力を示し、未知のダウンストリーム タスクへの好ましい転送能力を享受します。
Learning a generalized prior for natural image restoration is an important yet challenging task. Early methods mostly involved handcrafted priors including normalized sparsity, l_0 gradients, dark channel priors, etc. Recently, deep neural networks have been used to learn various image priors but do not guarantee to generalize. In this paper, we propose a novel approach that embeds a task-agnostic prior into a transformer. Our approach, named Task-Agnostic Prior Embedding (TAPE), consists of two stages, namely, task-agnostic pre-training and task-specific fine-tuning, where the first stage embeds prior knowledge about natural images into the transformer and the second stage extracts the knowledge to assist downstream image restoration. Experiments on various types of degradation validate the effectiveness of TAPE. The image restoration performance in terms of PSNR is improved by as much as 1.45dB and even outperforms task-specific algorithms. More importantly, TAPE shows the ability of disentangling generalized image priors from degraded images, which enjoys favorable transfer ability to unknown downstream tasks.
updated: Fri Aug 05 2022 16:17:51 GMT+0000 (UTC)
published: Fri Mar 11 2022 16:52:47 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アソシエイト