EMDA-Net: Earth Mover’s Distance (EMD) influenced Attention-aided Neural Network for Medical Image Classification
Medical image analysis methods need to be more generalized in order to meet the substantial problems associated with cancer detection and treatment, including skin, cervical, and breast cancer. In this work, we propose a novel lightweight convolutional neural network (CNN) architecture, named EMDA-Net (Earth Mover’s Distance (EMD) influenced Attention-aided Neural Network), specifically designed to meet the requirements of cancer subtype classification. It is built on the MobileNetV2 backbone and is equipped with hybrid statistical mechanism-aided attention mechanisms. Our model ensures high-performance while maintaining a lightweight design by leveraging the capabilities of the MobileNetV2 architecture and the attention mechanism. To improve similarity detection across data channels, we present a novel EMD influenced attention technique. We have examined datasets related to skin, cervical, and breast cancer for experimentation. Our model especially outperforms the state-of-the-art techniques, demonstrating its effectiveness in maximizing diagnosis precision for various cancer types across the datasets. The code of the proposed model will be made public at: https://t.ly/q5DlN
Authors
- Somenath Chakraborty (ORCID: https://orcid.org/0000-0002-3880-0308)
- Ram Sarkar (ORCID: https://orcid.org/0000-0001-8813-4086)
- Surya Majumder (ORCID: https://orcid.org/0009-0007-3272-7369)
- Kanishka Mallick
- Wrick Pal
Institutions
- Bose Institute (IN)
- Jadavpur University (IN)
- West Virginia University Institute of Technology (US)
Publication Details
- Published
- 2024-10-28
- DOI
- https://doi.org/10.1109/cogmi62246.2024.00040
- Primary Topic
- AI in cancer detection
- Type
- article
- Field-Weighted Citation Impact
- 0.00