构建可动态模拟胶质母细胞瘤与治疗相互作用的生成模型
Brain-WM: Brain Glioblastoma World Model
- 将治疗决策与肿瘤演化联合建模,通过共享潜空间实现协同预测
- 治疗规划准确率达91.5%,多序列MRI生成结构相似性超0.84
- 适合临床研究者用于个性化治疗方案优化与疗效预判
精确预测不同治疗干预下胶质母细胞瘤(GBM)的预后对优化临床结果至关重要。尽管生成式AI在模拟GBM演化方面已显潜力,但现有方法通常将干预措施视为静态条件输入,而非动态决策变量,因而无法捕捉肿瘤演化与治疗反应间的复杂互馈关系。为此,我们提出Brain-WM,首个统一下一步治疗预测与未来MRI生成的脑GBM世界模型,以捕获肿瘤与治疗的共演化动态。具体而言,Brain-WM将时空动态编码至共享潜空间,实现联合自回归治疗预测与基于流的未来MRI生成。不同于传统单体框架,Brain-WM采用新型Y型混合变压器(Y-shaped MoT)架构,结构性解耦异质目标,有效利用跨任务协同并防止特征坍塌。最后,一种协同的多时间点掩码对齐目标,显式锚定潜表示于解剖学相关的肿瘤结构与进展感知语义。在内部及外部多机构队列上的广泛验证表明,Brain-WM表现优异:治疗规划准确率达91.5%,对FLAIR、T1CE和T2W序列的结构相似性(SSIM)分别为0.8524、0.8581和0.8404。最终,Brain-WM为优化患者医疗提供了稳健的临床沙盒。源代码已公开于https://github.com/thibault-wch/Brain-GBM-world-model。
原文摘要 · Abstract (English)
Precise prognostic modeling of glioblastoma (GBM) under varying treatment interventions is essential for optimizing clinical outcomes. While generative AI has shown promise in simulating GBM evolution, existing methods typically treat interventions as static conditional inputs rather than dynamic decision variables. Consequently, they fail to capture the complex, reciprocal interplay between tumor evolution and treatment response. To bridge this gap, we present Brain-WM, a pioneering brain GBM world model that unifies next-step treatment prediction and future MRI generation, thereby capturing the co-evolutionary dynamics between tumor and treatment. Specifically, Brain-WM encodes spatiotemporal dynamics into a shared latent space for joint autoregressive treatment prediction and flow-based future MRI generation. Then, instead of a conventional monolithic framework, Brain-WM adopts a novel Y-shaped Mixture-of-Transformers (MoT) architecture. This design structurally disentangles heterogeneous objectives, successfully leveraging cross-task synergies while preventing feature collapse. Finally, a synergistic multi-timepoint mask alignment objective explicitly anchors latent representations to anatomically grounded tumor structures and progression-aware semantics. Extensive validation on internal and external multi-institutional cohorts demonstrates the superiority of Brain-WM, achieving 91.5% accuracy in treatment planning and SSIMs of 0.8524, 0.8581, and 0.8404 for FLAIR, T1CE, and T2W sequences, respectively. Ultimately, Brain-WM offers a robust clinical sandbox for optimizing patient healthcare. The source code is made available at https://github.com/thibault-wch/Brain-GBM-world-model.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。