畳み込みニューラルネットワーク(CNN)に基づく特徴学習手法は、画像分類タスクで大きな成果を上げることに成功しています。ただし、固有のノイズおよびその他のいくつかの要因により、畳み込み特徴統計の有効性が弱まる可能性があります。この論文では、周波数領域で離散ウェーブレット変換(DWT)を調査し、高周波領域でのみ注意を実装する新しいウェーブレット注意(WA)ブロックを設計します。これに基づいて、画像分類のためのウェーブレット注意畳み込みニューラルネットワーク(WA-CNN)を提案します。具体的には、WA-CNNは、特徴マップを低周波成分と高周波成分に分解して、基本オブジェクトの構造、詳細情報、ノイズをそれぞれ格納します。次に、WAブロックを利用して、さまざまな注意要因を使用して高周波ドメインの詳細情報をキャプチャしますが、低周波ドメインの基本的なオブジェクト構造を予約します。 CIFAR-10およびCIFAR-100データセットの実験結果は、提案されたWA-CNNが他の関連ネットワークと比較して分類精度の大幅な改善を達成することを示しています。具体的には、MobileNetV2バックボーンに基づいて、WA-CNNはCIFAR-10ベンチマークで1.26%のTop-1精度の向上を達成し、CIFAR-100ベンチマークで1.54%のTop-1精度の向上を達成します。
The feature learning methods based on convolutional neural network (CNN) have successfully produced tremendous achievements in image classification tasks. However, the inherent noise and some other factors may weaken the effectiveness of the convolutional feature statistics. In this paper, we investigate Discrete Wavelet Transform (DWT) in the frequency domain and design a new Wavelet-Attention (WA) block to only implement attention in the high-frequency domain. Based on this, we propose a Wavelet-Attention convolutional neural network (WA-CNN) for image classification. Specifically, WA-CNN decomposes the feature maps into low-frequency and high-frequency components for storing the structures of the basic objects, as well as the detailed information and noise, respectively. Then, the WA block is leveraged to capture the detailed information in the high-frequency domain with different attention factors but reserves the basic object structures in the low-frequency domain. Experimental results on CIFAR-10 and CIFAR-100 datasets show that our proposed WA-CNN achieves significant improvements in classification accuracy compared to other related networks. Specifically, based on MobileNetV2 backbones, WA-CNN achieves 1.26% Top-1 accuracy improvement on the CIFAR-10 benchmark and 1.54% Top-1 accuracy improvement on the CIFAR-100 benchmark.