arXiv reaDer
KBNet: イメージ復元のためのカーネル基盤ネットワーク
KBNet: Kernel Basis Network for Image Restoration
学習ベースの画像復元では、空間情報をどのように集約するかが重要な役割を果たします。ほとんどの既存の CNN ベースのネットワークは、静的畳み込みカーネルを採用して空間情報をエンコードしますが、空間情報を適応的に集約することはできません。最近のトランスフォーマー ベースのアーキテクチャは、適応型の空間集約を実現します。しかし、それらは畳み込みの望ましい誘導バイアスを欠いており、大量の計算コストを必要とします。この論文では、空間情報集約の代表的な画像パターンをモデル化するために学習可能なカーネルベースを導入するカーネルベースアテンション(KBA)モジュールを提案します。異なる局所構造をモデル化するために、異なるカーネル ベースがトレーニングされます。各空間位置で、それらは予測されたピクセル単位の係数によって線形かつ適応的に融合され、集約の重みが取得されます。 KBA モジュールに基づいて、多軸機能融合 (MFF) ブロックをさらに設計し、画像復元のためにチャネル単位、空間不変、およびピクセル適応機能をエンコードおよび融合します。カーネル ベーシス ネットワーク (KBNet) と名付けられた私たちのモデルは、画像のノイズ除去、デレイン、ブレ除去のタスクで 10 以上のベンチマークで最先端のパフォーマンスを達成しながら、以前の SOTA メソッドよりも少ない計算コストで済みます。
How to aggregate spatial information plays an essential role in learning-based image restoration. Most existing CNN-based networks adopt static convolutional kernels to encode spatial information, which cannot aggregate spatial information adaptively. Recent transformer-based architectures achieve adaptive spatial aggregation. But they lack desirable inductive biases of convolutions and require heavy computational costs. In this paper, we propose a kernel basis attention (KBA) module, which introduces learnable kernel bases to model representative image patterns for spatial information aggregation. Different kernel bases are trained to model different local structures. At each spatial location, they are linearly and adaptively fused by predicted pixel-wise coefficients to obtain aggregation weights. Based on the KBA module, we further design a multi-axis feature fusion (MFF) block to encode and fuse channel-wise, spatial-invariant, and pixel-adaptive features for image restoration. Our model, named kernel basis network (KBNet), achieves state-of-the-art performances on more than ten benchmarks over image denoising, deraining, and deblurring tasks while requiring less computational cost than previous SOTA methods.
updated: Mon Mar 06 2023 04:17:29 GMT+0000 (UTC)
published: Mon Mar 06 2023 04:17:29 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アソシエイト