arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2108.03124cs.CV

Contrastive Learning for View Classification of Echocardiograms

Ultromics Ltd.(奥特罗米克斯有限公司);Metalynx Ltd.(梅塔林克斯有限公司);Imperial College London(帝国理工学院);King’s College London(伦敦国王学院)

Agisilaos Chartsias, Shan Gao, Angela Mumith, Jorge Oliveira, Kanwal Bhatia, Bernhard Kainz, Arian Beqiri

英文摘要:

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 naive 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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑