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

Diffuse-UDA:通过外观与结构对齐的扩散模型解决医学图像分割中的无监督域自适应问题

Diffuse-UDA: Addressing Unsupervised Domain Adaptation in Medical Image Segmentation with Appearance and Structure Aligned Diffusion Models

Shenzhen Research Institute of Big Data(深圳大数据研究院);The Chinese University of Hong Kong (Shenzhen)(香港中文大学(深圳));Guangdong Provincial Key Laboratory of Big Data Computing(广东省大数据计算重点实验室)

Haifan Gong, Yitao Wang, Yihan Wang, Jiashun Xiao, Xiang Wan, Haofeng Li

AI总结:

针对医学图像分割中跨机构域差距导致的标注稀缺问题,提出融合可变形增强的扩散模型方法Diffuse-UDA,通过生成目标域图像-标签对增强无监督域自适应,性能超越主流方法且接近直接用目标域训练的理论上限。

摘要:

3D医学成像中体素级标注的稀缺性与复杂性带来了严峻挑战,尤其是资源充足机构的标注数据集与资源匮乏机构的无标注数据集之间存在域差距,这种差异影响了医疗领域人工智能算法的公平性。我们提出Diffuse-UDA,这是一种利用扩散模型解决医学图像分割中无监督域自适应(Unsupervised Domain Adaptation, UDA)问题的新方法。Diffuse-UDA能够生成具备目标域特征与多样结构的高质量图像-掩码对,从而增强UDA任务的效果。该方法首先生成目标域样本的伪标签;随后在来自两个域的图像-标签对或图像-伪标签对上训练经过专门定制、融合可变形增强的扩散模型;最后由源域标签引导扩散模型生成目标域的图像-标签对。在多个基准上的综合评估表明,Diffuse-UDA的性能优于主流的UDA与半监督策略,其性能接近甚至超过直接在目标域数据上训练的模型的理论上限。Diffuse-UDA为推动医学成像领域AI系统的开发与部署提供了可行路径,能够缓解不同医疗环境之间的资源差距,助力探索创新的AI驱动诊断工具,改善诊疗结果、节省时间并减少人为错误。

英文摘要:

The scarcity and complexity of voxel-level annotations in 3D medical imaging present significant challenges, particularly due to the domain gap between labeled datasets from well-resourced centers and unlabeled datasets from less-resourced centers. This disparity affects the fairness of artificial intelligence algorithms in healthcare. We introduce Diffuse-UDA, a novel method leveraging diffusion models to tackle Unsupervised Domain Adaptation (UDA) in medical image segmentation. Diffuse-UDA generates high-quality image-mask pairs with target domain characteristics and various structures, thereby enhancing UDA tasks. Initially, pseudo labels for target domain samples are generated. Subsequently, a specially tailored diffusion model, incorporating deformable augmentations, is trained on image-label or image-pseudo-label pairs from both domains. Finally, source domain labels guide the diffusion model to generate image-label pairs for the target domain. Comprehensive evaluations on several benchmarks demonstrate that Diffuse-UDA outperforms leading UDA and semi-supervised strategies, achieving performance close to or even surpassing the theoretical upper bound of models trained directly on target domain data. Diffuse-UDA offers a pathway to advance the development and deployment of AI systems in medical imaging, addressing disparities between healthcare environments. This approach enables the exploration of innovative AI-driven diagnostic tools, improves outcomes, saves time, and reduces human error.

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

↑