arXiv reaDer
軽量画像超解像のための畳み込みへの変換器設計の組み込み
Incorporating Transformer Designs into Convolutions for Lightweight Image Super-Resolution
近年、大規模な畳み込みカーネルの使用は、長期的な依存関係をキャプチャし、大きな受容野を提供する能力により、畳み込みニューラル ネットワークの設計で一般的になってきました。ただし、カーネル サイズの増加は、パラメーター数の 2 次増加にもつながり、計算量とメモリ要件が高くなります。この課題に対処するために、標準的な畳み込みを自己注意メカニズムでアップグレードする近傍注意 (NA) モジュールを提案します。 NA モジュールは、スライディング ウィンドウ パターンで長距離依存関係を効率的に抽出するため、大規模な畳み込みカーネルと同様のパフォーマンスをより少ないパラメーターで実現します。 NA モジュールに基づいて、TCSR という軽量の単一画像超解像 (SISR) ネットワークを提案します。さらに、TCSR に拡張フィードフォワード ネットワーク (EFFN) を導入して、SISR のパフォーマンスを向上させます。 EFFN は、効率的な機能集約のために、パラメーターを使用しない空間シフト操作を採用しています。私たちの広範な実験と切除研究は、TCSR が既存の軽量 SISR 法よりも優れており、最先端の性能を達成することを示しています。コードは https://github.com/Aitical/TCSR で入手できます。
In recent years, the use of large convolutional kernels has become popular in designing convolutional neural networks due to their ability to capture long-range dependencies and provide large receptive fields. However, the increase in kernel size also leads to a quadratic growth in the number of parameters, resulting in heavy computation and memory requirements. To address this challenge, we propose a neighborhood attention (NA) module that upgrades the standard convolution with a self-attention mechanism. The NA module efficiently extracts long-range dependencies in a sliding window pattern, thereby achieving similar performance to large convolutional kernels but with fewer parameters. Building upon the NA module, we propose a lightweight single image super-resolution (SISR) network named TCSR. Additionally, we introduce an enhanced feed-forward network (EFFN) in TCSR to improve the SISR performance. EFFN employs a parameter-free spatial-shift operation for efficient feature aggregation. Our extensive experiments and ablation studies demonstrate that TCSR outperforms existing lightweight SISR methods and achieves state-of-the-art performance. Our codes are available at https://github.com/Aitical/TCSR.
updated: Sat Mar 25 2023 01:32:18 GMT+0000 (UTC)
published: Sat Mar 25 2023 01:32:18 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アソシエイト