用扩散模型生成更符合脑部解剖结构的MRI,提升肿瘤诊断准确性。
BrainMRDiff: A Diffusion Model for Anatomically Consistent Brain MRI Synthesis
- 通过解剖结构与肿瘤信息联合建模,引导图像生成
- 在BraTS数据集上性能提升最高达33.33%
- 适合医学影像生成、辅助诊断研究者使用
精准的脑肿瘤诊断依赖于多序列磁共振成像(MRI)评估。然而临床中某些序列因运动伪影或对比剂禁忌等原因难以获取,导致图像质量下降,影响放射科医生判读。因此,合成高质量MRI成为关键研究方向。尽管可控生成AI取得进展,但保持解剖准确性仍是挑战。本文提出BrainMRDiff,一种拓扑一致性、解剖引导的扩散模型,利用脑部及肿瘤解剖结构作为条件输入。引入Tumor+Structure Aggregation(TSA)模块整合多类解剖结构与肿瘤信息,构建生成过程的完整条件;设计Topology-Guided Anatomy Preservation(TGAP)模块,在反向去噪过程中强制保持拓扑一致性,确保生成图像具备解剖完整性。实验表明,BrainMRDiff在BraTS-AG数据集上优于基线23.33%,在BraTS-Met数据集上提升33.33%。代码将很快公开。
原文摘要 · Abstract (English)
Accurate brain tumor diagnosis relies on the assessment of multiple Magnetic Resonance Imaging (MRI) sequences. However, in clinical practice, the acquisition of certain sequences may be affected by factors like motion artifacts or contrast agent contraindications, leading to suboptimal outcome, such as poor image quality. This can then affect image interpretation by radiologists. Synthesizing high quality MRI sequences has thus become a critical research focus. Though recent advancements in controllable generative AI have facilitated the synthesis of diagnostic quality MRI, ensuring anatomical accuracy remains a significant challenge. Preserving critical structural relationships between different anatomical regions is essential, as even minor structural or topological inconsistencies can compromise diagnostic validity. In this work, we propose BrainMRDiff, a novel topology-preserving, anatomy-guided diffusion model for synthesizing brain MRI, leveraging brain and tumor anatomies as conditioning inputs. To achieve this, we introduce two key modules: Tumor+Structure Aggregation (TSA) and Topology-Guided Anatomy Preservation (TGAP). TSA integrates diverse anatomical structures with tumor information, forming a comprehensive conditioning mechanism for the diffusion process. TGAP enforces topological consistency during reverse denoising diffusion process; both these modules ensure that the generated image respects anatomical integrity. Experimental results demonstrate that BrainMRDiff surpasses existing baselines, achieving performance improvements of 23.33% on the BraTS-AG dataset and 33.33% on the BraTS-Met dataset. Code will be made publicly available soon.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。