Driving · 2026-10-21
CaMeWM: A Causal Medical World Model for Generative Tumor Evolution and Treatment Planning
Huanlong Gao, Xuelei He, Fengjun Zhao, Xiaowei He
AI-assisted summary based on the paper. Consult the linked source for methods and limitations. Benchmark numbers appear after editor verification.
TL;DR
CaMeWM is a causal medical world model for support-restricted post-TACE tumor evolution and treatment planning. It forms one intervention-to-decision loop: admit a queried regimen only when fold-local empirical propensity support holds, generate a distribution of lesion-local follow-up worlds via response-grounded anchor plus residual 3D diffusion, then map sampled worlds to RMST using a response-Cox utility model and evaluate the frozen policy with cross-fitted censoring-adjusted doubly robust estimation.
Why it matters
The authors argue that treatment planning is counterfactual and that conditional generation alone can reproduce historical allocation rather than simulate interventional disease dynamics needed to rank regimens. CaMeWM couples an intervention query, a support/positivity restriction, and censoring-aware doubly robust evaluation, aiming to provide an inspectable basis for retrospective, support-restricted treatment planning.
Method
- Supported intervention interface: fix do(A=a) only when a complete regimen has fold-local joint-propensity support, using e_a(h)=P(A=a|H=h) and threshold η.
- Action-conditioned world simulation: generate follow-up worlds using an auxiliary response descriptor, a deterministic anchor preserving stable anatomy, and residual 3D diffusion transition (lesion-local residual variation).
- Utility-aware rollout and evaluation: map factual or sampled states to restricted mean survival time (RMST) with a response-Cox model; rank supported actions via generated-world RMST and evaluate with cross-fitted censoring-adjusted doubly robust estimation.
Limitation
The authors state that the learned networks cannot verify causal assumptions (e.g., consistency, conditional exchangeability, positivity); retrospective evaluation does not establish prospective clinical benefit, and they do not claim a unique alternate-world CT for an individual patient.
Abstract (from arXiv)
Treatment planning is intrinsically counterfactual: clinicians must compare how a patient's tumor and survival would evolve under competing regimens, whereas a retrospective cohort records only the regimen received and one follow-up scan. A treatment-conditioned generator trained on such data can therefore reproduce historical treatment allocation rather than simulate the interventional disease dynamics needed to rank regimens. We introduce \textbf{CaMeWM}, a causal medical world model for supported post-TACE tumor evolution and treatment planning. CaMeWM implements one intervention-to-decision loop rather than three independent prediction tasks: it first admits a queried regimen only for clinically comparable histories with empirical treatment support; it then generates a distribution of lesion-local follow-up worlds through an action-conditioned response descriptor, deterministic anchor, and residual 3D diffusion transition; and it finally maps sampled worlds to restricted mean survival time (RMST) with a response-Cox utility model. Under pre-treatment adjustment and positivity, this loop estimates a prespecified, support-restricted g-formula target; all selection and nuisance estimation are fold-local, and the induced policy is evaluated by cross-fitted, censoring-adjusted doubly robust estimation. Across 79 paired out-of-fold episodes, CaMeWM achieves 0.618 SSIM and 0.528 change-map SSIM for follow-up CT transition, while its generated-world survival pathway attains a 0.615 C-index; factual-only AIPW estimates a 9.60-week RMST gain (95\% CI 0.61--18.13) for the frozen policy. CaMeWM provides an inspectable basis for retrospective, support-restricted treatment planning.
Related papers
- CoRe-WM: Conditioned Response-State World Modeling for Longitudinal Treatment-Response Prediction in Breast DCE-MRI
- PatientWM: Towards Generalizable Clinical World Models via Unified Patient States
- Same Predictions, Different Harms: Causal Auditing of Patient World Models
- Patient, Place, Prior (P$^3$): What Counts as Personalization in Medical World Models?
- Repeated-Measure Leakage, Distribution Shift, and Reliability under Partial Observation in Patient World Models
Publication record
Bibliographic evidence describes where and when a record appeared. It does not validate scientific claims.
First public release: 2026-10-17 · Indexed by Pub-AI: 2026-10-21 UTC
CaMeWM: A Causal Medical World Model for Generative Tumor Evolution and Treatment Planning
Preprint posted: 2026-10-17 · Original record ↗
- arxiv
- 2610.26923
Explicit source versions (1)
- v1 · 2026-10-17
- Preprint available
Source date: 2026-10-17 · First detected: 2026-10-25 UTC · Historical baseline
Source evidence (3)
- Official evidence ↗ · OAI arXiv created; not header datestamp (official metadata)
- Official evidence ↗ · atom:published (official metadata)
- Official evidence ↗ · atom:published (official metadata)
Abstract · arxiv
Treatment planning is intrinsically counterfactual: clinicians must compare how a patient's tumor and survival would evolve under competing regimens, whereas a retrospective cohort records only the regimen received and one follow-up scan. A treatment-conditioned generator trained on such data can therefore reproduce historical treatment allocation rather than simulate the interventional disease dynamics needed to rank regimens. We introduce \textbf{CaMeWM}, a causal medical world model for supported post-TACE tumor evolution and treatment planning. CaMeWM implements one intervention-to-decision loop rather than three independent prediction tasks: it first admits a queried regimen only for clinically comparable histories with empirical treatment support; it then generates a distribution of lesion-local follow-up worlds through an action-conditioned response descriptor, deterministic anchor, and residual 3D diffusion transition; and it finally maps sampled worlds to restricted mean survival time (RMST) with a response-Cox utility model. Under pre-treatment adjustment and positivity, this loop estimates a prespecified, support-restricted g-formula target; all selection and nuisance estimation are fold-local, and the induced policy is evaluated by cross-fitted, censoring-adjusted doubly robust estimation. Across 79 paired out-of-fold episodes, CaMeWM achieves 0.618 SSIM and 0.528 change-map SSIM for follow-up CT transition, while its generated-world survival pathway attains a 0.615 C-index; factual-only AIPW estimates a 9.60-week RMST gain (95\% CI 0.61--18.13) for the frozen policy. CaMeWM provides an inspectable basis for retrospective, support-restricted treatment planning.