arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.09189eess.IVcs.CV

有限数据下的医学图像分割

Segmenting Medical Images with Limited Data

National University of Singapore(新加坡国立大学);Sun Yat-sen University(中山大学);Tan Tock Seng Hospital(陈笃生医院)

Zhaoshan Liua, Qiujie Lv, Chau Hung Lee, Lei Shen

AI总结:

针对医学图像分割中数据有限、未标注图像难利用的问题,提出融合在线自动增强、残差鲁棒性增强模块与敏感损失的半监督分割方法DEMS,在小数据场景下性能显著优于现有方法。

摘要:

尽管计算机视觉已被证实对医学图像分割具有重要价值,但其应用仍面临数据集规模有限、有效利用未标注图像复杂度高等挑战。为应对这些挑战,我们提出了一种全新的基于一致性的半监督方法,名为数据高效医学分割器(data-efficient medical segmenter,DEMS)。DEMS采用编码器-解码器架构,并整合了自研的在线自动增强器(online automatic augmenter,OAA)和残差鲁棒性增强(residual robustness enhancement,RRE)模块。OAA通过多种图像变换对输入数据进行增强,从而丰富数据集多样性以提升泛化能力。RRE丰富了特征多样性,并引入扰动为不同解码器生成多样化输入,进而增强可变性。此外,我们还引入了一种敏感损失函数,以进一步增强不同解码器之间的一致性并稳定训练过程。在自有数据集和三个公开数据集上开展的大量实验结果证实了DEMS的有效性。在极端数据短缺场景下,我们的DEMS在戴斯系数(dice score)上相比U-Net和性能最优的现有先进方法分别提升了16.85%和10.37%。凭借其优异的数据效率,DEMS有望为小数据场景下的医学分割领域带来重大进展。该项目主页可通过https://github.com/NUS-Tim/DEMS访问。

英文摘要:

While computer vision has proven valuable for medical image segmentation, its application faces challenges such as limited dataset sizes and the complexity of effectively leveraging unlabeled images. To address these challenges, we present a novel semi-supervised, consistency-based approach termed the data-efficient medical segmenter (DEMS). The DEMS features an encoder-decoder architecture and incorporates the developed online automatic augmenter (OAA) and residual robustness enhancement (RRE) blocks. The OAA augments input data with various image transformations, thereby diversifying the dataset to improve the generalization ability. The RRE enriches feature diversity and introduces perturbations to create varied inputs for different decoders, thereby providing enhanced variability. Moreover, we introduce a sensitive loss to further enhance consistency across different decoders and stabilize the training process. Extensive experimental results on both our own and three public datasets affirm the effectiveness of DEMS. Under extreme data shortage scenarios, our DEMS achieves 16.85\% and 10.37\% improvement in dice score compared with the U-Net and top-performed state-of-the-art method, respectively. Given its superior data efficiency, DEMS could present significant advancements in medical segmentation under small data regimes. The project homepage can be accessed at https://github.com/NUS-Tim/DEMS.

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

↑