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

StructDiff: 一种结构感知的扩散模型用于3D细粒度医学图像合成

StructDiff: Structure-aware Diffusion Model for 3D Fine-grained Medical Image Synthesis

Key Laboratory of New Generation Artificial Intelligence Technology and Its Interdisciplinary Applications (Southeast University), Ministry of Education, China(新一代人工智能技术及跨学科应用关键实验室(东南大学),中华人民共和国教育部,中国);Jiangsu Province Joint International Research Laboratory of Medical Information Processing, Southeast University, Nanjing, China(江苏省医疗信息处理联合国际研究实验室(东南大学),南京市,中国);Univ Rennes, CHU Rennes, Inserm, LTSI– UMR 1099, F-35000 Rennes, France(里昂大学,里昂大学医院,Inserm,LTSI– UMR 1099,法国里昂,法国);Shanghai AI Laboratory(上海人工智能实验室);School of Computer Science and Technology, Beijing Institute of Technology, Beijing, China(计算机科学与技术学院,北京理工大学,北京,中国);Department of Radiology, Jinling Hospital, Affiliated Hospital of Medical School, Nanjing University, Nanjing, China(放射科,南京大学附属医院,南京大学,南京,中国)

Jiahao Xia, Yutao Hu, Yaolei Qi, Zhenliang Li, Wenqi Shao, Junjun He, Ying Fu, Longjiang Zhang, Guanyu Yang

AI总结:

StructDiff通过结构感知扩散模型解决3D医学图像细粒度生成中的拓扑一致性和细节还原问题,提升下游分割性能。

摘要:

通过语义图像生成解决医学影像数据稀缺问题近年来受到越来越多关注。然而,现有的生成模型主要集中在合成整个器官或大组织结构,其在再现细粒度解剖细节方面的能力有限。由于医学数据对拓扑一致性有严格要求,且具有复杂的3D形态异质性,准确重建细粒度解剖细节仍是一个重大挑战。为了解决这些限制,我们提出了StructDiff,一种用于细粒度3D医学图像合成的结构感知扩散模型,能够精确生成拓扑复杂的解剖结构。除了传统的基于掩码的引导外,StructDiff还引入了一种配对图像-掩码模板来引导生成过程,提供结构约束,并提供明确的掩码到图像对应关系。此外,设计了一个掩码生成模块(MGM),以丰富掩码多样性并缓解高质量参考掩码的稀缺性。此外,我们提出了一种基于跳采样方差(SSV)的置信度感知自适应学习(CAL)策略,以减轻不完美合成数据在转移到下游任务时引入的不确定性。大量实验表明,StructDiff在拓扑一致性和视觉真实感方面达到了最先进的性能,并显著提升了下游分割性能。代码将在接受后发布。

英文摘要:

Solving medical imaging data scarcity through semantic image generation has attracted growing attention in recent years. However, existing generative models mainly focus on synthesizing whole-organ or large-tissue structures, showing limited capability in reproducing fine-grained anatomical details. Due to the stringent requirement of topological consistency and the complex 3D morphological heterogeneity of medical data, accurately reconstructing fine-grained anatomical details remains a significant challenge. To address these limitations, we propose StructDiff, a Structure-aware Diffusion Model for fine-grained 3D medical image synthesis, which enables precise generation of topologically complex anatomies. In addition to the conventional mask-based guidance, StructDiff further introduces a paired image-mask template to guide the generation process, providing structural constrains and offering explicit knowledge of mask-to-image correspondence. Moreover, a Mask Generation Module (MGM) is designed to enrich mask diversity and alleviate the scarcity of high-quality reference masks. Furthermore, we propose a Confidence-aware Adaptive Learning (CAL) strategy based on Skip-Sampling Variance (SSV), which mitigates uncertainty introduced by imperfect synthetic data when transferring to downstream tasks. Extensive experiments demonstrate that StructDiff achieves state-of-the-art performance in terms of topological consistency and visual realism, and significantly boosts downstream segmentation performance. Code will be released upon acceptance.

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

↑