DACB-Net: Dual Attention Guided Compact Bilinear Convolution Neural Network for Skin Disease Classification
Not on the map. Why?
Abstract (arXiv)
This paper introduces the three-branch Dual Attention-Guided Compact Bilinear CNN (DACB-Net) by focusing on learning from disease-specific regions to enhance accuracy and alignment. A global branch compensates for lost discriminative features, generating Attention Heat Maps (AHM) for relevant cropped regions. Finally, the last pooling layers of global and local branches are concatenated for fine-tuning, which offers a comprehensive solution to the challenges posed by skin disease diagnosis. Although current CNNs employ Stochastic Gradient Descent (SGD) for discriminative feature learning, using distinct pairs of local image patches to compute gradients and incorporating a modulation factor in the loss for focusing on complex data during training. However, this approach can lead to dataset imbalance, weight adjustments, and vulnerability to overfitting. The proposed solution combines two supervision branches and a novel loss function to address these issues, enhancing performance and interpretability. The framework integrates data augmentation, transfer learning, and fine-tuning to tackle data imbalance to improve classification performance, and reduce computational costs. Simulations on the HAM10000 and ISIC2019 datasets demonstrate the effectiveness of this approach, showcasing a 2.59% increase in accuracy compared to the state-of-the-art.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 12 of its 52 references that are in the map. First posted on arXiv 3 Jul 2024.
Builds on
12 of 52 references.
References · 12
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
- 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,590
- 2018CBAM: Convolutional Block Attention Modulecited by 2,463
- 2016Xception: Deep Learning with Depthwise Separable Convolutionscited by 2,320
- 2015Understanding Neural Networks Through Deep Visualizationcited by 332
- 2019GCNet: Non-local Networks Meet Squeeze-Excitation Networks and Beyondcited by 331
- 2015Bilinear CNNs for Fine-grained Visual Recognitioncited by 228
- 2020PAD-UFES-20: a skin lesion dataset composed of patient data and clinical images collected from smartphonescited by 116
- 2020Semi-supervised Medical Image Classification with Relation-driven Self-ensembling Modelcited by 59
- 2017Automatic segmentation of MR brain images with a convolutional neural networkcited by 26
- 2021Single Model Deep Learning on Imbalanced Small Datasets for Skin Lesion Classificationcited by 23
- 2018Attention-Gated Networks for Improving Ultrasound Scan Plane Detectioncited by 22
Built on by
Not cited yet.