arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2601.08078cs.CVcs.CEcs.CL

利用基于DINOv3的自监督特征进行鲁棒的小样本医学图像分割

Exploiting DINOv3-Based Self-Supervised Features for Robust Few-Shot Medical Image Segmentation

University of Texas Southwestern Medical Center(德克萨斯大学西南医学中心);University of Pennsylvania(宾夕法尼亚大学)

Guoping Xu, Jayaram K. Udupa, Weiguo Lu, You Zhang

AI总结:

本文提出DINO-AugSeg框架,利用DINOv3特征结合小波域增强和上下文融合,提升小样本医学图像分割的鲁棒性。

摘要:

基于深度学习的自动医学图像分割在临床诊断和治疗计划中发挥着关键作用,但在小样本场景中仍然具有挑战性,因为标注训练数据稀缺。最近,自监督基础模型如DINOv3,这些模型是在大规模自然图像数据集上训练的,已显示出在密集特征提取方面的强大潜力,可以帮助解决小样本学习的挑战。然而,它们直接应用于医学图像受到领域差异的阻碍。在本工作中,我们提出了DINO-AugSeg,一种新的框架,利用DINOv3特征来解决小样本医学图像分割的挑战。具体来说,我们引入了WT-Aug,一种基于小波的特征级增强模块,通过扰动频率成分来丰富DINOv3提取特征的多样性,并引入了CG-Fuse,一种基于上下文信息的融合模块,利用交叉注意力将语义丰富的低分辨率特征与空间细节丰富的高分辨率特征结合起来。在六个公共基准上进行了广泛的实验,涵盖了五种成像模态,包括MRI、CT、超声、内窥镜和皮肤镜,结果表明DINO-AugSeg在有限样本条件下始终优于现有方法。这些结果突显了将小波域增强和上下文融合纳入其中的有效性,表明DINO-AugSeg是推进小样本医学图像分割的有前景的方向。代码和数据将在https://github.com/apple1986/DINO-AugSeg上提供。

英文摘要:

Deep learning-based automatic medical image segmentation plays a critical role in clinical diagnosis and treatment planning but remains challenging in few-shot scenarios due to the scarcity of annotated training data. Recently, self-supervised foundation models such as DINOv3, which were trained on large natural image datasets, have shown strong potential for dense feature extraction that can help with the few-shot learning challenge. Yet, their direct application to medical images is hindered by domain differences. In this work, we propose DINO-AugSeg, a novel framework that leverages DINOv3 features to address the few-shot medical image segmentation challenge. Specifically, we introduce WT-Aug, a wavelet-based feature-level augmentation module that enriches the diversity of DINOv3-extracted features by perturbing frequency components, and CG-Fuse, a contextual information-guided fusion module that exploits cross-attention to integrate semantic-rich low-resolution features with spatially detailed high-resolution features. Extensive experiments on six public benchmarks spanning five imaging modalities, including MRI, CT, ultrasound, endoscopy, and dermoscopy, demonstrate that DINO-AugSeg consistently outperforms existing methods under limited-sample conditions. The results highlight the effectiveness of incorporating wavelet-domain augmentation and contextual fusion for robust feature representation, suggesting DINO-AugSeg as a promising direction for advancing few-shot medical image segmentation. Code and data will be made available on https://github.com/apple1986/DINO-AugSeg.

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

↑