Deeply Supervised Skin Lesions Diagnosis with Stage and Branch Attention

Wei Dai, Rui Liu, Tianyi Wu, Min Wang, Jianqin Yin, Jun Liu*

*Corresponding author for this work

Research output: Journal Publications and ReviewsRGC 21 - Publication in refereed journalpeer-review

17 Citations (Scopus)

Abstract

Accurate and unbiased examinations of skin lesions are critical for the early diagnosis and treatment of skin diseases. Visual features of skin lesions vary significantly because the images are collected from patients with different lesion colours and morphologies by using dissimilar imaging equipment. Recent studies have reported that ensembled convolutional neural networks (CNNs) are practical to classify the images for early diagnosis of skin disorders. However, the practical use of these ensembled CNNs is limited as these networks are heavyweight and inadequate for processing contextual information. Although lightweight networks (e.g., MobileNetV3 and EfficientNet) were developed to achieve parameter reduction for implementing deep neural networks on mobile devices, insufficient depth of feature representation restricts the performance. To address the existing limitations, we develop a new lite and effective neural network, namely HierAttn. The HierAttn applies a novel deep supervision strategy to learn the local and global features by using multi-stage and multi-branch attention mechanisms with only one training loss. The efficacy of HierAttn was evaluated by using the dermoscopy images dataset ISIC2019 and smartphone photos dataset PAD-UFES-20 (PAD2020). The experimental results show that HierAttn achieves the best accuracy and area under the curve (AUC) among the state-of-the-art lightweight networks. © 2023 IEEE.
Original languageEnglish
Pages (from-to)719-729
JournalIEEE Journal of Biomedical and Health Informatics
Volume28
Issue number2
Online published25 Aug 2023
DOIs
Publication statusPublished - Feb 2024

Funding

This work was supported in part by the Research Grant Council (RGC) of Hong Kong under Grants 11217922, 11212321, and ECS-21212720, in part by Guangdong Province Basic and Applied Basic Research Fund Project under Grant 2019A1515110175, and in part by the Science and Technology Innovation Committee of Shenzhen under Grant Type-C SGDX20210823104001011.

Research Keywords

  • Branch attention
  • Computational modeling
  • Convolution
  • Deep supervision
  • Feature extraction
  • Lesions
  • Skin
  • Skin lesions classification
  • Stage attention
  • Training
  • Transformers
  • Vision transformer
  • Attention
  • disease classification
  • skin lesion

Fingerprint

Dive into the research topics of 'Deeply Supervised Skin Lesions Diagnosis with Stage and Branch Attention'. Together they form a unique fingerprint.

Cite this