Demystifying Deep Learning Models for Retinal OCT Disease Classification using Explainable AI

被引:8
作者
Apon, Tasnim Sakib [1 ]
Hasan, Mohammad Mahmudul [2 ]
Islam, Abrar [2 ]
Alam, Md Golam Rabiul [1 ]
机构
[1] BRAC Univ, Dept Comp Sci & Engn, Dhaka, Bangladesh
[2] Islamic Univ Technol, Elect & Elect Engn, Dhaka, Bangladesh
来源
2021 IEEE ASIA-PACIFIC CONFERENCE ON COMPUTER SCIENCE AND DATA ENGINEERING (CSDE) | 2021年
关键词
Medical Image Processing; Explainable AI; Retinal OCT; Lime; Image Classification; Deep Neural Network; AI in Healthcare;
D O I
10.1109/CSDE53843.2021.9718400
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In the world of medical diagnostics, the adoption of various deep learning techniques is quite common as well as effective, and its statement is equally true when it comes to implementing it into the retina Optical Coherence Tomography (OCT) sector. However, firstly, these techniques have the black box characteristics that prevent the medical professionals from completely trusting the results generated from them. Secondly, the lack of precision of these methods restricts their implementation in clinical and complex cases, and finally, the existing works and models on the OCT classification are substantially large and complicated and they require a considerable amount of memory and computational power, reducing the quality of classifiers in real-time applications. To meet these problems, in this paper a self-developed CNN model has been proposed which is comparatively smaller and simpler along with the use of Lime that introduces Explainable AI to the study and helps to increase the interpretability of the model. This addition will be an asset to the medical experts for getting major and detailed information and will help them in making final decisions and will also reduce the opacity and vulnerability of the conventional deep learning models.
引用
收藏
页数:6
相关论文
共 9 条
[1]  
Awais M, 2017, IEEE I C SIGNAL IMAG, P489, DOI 10.1109/ICSIPA.2017.8120661
[2]   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
[3]   OPTICAL COHERENCE TOMOGRAPHY [J].
HUANG, D ;
SWANSON, EA ;
LIN, CP ;
SCHUMAN, JS ;
STINSON, WG ;
CHANG, W ;
HEE, MR ;
FLOTTE, T ;
GREGORY, K ;
PULIAFITO, CA ;
FUJIMOTO, JG .
SCIENCE, 1991, 254 (5035) :1178-1181
[4]  
Joshi A., 2020, INTERPRETABLE ANNOTA, P54, DOI [10.1007/978-3-030-61166-8_6, DOI 10.1007/978-3-030-61166-8_6]
[5]  
Kepp T, INMEDICAL IMAGING 20, V11314
[6]   Identifying Medical Diagnoses and Treatable Diseases by Image-Based Deep Learning [J].
Kermany, Daniel S. ;
Goldbaum, Michael ;
Cai, Wenjia ;
Valentim, Carolina C. S. ;
Liang, Huiying ;
Baxter, Sally L. ;
McKeown, Alex ;
Yang, Ge ;
Wu, Xiaokang ;
Yan, Fangbing ;
Dong, Justin ;
Prasadha, Made K. ;
Pei, Jacqueline ;
Ting, Magdalena ;
Zhu, Jie ;
Li, Christina ;
Hewett, Sierra ;
Dong, Jason ;
Ziyar, Ian ;
Shi, Alexander ;
Zhang, Runze ;
Zheng, Lianghong ;
Hou, Rui ;
Shi, William ;
Fu, Xin ;
Duan, Yaou ;
Huu, Viet A. N. ;
Wen, Cindy ;
Zhang, Edward D. ;
Zhang, Charlotte L. ;
Li, Oulan ;
Wang, Xiaobo ;
Singer, Michael A. ;
Sun, Xiaodong ;
Xu, Jie ;
Tafreshi, Ali ;
Lewis, M. Anthony ;
Xia, Huimin ;
Zhang, Kang .
CELL, 2018, 172 (05) :1122-+
[7]  
Lee CS, 2017, OPHTHALMOL RETINA, V1, P322, DOI 10.1016/j.oret.2016.12.009
[8]  
Singh A, 2020, IEEE INT CONF ROBOT, P2167, DOI [10.1109/icra40945.2020.9197020, 10.1109/ICRA40945.2020.9197020]
[9]   Deep learning for quality assessment of retinal OCT images [J].
Wang, Jing ;
Deng, Guohua ;
Li, Wanyue ;
Chen, Yiwei ;
Gao, Feng ;
Liu, Hu ;
He, Yi ;
Shi, Guohua .
BIOMEDICAL OPTICS EXPRESS, 2019, 10 (12) :6057-6072