arXiv reaDer
マルチパスアテンションを備えたマルチヘッド畳み込みニューラルネットワークは、画像のノイズ除去を改善します
A Multi-Head Convolutional Neural Network With Multi-path Attention improves Image Denoising
最近では、畳み込みニューラル ネットワーク (CNN) とアテンション メカニズムが画像のノイズ除去に広く使用されており、満足のいくパフォーマンスが達成されています。ただし、以前の研究では、ほとんどの場合、単一のヘッドを使用してノイズの多い画像を受信し、抽出される特徴の豊富さを制限していました。したがって、MHCNN という名前の複数のヘッド (MH) を備えた新しい CNN がこの論文で提案され、そのヘッドは異なる回転角度で回転された入力画像を受け取ります。 MH は、MHCNN が回転した画像の特徴を同時に利用してノイズを除去するようにします。これらの機能を効果的に統合するために、新しいマルチパス アテンション メカニズム (MPA) を提示します。ピクセル レベル、チャネル レベル、またはパッチ レベルの機能を処理する以前のアテンション メカニズムとは異なり、MPA は画像レベルの機能に焦点を当てています。実験によると、MHCNN は加法性ホワイト ガウス ノイズ (AWGN) ノイズ除去および実世界の画像ノイズ除去において、他の最先端の CNN モデルを凌駕しています。そのピーク信号対雑音比 (PSNR) の結果は、BRDNet、RIDNet、PAN-Net、および CSANN などの他のネットワークよりも高くなっています。コードは https://github.com/JiaHongZ/MHCNN でアクセスできます。
Recently, convolutional neural networks (CNNs) and attention mechanisms have been widely used in image denoising and achieved satisfactory performance. However, the previous works mostly use a single head to receive the noisy image, limiting the richness of extracted features. Therefore, a novel CNN with multiple heads (MH) named MHCNN is proposed in this paper, whose heads will receive the input images rotated by different rotation angles. MH makes MHCNN simultaneously utilize features of rotated images to remove noise. To integrate these features effectively, we present a novel multi-path attention mechanism (MPA). Unlike previous attention mechanisms that handle pixel-level, channel-level, or patch-level features, MPA focuses on features at the image level. Experiments show MHCNN surpasses other state-of-the-art CNN models on additive white Gaussian noise (AWGN) denoising and real-world image denoising. Its peak signal-to-noise ratio (PSNR) results are higher than other networks, such as BRDNet, RIDNet, PAN-Net, and CSANN. The code is accessible at https://github.com/JiaHongZ/MHCNN.
updated: Fri Nov 04 2022 02:21:48 GMT+0000 (UTC)
published: Wed Apr 27 2022 07:11:41 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アソシエイト