Neural architecture search of echocardiography view classifiers

被引:11
作者
Azarmehr, Neda [1 ]
Ye, Xujiong [1 ]
Howard, James P. [2 ]
Lane, Elisabeth S. [3 ]
Labs, Robert [3 ]
Shun-Shin, Matthew J. [2 ]
Cole, Graham D. [2 ]
Bidaut, Luc [1 ]
Francis, Darrel P. [2 ]
Zolgharni, Massoud [2 ,3 ]
机构
[1] Univ Lincoln, Sch Comp Sci, Lincoln, England
[2] Imperial Coll London, Natl Heart & Lung Inst, London, England
[3] Univ West London, Sch Comp & Engn, London, England
关键词
deep learning; echocardiography; neural architecture search; view classification; AutoML; COMPUTER-AIDED DIAGNOSIS; CLASSIFICATION;
D O I
10.1117/1.JMI.8.3.034002
中图分类号
R8 [特种医学]; R445 [影像诊断学];
学科分类号
1002 ; 100207 ; 1009 ;
摘要
Purpose: Echocardiography is the most commonly used modality for assessing the heart in clinical practice. In an echocardiographic exam, an ultrasound probe samples the heart from different orientations and positions, thereby creating different viewpoints for assessing the cardiac function. The determination of the probe viewpoint forms an essential step in automatic echocardiographic image analysis. Approach: In this study, convolutional neural networks are used for the automated identification of 14 different anatomical echocardiographic views (larger than any previous study) in a dataset of 8732 videos acquired from 374 patients. Differentiable architecture search approach was utilized to design small neural network architectures for rapid inference while maintaining high accuracy. The impact of the image quality and resolution, size of the training dataset, and number of echocardiographic view classes on the efficacy of the models were also investigated. Results: In contrast to the deeper classification architectures, the proposed models had significantly lower number of trainable parameters (up to 99.9% reduction), achieved comparable classification performance (accuracy 88.4% to 96%, precision 87.8% to 95.2%, recall 87.1% to 95.1%) and real-time performance with inference time per image of 3.6 to 12.6 ms. Conclusion: Compared with the standard classification neural network architectures, the proposed models are faster and achieve comparable classification performance. They also require less training data. Such models can be used for real-time detection of the standard views. (C) 2021 Society of Photo-Optical Instrumentation Engineers (SPIE)
引用
收藏
页数:21
相关论文
共 49 条
[1]  
Agarwal D, 2013, I S BIOMED IMAGING, P1368
[2]  
[Anonymous], 2018, INT C LEARN REPR
[3]  
[Anonymous], 2015, 3 INT C LEARN REP IC
[4]  
Azarmehr N., 2019, INT C MED IMAG DEEP
[5]  
Bai W, 2018, CLASSIFICATION BIOAP, P323
[6]  
Bello I., 2017, INT C MACHINE LEARNI, P459, DOI DOI 10.5555/3305381.3305429
[7]  
Beymer David, 2008, 2008 IEEE Computer Society Conference on Computer Vision and Pattern Recognition Workshops (CVPR Workshops), P1, DOI 10.1109/CVPRW.2008.4563008
[8]  
Beymer D., 2010, IEEE COMP SOC C COMP, P162
[9]  
Boski M, 2017, 2017 10TH INTERNATIONAL WORKSHOP ON MULTIDIMENSIONAL (ND) SYSTEMS (NDS)
[10]   Reinforcement Learning, Fast and Slow [J].
Botvinick, Matthew ;
Ritter, Sam ;
Wang, Jane X. ;
Kurth-Nelson, Zeb ;
Blundell, Charles ;
Hassabis, Demis .
TRENDS IN COGNITIVE SCIENCES, 2019, 23 (05) :408-422