arXiv reaDer
TAPE:画像復元のためのタスクにとらわれない事前埋め込み
TAPE: Task-Agnostic Prior Embedding for Image Restoration
自然な画像復元のための一般化された事前学習は、重要でありながら挑戦的な作業です。初期の方法は、正規化されたスパース性、L0勾配、ダークチャネル事前分布などを含む手作りの事前分布を主に含んでいました。最近、深いニューラルネットワークがさまざまな画像事前分布を学習するために使用されていますが、一般化を保証するものではありません。この論文では、タスクにとらわれない事前設定をトランスフォーマーに組み込む新しいアプローチを提案します。 Task-Agnostic Prior Embedding(TAPE)という名前のアプローチは、タスクに依存しない事前トレーニング、タスクに依存しない微調整、およびタスク固有の微調整の3つの段階で構成され、最初の段階で自然画像を変圧器に取り込み、後者の2つは、下流の画像復元を支援するための知識を抽出します。さまざまなタイプの劣化に関する実験により、TAPEの有効性が検証されます。 PSNRに関する画像復元のパフォーマンスは、1.45 dBも向上し、タスク固有のアルゴリズムよりも優れています。さらに重要なことに、TAPEは、劣化した画像から一般化された画像の事前情報を解きほぐす能力を示します。これは、未知のダウンストリームタスクへの好ましい転送能力を享受します。
Learning an generalized prior for natural image restoration is an important yet challenging task. Early methods mostly involved handcrafted priors including normalized sparsity, L0 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 three stages, namely, task-agnostic pre-training, task-agnostic fine-tuning, and task-specific fine-tuning, where the first one embeds prior knowledge about natural images into the transformer and the latter two 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.45 dB 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 Mar 11 2022 16:52:47 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アソシエイト