Skin lesion recognition via global-local attention and dual-branch input network

被引:1
|
作者
Tan, Ling [1 ]
Wu, Hui [1 ]
Xia, Jingming [2 ]
Liang, Ying [1 ]
Zhu, Jining [1 ]
机构
[1] Nanjing Univ Informat Sci & Technol, Sch Comp Sci, 219, Ningliu Rd, Nanjing 210044, Jiangsu, Peoples R China
[2] Nanjing Univ Informat Sci & Technol, Sch Artificial Intelligence, 219, Ningliu Rd, Nanjing 210044, Jiangsu, Peoples R China
关键词
Dual -branch input network; Global context modeling; Global -local attention mechanism; Residual network; Skin lesion recognition; CANCER DETECTION; MELANOMA; CLASSIFICATION; MODEL;
D O I
10.1016/j.engappai.2023.107385
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Owing to the deficiency of training data, the variety of skin lesion shapes for different patients, inter-class similarity, and intra-class variation, the recognition accuracy and speed in skin lesion recognition based on deep learning remain challenging. To solve the above issues, we propose a Dual-Branch and Global-Local Attention network based on ResNet50 (DGLA-ResNet50) to reduce model parameters and improve identification accuracy. For inter-class and intra-class problems, a global-local attention mechanism is designed to obtain the dependency information of query points in horizontal and vertical directions in turn for obtaining the global information indirectly. And we choose to obtain the local information through multiple convolutions simultaneously. Aiming at insufficient samples, the attention mechanism is lightened from the perspectives of reducing query points and the points associated with them, so as to reduce parameter redundancy. In addition, we present a dual-branch input network to deal with the problem of image variety, in which two branches are used to extract image features with different resolutions for fusion, so as to expand the network receptive field. We evaluated DGLA-ResNet50 on ISIC2018 and ISIC2019. The experimental results on ISIC2018 demonstrate that DGLA-ResNet50 has a recognition accuracy of 90.71%, while the parameters of the model are only 104.2M and the FLOPs value is 15.6G. Compared with the common models, our model obtains significantly better performance. The results indicate that DGLA-ResNet50 can improve the accuracy well while ensuring the lightweight of the model, and can prospectively assist doctors in the rapid diagnosis of skin lesions.
引用
收藏
页数:14
相关论文
共 50 条
  • [1] DBGA-Net: Dual-Branch Global-Local Attention Network for Remote Sensing Scene Classification
    Xia, Jingming
    Zhou, Yao
    Tan, Ling
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2023, 20
  • [2] DHUnet: Dual-branch hierarchical global-local fusion network for whole slide image segmentation
    Wang, Lian
    Pan, Liangrui
    Wang, Hetian
    Liu, Mingting
    Feng, Zhichao
    Rong, Pengfei
    Chen, Zuo
    Peng, Shaoliang
    [J]. BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2023, 85
  • [3] Residual cosine similar attention and bidirectional convolution in dual-branch network for skin lesion image classification
    Li, Aolun
    Zhang, Dezhi
    Yu, Long
    Kang, Xiaojing
    Tian, Shengwei
    Wu, Weidong
    You, Hongfeng
    Huo, Xiangzuo
    [J]. ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2024, 133
  • [4] Dual-branch self-attention network for pedestrian attribute recognition
    Liu, Zhenyu
    Zhang, Zhang
    Li, Da
    Zhang, Peng
    Shan, Caifeng
    [J]. PATTERN RECOGNITION LETTERS, 2022, 163 : 112 - 120
  • [5] An attention-based RGBD dual-branch gesture recognition network
    Chen, Bo
    Xie, Pengwei
    Hao, Nan
    [J]. 2021 PROCEEDINGS OF THE 40TH CHINESE CONTROL CONFERENCE (CCC), 2021, : 8022 - 8027
  • [6] A Global-Local Dual Branch Network for Congested Crowd Counting
    Di H.
    Song L.
    Yu X.
    Wang W.
    [J]. Beijing Ligong Daxue Xuebao/Transaction of Beijing Institute of Technology, 2022, 42 (11): : 1175 - 1183
  • [7] Brain Tumor Image Segmentation Based on Global-Local Dual-Branch Feature Fusion
    Jia, Zhaonian
    Hong, Yi
    Ma, Tiantian
    Ren, Zihang
    Shi, Shuang
    Hou, Alin
    [J]. PATTERN RECOGNITION AND COMPUTER VISION, PRCV 2023, PT V, 2024, 14429 : 381 - 393
  • [8] Dual-ATME: Dual-Branch Attention Network for Micro-Expression Recognition
    Zhou, Haoliang
    Huang, Shucheng
    Li, Jingting
    Wang, Su-Jing
    [J]. ENTROPY, 2023, 25 (03)
  • [9] TrUNet: Dual-Branch Network by Fusing CNN and Transformer for Skin Lesion Segmentation
    Chen, Wei
    Mu, Qian
    Qi, Jie
    [J]. IEEE Access, 2024, 12 : 144174 - 144185
  • [10] Global-local attention for emotion recognition
    Le, Nhat
    Nguyen, Khanh
    Nguyen, Anh
    Le, Bac
    [J]. NEURAL COMPUTING & APPLICATIONS, 2022, 34 (24): : 21625 - 21639