arXiv reaDer
画像超解像度のためのデュアル集約トランスフォーマー
Dual Aggregation Transformer for Image Super-Resolution
Transformer は最近、画像の超解像度 (SR) などの低レベルの視覚タスクでかなりの人気を集めています。これらのネットワークは、さまざまな次元、空間またはチャネルに沿って自己注意を利用し、優れたパフォーマンスを実現します。このことから、Transformer で 2 つの次元を組み合わせて、より強力な表現機能を実現することができました。上記の考えに基づいて、画像 SR 用の新しい Transformer モデルである Dual Aggregation Transformer (DAT) を提案します。当社の DAT は、ブロック間およびブロック内のデュアル方式で、空間次元とチャネル次元にわたる機能を集約します。具体的には、連続する Transformer ブロックで空間的セルフ アテンションとチャネル セルフ アテンションを交互に適用します。代替戦略により、DAT はグローバル コンテキストをキャプチャし、ブロック間の機能集約を実現できます。さらに、ブロック内特徴集約を実現するために、適応型相互作用モジュール (AIM) と空間ゲート フィードフォワード ネットワーク (SGFN) を提案します。 AIM は、対応する次元からの 2 つの自己注意メカニズムを補完します。一方、SGFN は、フィードフォワード ネットワークに追加の非線形空間情報を導入します。広範な実験により、当社の DAT が現在の方法を上回ることが示されました。コードとモデルは https://github.com/zhengchen1999/DAT から入手できます。
Transformer has recently gained considerable popularity in low-level vision tasks, including image super-resolution (SR). These networks utilize self-attention along different dimensions, spatial or channel, and achieve impressive performance. This inspires us to combine the two dimensions in Transformer for a more powerful representation capability. Based on the above idea, we propose a novel Transformer model, Dual Aggregation Transformer (DAT), for image SR. Our DAT aggregates features across spatial and channel dimensions, in the inter-block and intra-block dual manner. Specifically, we alternately apply spatial and channel self-attention in consecutive Transformer blocks. The alternate strategy enables DAT to capture the global context and realize inter-block feature aggregation. Furthermore, we propose the adaptive interaction module (AIM) and the spatial-gate feed-forward network (SGFN) to achieve intra-block feature aggregation. AIM complements two self-attention mechanisms from corresponding dimensions. Meanwhile, SGFN introduces additional non-linear spatial information in the feed-forward network. Extensive experiments show that our DAT surpasses current methods. Code and models are obtainable at https://github.com/zhengchen1999/DAT.
updated: Fri Aug 11 2023 05:21:15 GMT+0000 (UTC)
published: Mon Aug 07 2023 07:39:39 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アソシエイト