用解剖结构引导扩散模型,更准预测阿尔茨海默病脑部变化。
Anatomically Guided Latent Diffusion for Brain MRI Progression Modeling
- 输入层融合解剖、噪声随访和临床数据,端到端建模进展过程。
- 生成图像体积误差降低15%-20%,解剖一致性更强。
- 适合研究神经退行性疾病进展,尤其关注脑结构变化的学者。
准确建模纵向脑部MRI变化对理解神经退行性疾病和预测个体化结构改变至关重要。现有先进方法如BrLP虽性能优越,但依赖多阶段训练与辅助条件模块,存在架构复杂、临床协变量利用不足及解剖一致性难以保障等问题。本文提出解剖引导的潜在扩散模型(AG-LDM),通过直接在输入层融合基线解剖、噪声随访状态与临床协变量,避免使用辅助控制网络,实现统一端到端建模,同时显式引入轻量级3D组织分割模型WarpSeg进行解剖监督,在自编码器微调与扩散模型训练中确保脑组织边界一致与形态保真度。在31,713例ADNI纵向数据对上实验,以及在OASIS-3上的零样本评估表明,AG-LDM性能达到或超越更复杂的扩散模型,生成图像质量优异,体积误差降低15%-20%;其对时间与临床协变量的敏感性比BrLP高3.5至31.5倍,能生成符合生物学规律的反事实轨迹,准确捕捉阿尔茨海默病特征如边缘系统萎缩与脑室扩张。结果表明AG-LDM是一种高效且解剖基础坚实的脑部MRI进展建模框架。
原文摘要 · Abstract (English)
Accurately modeling longitudinal brain MRI progression is crucial for understanding neurodegenerative diseases and predicting individualized structural changes. Existing state-of-the-art approaches, such as Brain Latent Progression (BrLP), often use multi-stage training pipelines with auxiliary conditioning modules but suffer from architectural complexity, suboptimal use of conditional clinical covariates, and limited guarantees of anatomical consistency. We propose Anatomically Guided Latent Diffusion Model (AG-LDM), a segmentation-guided framework that enforces anatomically consistent progression while substantially simplifying the training pipeline. AG-LDM conditions latent diffusion by directly fusing baseline anatomy, noisy follow-up states, and clinical covariates at the input level, a strategy that avoids auxiliary control networks by learning a unified, end-to-end model that represents both anatomy and progression. A lightweight 3D tissue segmentation model (WarpSeg) provides explicit anatomical supervision during both autoencoder fine-tuning and diffusion model training, ensuring consistent brain tissue boundaries and morphometric fidelity. Experiments on 31,713 ADNI longitudinal pairs and zero-shot evaluation on OASIS-3 demonstrate that AG-LDM matches or surpasses more complex diffusion models, achieving highly competitive image quality and 15-20% reduction in volumetric errors in generated images. AG-LDM also exhibits markedly stronger utilization of temporal and clinical covariates (3.5-31.5x higher covariate sensitivity than BrLP) and generates biologically plausible counterfactual trajectories, accurately capturing hallmarks of Alzheimer's progression such as limbic atrophy and ventricular expansion. These results highlight AG-LDM as an efficient, anatomically grounded framework for reliable brain MRI progression modeling.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。