Neural architecture search of echocardiography view classifiers

Azarmehr, Neda, Ye, Xujiong, Howard, James P , Lane, Elisabeth S, Labs, Robert, Shun-Shin, Matthew J, Cole, Graham D, Bidaut, Luc, Francis, Darrel P and Zolgharni, Massoud (2021) Neural architecture search of echocardiography view classifiers. Journal of Medical Imaging, 8 (3). 034002. ISSN 2329-4302

Full content URL:

Neural architecture search of echocardiography view classifiers
Published Manuscript
JMI-20323R_online.pdf - Whole Document

Item Type:Article
Item Status:Live Archive


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.

Keywords:deep learning; echocardiography; neural architecture search; view classification; AutoML
Subjects:G Mathematical and Computer Sciences > G700 Artificial Intelligence
G Mathematical and Computer Sciences > G400 Computer Science
G Mathematical and Computer Sciences > G760 Machine Learning
Divisions:College of Science > School of Computer Science
Related URLs:
ID Code:45384
Deposited On:28 Jun 2021 12:01

Repository Staff Only: item control page