An Attention Mechanism and Ensemble Learning based on Dermoscopic Image Classification

被引:0
|
作者
Huang, Shanchuan [1 ]
Lei, Hongwei [2 ]
Yang, Jinzhu [3 ]
Jiang, Tao [4 ]
Jin, Liuhan [1 ]
Yao, Yudong [5 ]
Grzegorzek, Marcin [6 ,7 ]
Li, Chen [1 ,3 ]
机构
[1] Northeastern Univ, Coll Med & Biol Informat Engn, Microscop Image & Med Image Anal Grp, Shenyang, Peoples R China
[2] Northeastern Univ, Coll Informat Sci & Engn, Shenyang, Peoples R China
[3] Minist Educ, Key Lab Intelligent Comp Med Image, Shenyang, Liaoning, Peoples R China
[4] Chengdu Univ Tradit Chinese Med, Sch Intelligent Med, Chengdu, Peoples R China
[5] Stevens Inst Technol, Dept Elect & Comp Engn, Hoboken, NJ USA
[6] Univ Lubeck, Inst Med Informat, Lubeck, Germany
[7] German Res Ctr Artificial Intelligence, Lubeck, Germany
基金
中国国家自然科学基金;
关键词
Skin cancer; Image classification; Attention mechanism; Ensemble learning; Late fusion; MELANOMA; MUTATIONS; FREQUENCY; DIAGNOSIS; NEVI;
D O I
10.2352/J.ImagingSci.Technol.2024.68.4.040403
中图分类号
TB8 [摄影技术];
学科分类号
0804 ;
摘要
Skin tumors have become one of the most common diseases worldwide. Usually, benign skin tumors are not harmful to human health, but malignant skin tumors are highly likely to develop into skin cancer, which is life-threatening. Dermoscopy is currently the most effective method of diagnosing skin tumors. However, the complexity of skin tumor cells makes doctors' diagnoses subject to error. Therefore, it is essential to use computers for assisted diagnosis, thereby improving the diagnostic accuracy of skin tumors. In this paper, we propose Deep-skin, a model for dermoscopic image classification, which is based on both attention mechanism and ensemble learning. Considering the characteristics of dermoscopic images, we suggest embedding different attention mechanisms on top of Inception-V3 to obtain more potential features. We then improve the classification performance by late fusion of the different models. To demonstrate the effectiveness of Deep-skin, we conduct experiments and evaluations on the publicly available dataset Skin Cancer: Malignant vs. Benign and compare the performance of Deep-skin with other classification models. The experimental results indicate that Deep-skin performs well on the dataset in comparison to other models, achieving a maximum accuracy of 87.8%. In the future, we intend to investigate better classification models for automatic diagnosis of skin tumors. Such models can potentially assist physicians and patients in clinical settings.
引用
收藏
页数:14
相关论文
共 50 条
  • [21] Deep Ensemble Learning for Retinal Image Classification
    Ho, Edward
    Wang, Edward
    Youn, Saerom
    Sivajohan, Asaanth
    Lane, Kevin
    Chun, Jin
    Hutnik, Cindy M. L.
    TRANSLATIONAL VISION SCIENCE & TECHNOLOGY, 2022, 11 (10):
  • [22] Deep Learning Ensemble for Hyperspectral Image Classification
    Chen, Yushi
    Wang, Ying
    Gu, Yanfeng
    He, Xin
    Ghamisi, Pedram
    Jia, Xiuping
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2019, 12 (06) : 1882 - 1897
  • [23] Camouflage image segmentation based on transfer learning and attention mechanism
    Wu T.
    Wang L.
    Zhu J.
    Xi Tong Gong Cheng Yu Dian Zi Ji Shu/Systems Engineering and Electronics, 2022, 44 (02): : 376 - 384
  • [24] Neural architecture search based on dual attention mechanism for image classification
    Jin, Cong
    Huang, Jinjie
    Wei, Tianshu
    Chen, Yuanjian
    MATHEMATICAL BIOSCIENCES AND ENGINEERING, 2023, 20 (02) : 2691 - 2715
  • [25] A Novel ResNet50-based Attention Mechanism For Image Classification
    Zhang, Jingsi
    Yu, Xiaosheng
    Lei, Xiaoliang
    Wu, Chengdong
    JOURNAL OF APPLIED SCIENCE AND ENGINEERING, 2024, 27 (08): : 2891 - 2899
  • [26] Hyperspectral Image Classification Based on Convolution Neural Network with Attention Mechanism
    Chen Wenhao
    Jing, He
    Gang, Liu
    LASER & OPTOELECTRONICS PROGRESS, 2022, 59 (18)
  • [27] A Synergic Neural Network For Medical Image Classification Based On Attention Mechanism
    Wang Shanshan
    Zhang Tao
    Li Fei
    Ruan ZhenPing
    Yang Zhen
    Zhan Shu
    Zhang ZhiQiang
    2022 ASIA CONFERENCE ON ALGORITHMS, COMPUTING AND MACHINE LEARNING (CACML 2022), 2022, : 82 - 87
  • [28] A framework for immunofluorescence image augmentation and classification based on unsupervised attention mechanism
    Wang, Ziyi
    Zhang, Qing
    Wang, Yan
    Zhu, Min
    Li, Qingli
    JOURNAL OF BIOPHOTONICS, 2023, 16 (12)
  • [29] Brain CT image classification based on mask RCNN and attention mechanism
    Shoulin Yin
    Hang Li
    Lin Teng
    Asif Ali Laghari
    Ahmad Almadhor
    Michal Gregus
    Gabriel Avelino Sampedro
    Scientific Reports, 14 (1)
  • [30] Extreme Learning Machine-Based Ensemble Transfer Learning for Hyperspectral Image Classification
    Liu, Xiaobo
    Hu, Qiubo
    Cai, Yaoming
    Cai, Zhihua
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2020, 13 : 3892 - 3902