提出新型注意力机制,让模型精准聚焦图像关键区域。
Vision Eagle Attention: a new lens for advancing image classification
- 用卷积生成注意力图,自动识别重要视觉区域
- 在三个数据集上准确率显著提升,最高达98.7%
- 轻量设计适合部署,可扩展至检测与分割任务
在计算机视觉任务中,精准聚焦图像中的关键区域对提升模型性能至关重要,尤其当特征微小、隐蔽或分散时。传统卷积神经网络(CNN)对图像各区域一视同仁,导致特征提取效率低下。为此,本文提出Vision Eagle Attention,一种基于卷积的空间注意力机制,通过卷积捕捉局部空间特征并生成注意力图,有选择性地强化信息丰富的区域,抑制无关背景。该机制被集成于轻量级ResNet-18架构中,实验在FashionMNIST、Intel Image Classification和OracleMNIST三个主流基准数据集上验证,结果表明分类准确率显著提升,最高达98.7%。该方法具备良好的可扩展性,适用于目标检测、分割与视觉跟踪等任务,为多种视觉应用提供高效解决方案。代码已开源:https://github.com/MahmudulHasan11085/Vision-Eagle-Attention.git。
原文摘要 · Abstract (English)
In computer vision tasks, the ability to focus on relevant regions within an image is crucial for improving model performance, particularly when key features are small, subtle, or spatially dispersed. Convolutional neural networks (CNNs) typically treat all regions of an image equally, which can lead to inefficient feature extraction. To address this challenge, I have introduced Vision Eagle Attention, a novel attention mechanism that enhances visual feature extraction using convolutional spatial attention. The model applies convolution to capture local spatial features and generates an attention map that selectively emphasizes the most informative regions of the image. This attention mechanism enables the model to focus on discriminative features while suppressing irrelevant background information. I have integrated Vision Eagle Attention into a lightweight ResNet-18 architecture, demonstrating that this combination results in an efficient and powerful model. I have evaluated the performance of the proposed model on three widely used benchmark datasets: FashionMNIST, Intel Image Classification, and OracleMNIST, with a primary focus on image classification. Experimental results show that the proposed approach improves classification accuracy. Additionally, this method has the potential to be extended to other vision tasks, such as object detection, segmentation, and visual tracking, offering a computationally efficient solution for a wide range of vision-based applications. Code is available at: https://github.com/MahmudulHasan11085/Vision-Eagle-Attention.git
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。