Contrastive learning for view classification of echocardiograms
File(s)2108.03124.pdf (1.94 MB)
Accepted version
OA Location
Author(s)
Type
Conference Paper
Abstract
Analysis of cardiac ultrasound images is commonly performed in routine clinical practice for quantification of cardiac function. Its increasing automation frequently employs deep learning networks that are trained to predict disease or detect image features. However, such models are extremely data-hungry and training requires labelling of many thousands of images by experienced clinicians. Here we propose the use of contrastive learning to mitigate the labelling bottleneck. We train view classification models for imbalanced cardiac ultrasound datasets and show improved performance for views/classes for which minimal labelled data is available. Compared to a naïve baseline model, we achieve an improvement in F1 score of up to 26% in those views while maintaining state-of-the-art performance for the views with sufficiently many labelled training observations.
Date Issued
2021-09-21
Date Acceptance
2021-06-11
Citation
Lecture Notes in Computer Science, 2021, 12967, pp.149-158
ISSN
0302-9743
Publisher
Springer
Start Page
149
End Page
158
Journal / Book Title
Lecture Notes in Computer Science
Volume
12967
Copyright Statement
© 2021 Springer Nature Switzerland AG. The final publication is available at Springer via https://link.springer.com/chapter/10.1007/978-3-030-87583-1_15
Source
24th International Conference on Medical Image Computing and Computer Assisted Intervention
Subjects
Science & Technology
Technology
Life Sciences & Biomedicine
Computer Science, Artificial Intelligence
Engineering, Biomedical
Radiology, Nuclear Medicine & Medical Imaging
Computer Science
Engineering
Contrastive learning
Classification
Echocardiography
RECOMMENDATIONS
LOCALIZATION
Artificial Intelligence & Image Processing
Publication Status
Published
Start Date
2021-09-27
Coverage Spatial
Virtual
Date Publish Online
2021-09-21