Comparison of machine learning and deep learning for view identification from cardiac magnetic resonance images.
Comparison of machine learning and deep learning for view identification from cardiac magnetic resonance images.
复制标题
DOI:
10.1016/j.clinimag.2021.11.013
复制
发表时间:
2022-03
期刊:
影响因子:
2.1
通讯作者:
Patel AR
中科院分区:
文献类型:
--
作者:
Chauhan D;Anyanwu E;Goes J;Besser SA;Anand S;Madduri R;Getty N;Kelle S;Kawaji K;Mor-Avi V;Patel AR
Artificial intelligence based interpretation of cardiac magnetic resonance (CMR) studies relies on the ability to accurately identify the imaging plane, which can be achieved by both deep learning (DL) and machine learning (ML) techniques. We aimed to compare the accuracy of ML and DL for CMR view classification and to identify potential pitfalls during training and testing of the algorithms. Both DL and ML algorithms accurately classified CMR images, but DL outperformed ML (95% and 90%) when classifying images with complex heart anatomy. Reaching this level of accuracy required training on a carefully curated cohort of studies. Artificial intelligence is increasingly utilized to aid in the interpretation of cardiac magnetic resonance (CMR) studies. One of the first steps is the identification of the imaging plane depicted, which can be achieved by both deep learning (DL) and classical machine learning (ML) techniques without user input. We aimed to compare the accuracy of ML and DL for CMR view classification and to identify potential pitfalls during training and testing of the algorithms. To train our DL and ML algorithms, we first established datasets by retrospectively selecting 200 CMR cases. The models were trained using two different cohorts (passively and actively curated) and applied data augmentation to enhance training. Once trained, the models were validated on an external dataset, consisting of 20 cases acquired at another center. We then compared accuracy metrics and applied class activation mapping (CAM) to visualize DL model performance. The DL and ML models trained with the passively-curated CMR cohort were 99.1% and 99.3% accurate on the validation set, respectively. However, when tested on the CMR cases with complex anatomy, both models performed poorly. After training and testing our models again on all 200 cases (active cohort), validation on the external dataset resulted in 95% and 90% accuracy, respectively. The CAM analysis depicted heat maps that demonstrated the importance of carefully curating the datasets to be used for training. Both DL and ML models can accurately classify CMR images, but DL outperformed ML when classifying images with complex heart anatomy. Deep learning outperformed machine learning (95% vs 90%) when classifying images with complex heart anatomy. Reaching this level of accuracy required training on a carefully curated cohort of studies.
登录
查看更多内容
DOI:
10.1148/ryai.2020190043
发表时间:
2020-05-01
期刊:
RADIOLOGY-ARTIFICIAL INTELLIGENCE
影响因子:
--
作者:
Reyes, Mauricio;Meier, Raphael;Wiest, Roland
通讯作者:
Wiest, Roland
影响因子:
4.5
作者:
Thrall, James H.;Li, Xiang;Brink, James
通讯作者:
Brink, James
影响因子:
15.2
作者:
Madani, Ali;Arnaout, Ramy;Arnaout, Rima
通讯作者:
Arnaout, Rima
影响因子:
8.1
作者:
Johnson, Justin M.;Khoshgoftaar, Taghi M.
通讯作者:
Khoshgoftaar, Taghi M.
影响因子:
3
作者:
McNemar, Quinn
通讯作者:
McNemar, Quinn