arXiv reaDer
U2-Former:画像復元用のネストされたU字型トランスフォーマー
U2-Former: A Nested U-shaped Transformer for Image Restoration
Transformerは、さまざまな高レベルの視覚タスクで優れたパフォーマンスを達成していますが、画像の復元でTransformerの可能性を最大限に活用することは依然として困難です。重要なのは、画像復元のための一般的なエンコーダー-デコーダーフレームワークでTransformerを適用する深さが限られていることです。これは、自己注意の計算負荷が大きく、レイヤーのさまざまな深さ(スケール)での通信が非効率的であるためです。この論文では、U2-Formerと呼ばれる、画像復元のための深くて効果的なTransformerベースのネットワークを紹介します。これは、Transformerをコア操作として使用して、深いエンコードおよびデコード空間で画像復元を実行できます。具体的には、ネストされたU字型の構造を利用して、特徴マップのさまざまなスケールでさまざまなレイヤー間の相互作用を促進します。さらに、トークン表現を圧縮する機能フィルタリングメカニズムを導入することにより、基本的なTransformerブロックの計算効率を最適化します。画像復元の一般的な監視方法とは別に、U2-Formerは複数の側面で対照学習を実行して、ノイズ成分を背景画像からさらに分離します。それぞれ反射除去、レインストリーク除去、および曇り除去を含むさまざまな画像復元タスクに関する広範な実験は、提案されたU2-Formerの有効性を示しています。
While Transformer has achieved remarkable performance in various high-level vision tasks, it is still challenging to exploit the full potential of Transformer in image restoration. The crux lies in the limited depth of applying Transformer in the typical encoder-decoder framework for image restoration, resulting from heavy self-attention computation load and inefficient communications across different depth (scales) of layers. In this paper, we present a deep and effective Transformer-based network for image restoration, termed as U2-Former, which is able to employ Transformer as the core operation to perform image restoration in a deep encoding and decoding space. Specifically, it leverages the nested U-shaped structure to facilitate the interactions across different layers with different scales of feature maps. Furthermore, we optimize the computational efficiency for the basic Transformer block by introducing a feature-filtering mechanism to compress the token representation. Apart from the typical supervision ways for image restoration, our U2-Former also performs contrastive learning in multiple aspects to further decouple the noise component from the background image. Extensive experiments on various image restoration tasks, including reflection removal, rain streak removal and dehazing respectively, demonstrate the effectiveness of the proposed U2-Former.
updated: Wed Dec 08 2021 12:09:20 GMT+0000 (UTC)
published: Sat Dec 04 2021 08:37:04 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アソシエイト