arXiv:2509.03141cs.CVcs.LG2025-09被引 1

用双向时间正则化生成更真实的脑部MRI,预测疾病进展。

Temporally-Aware Diffusion Model for Brain Progression Modelling with Bidirectional Temporal Regularisation

  • 引入脑龄估计器引导扩散模型生成符合年龄差的影像。
  • 通过双向训练提升时间建模能力,生成更准确的未来扫描。
  • 基于3D全脑结构,适合研究神经退行性疾病进展。

生成真实的脑部MRI以准确预测脑结构未来的改变,对临床评估和疾病进展分析具有重要价值。现有方法存在三方面局限:(i) 未显式建模结构变化与时间间隔的关系,尤其在年龄不平衡数据上表现不佳;(ii) 依赖扫描插值,生成的是时间点之间的中间图像而非真正的病理进展;(iii) 多采用2D切片架构,忽略完整的3D解剖上下文,影响纵向预测精度。本文提出3D时序感知扩散模型(TADM-3D),可准确预测脑部MRI的进展。为更好建模时间与结构变化关系,TADM-3D使用预训练脑龄估计器(BAE)指导生成过程,确保基线与随访扫描间的年龄差合理。为进一步增强时序感知,提出回溯时间正则化(BITR),通过双向训练实现从基线到随访(前向)和从随访到基线(后向)的生成。尽管回溯生成临床应用有限,但有助于提升生成结果的时间一致性。在OASIS-3数据集上训练并验证,外部NACC数据集上评估泛化性能。代码将在接受后公开。

原文摘要 · Abstract (English)

Generating realistic MRIs to accurately predict future changes in the structure of brain is an invaluable tool for clinicians in assessing clinical outcomes and analysing the disease progression at the patient level. However, current existing methods present some limitations: (i) some approaches fail to explicitly capture the relationship between structural changes and time intervals, especially when trained on age-imbalanced datasets; (ii) others rely only on scan interpolation, which lack clinical utility, as they generate intermediate images between timepoints rather than future pathological progression; and (iii) most approaches rely on 2D slice-based architectures, thereby disregarding full 3D anatomical context, which is essential for accurate longitudinal predictions. We propose a 3D Temporally-Aware Diffusion Model (TADM-3D), which accurately predicts brain progression on MRI volumes. To better model the relationship between time interval and brain changes, TADM-3D uses a pre-trained Brain-Age Estimator (BAE) that guides the diffusion model in the generation of MRIs that accurately reflect the expected age difference between baseline and generated follow-up scans. Additionally, to further improve the temporal awareness of TADM-3D, we propose the Back-In-Time Regularisation (BITR), by training TADM-3D to predict bidirectionally from the baseline to follow-up (forward), as well as from the follow-up to baseline (backward). Although predicting past scans has limited clinical applications, this regularisation helps the model generate temporally more accurate scans. We train and evaluate TADM-3D on the OASIS-3 dataset, and we validate the generalisation performance on an external test set from the NACC dataset. The code will be available upon acceptance.

脑影像生成扩散模型时间建模疾病进展

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。