arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2407.21738eess.IVcs.AIcs.CVcs.LG

利用自监督学习对超声扫描视频中的胎儿心脏切面进行分类

Leveraging Self-Supervised Learning for Fetal Cardiac Planes Classification using Ultrasound Scan Videos

Mohamed bin Zayed University of Artificial Intelligence(穆罕默德·本·扎耶德人工智能大学);Abu Dhabi Health Services Company (SEHA)(阿布扎比卫生服务公司(SEHA))

Joseph Geo Benjamin, Mothilal Asokan, Amna Alhosani, Hussain Alasmawi, Werner Gerhard Diehl, Leanne Bricker, Karthik Nandakumar, Mohammad Yaqub

AI总结:

本文研究利用双编码器自监督学习从超声视频中学习表示,以提升有限标注下胎儿心脏切面分类性能,发现数据方差比规模更重要,且 BarlowTwins 初始化结合 1% 标注数据微调可显著超越 ImageNet 和其他 SSL 方法。

摘要:

自监督学习(SSL)方法因能够直接利用底层数据分布来解决标注数据有限的情况而受到欢迎。然而,这类方法在超声(US)成像中的采用尚未得到充分探索,尤其是在胎儿评估方面。我们研究了双编码器 SSL 在利用未标注的 US 视频数据方面的潜力,以在使用有限的已标注 2D US 图像的情况下,提升具有挑战性的下游标准胎儿心脏切面(SFCP)分类性能。我们研究了 7 种基于重建、对比损失、蒸馏和信息论的 SSL 方法,并在一个大型私有 US 数据集上对其进行了广泛评估。我们的观察和发现综合了不同设置下超过 500 次下游训练实验的结果。我们的主要观察表明,对于 SSL 训练而言,数据集的方差比其规模更为关键,因为它使模型能够学习可泛化的表示,从而提升下游任务的性能。总体而言,BarlowTwins 方法在作为下游任务初始化时,无论训练设置和数据变化如何,都表现出稳健的性能。值得注意的是,使用 1% 的标注数据进行完整微调,在 F1-score 上比 ImageNet 初始化高出 12%,并且比其他 SSL 初始化至少高出 4% 的 F1-score,因此使其成为从 US 视频迁移学习到图像数据的有前景的候选方法。

英文摘要:

Self-supervised learning (SSL) methods are popular since they can address situations with limited annotated data by directly utilising the underlying data distribution. However, the adoption of such methods is not explored enough in ultrasound (US) imaging, especially for fetal assessment. We investigate the potential of dual-encoder SSL in utilizing unlabelled US video data to improve the performance of challenging downstream Standard Fetal Cardiac Planes (SFCP) classification using limited labelled 2D US images. We study 7 SSL approaches based on reconstruction, contrastive loss, distillation, and information theory and evaluate them extensively on a large private US dataset. Our observations and findings are consolidated from more than 500 downstream training experiments under different settings. Our primary observation shows that for SSL training, the variance of the dataset is more crucial than its size because it allows the model to learn generalisable representations, which improve the performance of downstream tasks. Overall, the BarlowTwins method shows robust performance, irrespective of the training settings and data variations, when used as an initialisation for downstream tasks. Notably, full fine-tuning with 1% of labelled data outperforms ImageNet initialisation by 12% in F1-score and outperforms other SSL initialisations by at least 4% in F1-score, thus making it a promising candidate for transfer learning from US video to image data.

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

↑