Endoscopic Image Classification Based on Explainable Deep Learning

被引:29
作者
Mukhtorov, Doniyorjon [1 ]
Rakhmonova, Madinakhon [1 ]
Muksimova, Shakhnoza [1 ]
Cho, Young-Im [1 ]
机构
[1] Gachon Univ, Dept IT Convergence Engn, Seongnam Si 461701, South Korea
关键词
explainable ai; deep learning; classification; endoscopic image; CONVOLUTIONAL NEURAL-NETWORKS; COMPUTER-AIDED DETECTION; DIAGNOSIS; CANCER;
D O I
10.3390/s23063176
中图分类号
O65 [分析化学];
学科分类号
070302 ; 081704 ;
摘要
Deep learning has achieved remarkably positive results and impacts on medical diagnostics in recent years. Due to its use in several proposals, deep learning has reached sufficient accuracy to implement; however, the algorithms are black boxes that are hard to understand, and model decisions are often made without reason or explanation. To reduce this gap, explainable artificial intelligence (XAI) offers a huge opportunity to receive informed decision support from deep learning models and opens the black box of the method. We conducted an explainable deep learning method based on ResNet152 combined with Grad-CAM for endoscopy image classification. We used an open-source KVASIR dataset that consisted of a total of 8000 wireless capsule images. The heat map of the classification results and an efficient augmentation method achieved a high positive result with 98.28% training and 93.46% validation accuracy in terms of medical image classification.
引用
收藏
页数:14
相关论文
共 54 条
[11]  
Fu RG, 2020, Arxiv, DOI arXiv:2008.02312
[12]   SD-CNN: A shallow-deep CNN for improved breast cancer diagnosis [J].
Gao, Fei ;
Wu, Teresa ;
Li, Jing ;
Zheng, Bin ;
Ruan, Lingxiang ;
Shang, Desheng ;
Patel, Bhavika .
COMPUTERIZED MEDICAL IMAGING AND GRAPHICS, 2018, 70 :53-62
[13]   Convolutional neural networks for computer-aided detection or diagnosis in medical image analysis: An overview [J].
Gao, Jun ;
Jiang, Qian ;
Zhou, Bo ;
Chen, Daozheng .
MATHEMATICAL BIOSCIENCES AND ENGINEERING, 2019, 16 (06) :6536-6561
[14]   A self-learning teacher-student framework for gastrointestinal image classification [J].
Gjestang, Henrik L. ;
Hicks, Steven A. ;
Thambawita, Vajira ;
Halvorsen, Pal ;
Riegler, Michael A. .
2021 IEEE 34TH INTERNATIONAL SYMPOSIUM ON COMPUTER-BASED MEDICAL SYSTEMS (CBMS), 2021, :539-544
[15]   TCD-Net: A Novel Deep Learning Framework for Fully Polarimetric Change Detection Using Transfer Learning [J].
Habibollahi, Rezvan ;
Seydi, Seyd Teymoor ;
Hasanlou, Mahdi ;
Mahdianpari, Masoud .
REMOTE SENSING, 2022, 14 (03)
[16]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[17]   Application of artificial intelligence using a convolutional neural network for detecting gastric cancer in endoscopic images [J].
Hirasawa, Toshiaki ;
Aoyama, Kazuharu ;
Tanimoto, Tetsuya ;
Ishihara, Soichiro ;
Shichijo, Satoki ;
Ozawa, Tsuyoshi ;
Ohnishi, Tatsuya ;
Fujishiro, Mitsuhiro ;
Matsuo, Keigo ;
Fujisaki, Junko ;
Tada, Tomohiro .
GASTRIC CANCER, 2018, 21 (04) :653-660
[18]   Densely Connected Convolutional Networks [J].
Huang, Gao ;
Liu, Zhuang ;
van der Maaten, Laurens ;
Weinberger, Kilian Q. .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :2261-2269
[19]   Efficient Detection of Knee Anterior Cruciate Ligament from Magnetic Resonance Imaging Using Deep Learning Approach [J].
Javed Awan, Mazhar ;
Mohd Rahim, Mohd Shafry ;
Salim, Naomie ;
Mohammed, Mazin Abed ;
Garcia-Zapirain, Begonya ;
Abdulkareem, Karrar Hameed .
DIAGNOSTICS, 2021, 11 (01)
[20]   LayerCAM: Exploring Hierarchical Class Activation Maps for Localization [J].
Jiang, Peng-Tao ;
Zhang, Chang-Bin ;
Hou, Qibin ;
Cheng, Ming-Ming ;
Wei, Yunchao .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2021, 30 :5875-5888