arXiv:2502.03103cs.CVcs.LG2025-02被引 3

提出边缘注意力模块,显著提升图像分类准确率。

Edge Attention Module for Object Classification

  • 设计最大最小池化层提取关键边缘特征,融入注意力机制。
  • 在Caltech-101和Caltech-256上分别达到95.5%和86%准确率。
  • 适用于需要精细边缘识别的图像分类任务。

本研究提出一种基于边缘注意力的卷积神经网络(CNN),以应对类别不平衡与类间相似性问题。首次引入边缘注意力模块(EAM),由最大最小池化层与卷积层组成,专为捕捉对物体分类至关重要的边缘信息而设计。通过将该新型池化技术嵌入注意力模块,模型能自动聚焦于关键边缘特征,显著提升分类准确率与F1分数。我们在Caltech-101、Caltech-256、CIFAR-100和Tiny ImageNet-200等标准数据集上,将EAM或双EAM(2EAMs)集成至多个预训练CNN模型中进行测试。实验表明,所提框架在上述数据集上的表现优于所有基线CNN模型及近期主流模型(如PiT、CBAM、ConvNext)。在Caltech-101和Caltech-256上分别取得95.5%和86%的准确率,目前为该数据集的最佳结果。

原文摘要 · Abstract (English)

A novel ``edge attention-based Convolutional Neural Network (CNN)'' is proposed in this research for object classification task. With the advent of advanced computing technology, CNN models have achieved to remarkable success, particularly in computer vision applications. Nevertheless, the efficacy of the conventional CNN is often hindered due to class imbalance and inter-class similarity problems, which are particularly prominent in the computer vision field. In this research, we introduce for the first time an ``Edge Attention Module (EAM)'' consisting of a Max-Min pooling layer, followed by convolutional layers. This Max-Min pooling is entirely a novel pooling technique, specifically designed to capture only the edge information that is crucial for any object classification task. Therefore, by integrating this novel pooling technique into the attention module, the CNN network inherently prioritizes on essential edge features, thereby boosting the accuracy and F1-score of the model significantly. We have implemented our proposed EAM or 2EAMs on several standard pre-trained CNN models for Caltech-101, Caltech-256, CIFAR-100 and Tiny ImageNet-200 datasets. The extensive experiments reveal that our proposed framework (that is, EAM with CNN and 2EAMs with CNN), outperforms all pre-trained CNN models as well as recent trend models ``Pooling-based Vision Transformer (PiT)'', ``Convolutional Block Attention Module (CBAM)'', and ConvNext, by substantial margins. We have achieved the accuracy of 95.5% and 86% by the proposed framework on Caltech-101 and Caltech-256 datasets, respectively. So far, this is the best results on these datasets, to the best of our knowledge.

图像分类注意力机制边缘特征

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。