A Pipeline for the Implementation and Visualization of Explainable Machine Learning for Medical Imaging Using Radiomics Features

被引:27
作者
Severn, Cameron [1 ]
Suresh, Krithika [1 ]
Gorg, Carsten [1 ]
Choi, Yoon Seong [2 ]
Jain, Rajan [3 ,4 ]
Ghosh, Debashis [1 ]
机构
[1] Univ Colorado, Dept Biostat & Informat, Aurora, CO 80045 USA
[2] Yonsei Univ, Dept Radiol, Coll Med, Seoul 03722, South Korea
[3] NYU, Dept Radiol, Grossman Sch Med, New York, NY 10016 USA
[4] NYU, Dept Neurosurg, Grossman Sch Med, New York, NY 10016 USA
关键词
explainable machine learning; medical imaging; information visualization; radiomics; PREDICTION; INFORMATION; BIOMARKER; SELECTION; DISEASE; IMAGES; MODELS; IDH;
D O I
10.3390/s22145205
中图分类号
O65 [分析化学];
学科分类号
070302 ; 081704 ;
摘要
Machine learning (ML) models have been shown to predict the presence of clinical factors from medical imaging with remarkable accuracy. However, these complex models can be difficult to interpret and are often criticized as "black boxes". Prediction models that provide no insight into how their predictions are obtained are difficult to trust for making important clinical decisions, such as medical diagnoses or treatment. Explainable machine learning (XML) methods, such as Shapley values, have made it possible to explain the behavior of ML algorithms and to identify which predictors contribute most to a prediction. Incorporating XML methods into medical software tools has the potential to increase trust in ML-powered predictions and aid physicians in making medical decisions. Specifically, in the field of medical imaging analysis the most used methods for explaining deep learning-based model predictions are saliency maps that highlight important areas of an image. However, they do not provide a straightforward interpretation of which qualities of an image area are important. Here, we describe a novel pipeline for XML imaging that uses radiomics data and Shapley values as tools to explain outcome predictions from complex prediction models built with medical imaging with well-defined predictors. We present a visualization of XML imaging results in a clinician-focused dashboard that can be generalized to various settings. We demonstrate the use of this workflow for developing and explaining a prediction model using MRI data from glioma patients to predict a genetic mutation.
引用
收藏
页数:16
相关论文
共 53 条
[1]   Prediction of gestational diabetes based on nationwide electronic health records [J].
Artzi, Nitzan Shalom ;
Shilo, Smadar ;
Hadar, Eran ;
Rossman, Hagai ;
Barbash-Hazan, Shiri ;
Ben-Haroush, Avi ;
Balicer, Ran D. ;
Feldman, Becca ;
Wiznitzer, Arnon ;
Segal, Eran .
NATURE MEDICINE, 2020, 26 (01) :71-+
[2]   Explainable Artificial Intelligence (XAI): Concepts, taxonomies, opportunities and challenges toward responsible AI [J].
Barredo Arrieta, Alejandro ;
Diaz-Rodriguez, Natalia ;
Del Ser, Javier ;
Bennetot, Adrien ;
Tabik, Siham ;
Barbado, Alberto ;
Garcia, Salvador ;
Gil-Lopez, Sergio ;
Molina, Daniel ;
Benjamins, Richard ;
Chatila, Raja ;
Herrera, Francisco .
INFORMATION FUSION, 2020, 58 :82-115
[3]   Hyperopt: A Python library for model selection and hyperparameter optimization [J].
Bergstra, James ;
Komer, Brent ;
Eliasmith, Chris ;
Yamins, Dan ;
Cox, David D .
Computational Science and Discovery, 2015, 8 (01)
[4]   Impact of rescanning and repositioning on radiomic features employing a multi-object phantom in magnetic resonance imaging [J].
Bernatz, Simon ;
Zhdanovich, Yauheniya ;
Ackermann, Jorg ;
Koch, Ina ;
Wild, Peter J. ;
dos Santos, Daniel Pinto ;
Vogl, Thomas J. ;
Kaltenbach, Benjamin ;
Rosbach, Nicolas .
SCIENTIFIC REPORTS, 2021, 11 (01)
[5]   Random forests [J].
Breiman, L .
MACHINE LEARNING, 2001, 45 (01) :5-32
[6]   XGBoost: A Scalable Tree Boosting System [J].
Chen, Tianqi ;
Guestrin, Carlos .
KDD'16: PROCEEDINGS OF THE 22ND ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2016, :785-794
[7]   The Cancer Imaging Archive (TCIA): Maintaining and Operating a Public Information Repository [J].
Clark, Kenneth ;
Vendt, Bruce ;
Smith, Kirk ;
Freymann, John ;
Kirby, Justin ;
Koppel, Paul ;
Moore, Stephen ;
Phillips, Stanley ;
Maffitt, David ;
Pringle, Michael ;
Tarbox, Lawrence ;
Prior, Fred .
JOURNAL OF DIGITAL IMAGING, 2013, 26 (06) :1045-1057
[8]  
Covert I., 2021, J. Mach. Learn. Res., V22, p209:1
[9]   IDH mutations in glioma and acute myeloid leukemia [J].
Dang, Lenny ;
Jin, Shengfang ;
Su, Shinsan M. .
TRENDS IN MOLECULAR MEDICINE, 2010, 16 (09) :387-397
[10]   Convolutional Neural Networks for the evaluation of cancer in Barrett's esophagus: Explainable AI to lighten up the black-box [J].
de Souza Jr, Luis A. ;
Mendel, Robert ;
Strasser, Sophia ;
Ebigbo, Alanna ;
Probst, Andreas ;
Messmann, Helmut ;
Papa, Joao P. ;
Palm, Christoph .
COMPUTERS IN BIOLOGY AND MEDICINE, 2021, 135