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 条
  • [41] Classification of ECG Using Ensemble of Residual CNNs with Attention Mechanism
    Nejedly, Petr
    Ivora, Adam
    Viscor, Ivo
    Koscova, Zuzana
    Smisek, Radovan
    Jurak, Pavel
    Plesinger, Filip
    2021 COMPUTING IN CARDIOLOGY (CINC), 2021,
  • [42] Mass Classification of Breast Mammogram Based on Attention Mechanism and Transfer Learning
    Xu Wenhui
    Pei Yijian
    Gao Donglin
    Zhu Jiude
    Liu Yunkai
    LASER & OPTOELECTRONICS PROGRESS, 2021, 58 (04)
  • [43] A novel remotely sensed image classification based on ensemble learning and feature integration
    Liu Pei
    Du Pei-Jun
    Tan Kun
    JOURNAL OF INFRARED AND MILLIMETER WAVES, 2014, 33 (03) : 311 - 317
  • [44] A Few Shot Image Ensemble Classification Algorithm Based on Meta-Learning
    Peng, He
    Qun, Liu
    PROCEEDINGS OF 2021 2ND INTERNATIONAL CONFERENCE ON ARTIFICIAL INTELLIGENCE AND INFORMATION SYSTEMS (ICAIIS '21), 2021,
  • [45] DEEP ENSEMBLE LEARNING FOR LAND COVER CLASSIFICATION BASED ON HYPERSPECTRAL PRISMA IMAGE
    Kalantar, Bahareh
    Seydi, Seyd Teymoor
    Ueda, Naonori
    Saeidi, Vahideh
    Halin, Alfian Abdul
    Shabani, Farzin
    2022 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS 2022), 2022, : 3612 - 3615
  • [46] Fine-Grained Image Classification Based on Feature Fusion and Ensemble Learning
    Zhang, Wenli
    Wei, Song
    LASER & OPTOELECTRONICS PROGRESS, 2024, 61 (22)
  • [47] Multiscale CNNs Ensemble Based Self-Learning for Hyperspectral Image Classification
    Fang, Leyuan
    Zhao, Wenke
    He, Nanjun
    Zhu, Jian
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2020, 17 (09) : 1593 - 1597
  • [48] Learning a Hierarchical Global Attention for Image Classification
    Cao, Kerang
    Gao, Jingyu
    Choi, Kwang-nam
    Duan, Lini
    FUTURE INTERNET, 2020, 12 (11): : 1 - 11
  • [49] Deep ensemble transfer learning-based framework for mammographic image classification
    Parita Oza
    Paawan Sharma
    Samir Patel
    The Journal of Supercomputing, 2023, 79 : 8048 - 8069
  • [50] Genre-based Image Classification Using Ensemble Learning for Online Flyers
    Pourashraf, Payam
    Tomuro, Noriko
    Apostolova, Emilia
    SEVENTH INTERNATIONAL CONFERENCE ON DIGITAL IMAGE PROCESSING (ICDIP 2015), 2015, 9631