用扩散模型预测胶质瘤进展,支持任意时间点的肿瘤演化概率图
Multi-Task Diffusion Approach For Prediction of Glioma Tumor Progression
- 多任务扩散框架同时生成未来MRI和空间概率演化图
- 仅需两次随访扫描,即可生成任意时间点的进展预测
- 引入放疗剂量加权损失,提升临床关键区域预测精度
胶质瘤是一种进展迅速、预后差的脑部恶性肿瘤,其演变预测面临巨大挑战。临床实践中纵向MRI数据稀疏且采集不规律,缺失随访序列导致数据不平衡,难以建立可靠模型。本文提出一种多任务扩散框架,实现时间无关、像素级的胶质瘤进展预测。模型可同时生成任意时间点的未来FLAIR序列,并基于符号距离场(SDF)估计空间概率演化图,实现不确定性量化。为捕捉跨任意时间间隔的肿瘤动态,集成预训练形变模块,利用形变场建模扫描间变化。针对数据稀缺问题,设计针对性增强管道,合成三阶段随访序列并补全缺失的影像模态,提升模型稳定性和准确性。仅基于早期两个随访扫描,即可生成灵活的时间依赖概率图,使临床医生可在任意未来时间点评估肿瘤进展风险。此外,引入放疗加权焦点损失,利用放疗剂量图在训练中强调临床重要区域。该方法在公开数据集上训练,在内部私有数据集上验证,结果表现优异。
原文摘要 · Abstract (English)
Glioma, an aggressive brain malignancy characterized by rapid progression and its poor prognosis, poses significant challenges for accurate evolution prediction. These challenges are exacerbated by sparse, irregularly acquired longitudinal MRI data in clinical practice, where incomplete follow-up sequences create data imbalances and make reliable modeling difficult. In this paper, we present a multitask diffusion framework for time-agnostic, pixel-wise prediction of glioma progression. The model simultaneously generates future FLAIR sequences at any chosen time point and estimates spatial probabilistic tumor evolution maps derived using signed distance fields (SDFs), allowing uncertainty quantification. To capture temporal dynamics of tumor evolution across arbitrary intervals, we integrate a pretrained deformation module that models inter-scan changes using deformation fields. Regarding the common clinical limitation of data scarcity, we implement a targeted augmentation pipeline that synthesizes complete sequences of three follow-up scans and imputes missing MRI modalities from available patient studies, improving the stability and accuracy of predictive models. Based on merely two follow-up scans at earlier timepoints, our framework produces flexible time-depending probability maps, enabling clinicians to interrogate tumor progression risks at any future temporal milestone. We further introduce a radiotherapy-weighted focal loss term that leverages radiation dose maps, as these highlight regions of greater clinical importance during model training. The proposed method was trained on a public dataset and evaluated on an internal private dataset, achieving promising results in both cases
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。