Edge Attention Module for Object Classification

Title

Edge Attention Module for Object Classification

Creator

Roy, Santanu; Suresh, Ashvath; Adhikari, Prashant; Gupta, Archit

Description

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. All the codes along with graphs, and their classification reports are shared on an anonymous GitHub link: https://anonymous.4open.science/r/Object-Classification-7BE5. 2025 IEEE.

Source

Proceedings of the International Joint Conference on Neural Networks;

Date

01-01-2025

Publisher

Institute of Electrical and Electronics Engineers Inc.

Subject

Convolutional Neural Network (CNN); Edge Attention Module (EAM); Max-Min Pooling; Vision Transformer (ViT)

Coverage

Roy S., Dept. of Computer Science, NIIT University, Jaipur, India; Suresh A., Dept. of Computer Science and Engg, Christ (Deemed to be University), Bangalore, India; Adhikari P., Dept. of Computer Science, Christ (Deemed to be University), Bangalore, India; Gupta A., Dept. of Computer Science, NIIT University, Jaipur, India

Rights

Restricted Access; Hardcopy may be available in the library

Relation

ISSN: 21614393; ISBN: 979-833151042-8; CODEN: 85OFA

Format

online

Language

English

Type

Conference paper

Identifier

https://doi.org/10.1109/IJCNN64981.2025.11227631

https://www.scopus.com/pages/publications/105023977072?origin=resultslist