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

在低资源设置下使用高效微调策略分割胎头:基于U-Net的实证研究

Segmenting Fetal Head with Efficient Fine-tuning Strategies in Low-resource Settings: an empirical study with U-Net

Science Foundation Ireland Centre for Research Training in Machine Learning(爱尔兰科学基金会机器学习研究培训中心);School of Medicine, University College Dublin(都柏林大学医学院);School of Computer Science, University College Dublin(都柏林大学计算机科学学院)

Fangyijie Wang, Guénolé Silvestre, Kathleen M. Curran

AI总结:

本研究针对低资源下胎儿超声图像分割问题,基于U-Net系统评估多种微调策略,发现微调优于从头训练、解码器微调策略最优且少参数架构表现相当,并公开代码与权重。

摘要:

准确测量胎儿头围对于在常规产前筛查中评估胎儿生长至关重要。在测量之前,需要在超声图像中准确识别并分割感兴趣区域,特别是胎头。深度学习技术的最新进展表明,使用编码器-解码器模型分割胎头取得了显著进展。在这些模型中,U-Net已成为准确分割的标准方法。然而,训练编码器-解码器模型可能是一个耗时的过程,需要大量计算资源。此外,当可用数据有限时,微调这些模型尤其具有挑战性。目前仍然没有针对胎儿超声图像分割的U-Net最佳微调实践指南。本工作总结了现有的微调策略,包括不同的主干架构、模型组件和微调策略,使用了来自荷兰、西班牙、马拉维、埃及和阿尔及利亚的超声数据。我们的研究表明:(1)微调U-Net比从头训练获得更好的性能;(2)解码器中的微调策略优于其他策略;(3)参数数量较少的网络架构可以实现相似或更好的性能。我们还展示了微调策略在低资源设置中的有效性,并进一步将实验扩展到少样本学习。最后,我们公开发布了代码和特定的微调权重。

英文摘要:

Accurate measurement of fetal head circumference is crucial for estimating fetal growth during routine prenatal screening. Prior to measurement, it is necessary to accurately identify and segment the region of interest, specifically the fetal head, in ultrasound images. Recent advancements in deep learning techniques have shown significant progress in segmenting the fetal head using encoder-decoder models. Among these models, U-Net has become a standard approach for accurate segmentation. However, training an encoder-decoder model can be a time-consuming process that demands substantial computational resources. Moreover, fine-tuning these models is particularly challenging when there is a limited amount of data available. There are still no "best-practice" guidelines for optimal fine-tuning of U-net for fetal ultrasound image segmentation. This work summarizes existing fine-tuning strategies with various backbone architectures, model components, and fine-tuning strategies across ultrasound data from Netherlands, Spain, Malawi, Egypt and Algeria. Our study shows that (1) fine-tuning U-Net leads to better performance than training from scratch, (2) fine-tuning strategies in decoder are superior to other strategies, (3) network architecture with less number of parameters can achieve similar or better performance. We also demonstrate the effectiveness of fine-tuning strategies in low-resource settings and further expand our experiments into few-shot learning. Lastly, we publicly released our code and specific fine-tuned weights.

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

↑