An Evolutionary Attention-Based Network for Medical Image Classification

Abstract

Deep learning has become a primary choice in medical image analysis due to its powerful representation capability. However, most existing deep learning models designed for medical image classification can only perform well on a specific disease. The performance drops dramatically when it comes to other diseases. Generalizability remains a challenging problem. In this paper, we propose an evolutionary attention-based network (EDCA-Net), which is an effective and robust network for medical image classification tasks. To extract task-related features from a given medical dataset, we first propose the densely connected attentional network (DCA-Net) where feature maps are automatically channel-wise weighted, and the dense connectivity pattern is introduced to improve the efficiency of information flow. To improve the model capability and generalizability, we introduce two types of evolution: intra- and inter-evolution. The intra-evolution optimizes the weights of DCA-Net, while the inter-evolution allows two instances of DCA-Net to exchange training experience during training. The evolutionary DCA-Net is referred to as EDCA-Net. The EDCA-Net is evaluated on four publicly accessible medical datasets of different diseases. Experiments showed that the EDCA-Net outperforms the state-of-the-art methods on three datasets and achieves comparable performance on the last dataset, demonstrating good generalizability for medical image classification.

EENet: Application of convolutional neural network‐based deep learning methods in bone tumor pathological diagnosis

“…Asif et al 23 proposed the DCDS‐Net model introduced for the diagnosis of Gastrointestinal diseases and explored the feasibility of block‐wise fine‐tuning using transfer learning on the proposed model to reduce overfitting. Zhu et al 24 proposed an evolutionary attention‐based network (EDCA‐Net). The weights of EDCA‐Net were also initialized through the adoption of transfer learning, acquiring prior knowledge from ImageNet, and leading to performance improvement.…”

Blockchain-Based Deep CNN for Brain Tumor Prediction Using MRI Scans

“…More trustworthy early tumor detection methods, including computer-aided diagnosis (CAD) [ 8 ], are therefore urgently needed. Applications in medical diagnosis that depend on feature extraction from medical images, such as attempting to differentiate between healthy and pathological tissue, have relied heavily on CAD techniques [ 9 , 10 , 11 ]. Techniques for medical imaging are essential for early tumor detection and for improving treatment options.…”