Pub-AI.AI in Science

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

arXiv:2610.26923PDF

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

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

preprint

CaMeWM: A Causal Medical World Model for Generative Tumor Evolution and Treatment Planning

Huanlong Gao, Xuelei He, Fengjun Zhao, Xiaowei He

Preprint posted: 2026-10-17 · Original record ↗

arxiv
2610.26923
Explicit source versions (1)
  • v1 · 2026-10-17
  1. Preprint available

    Source date: 2026-10-17 · First detected: 2026-10-25 UTC · Historical baseline

    Source evidence (3)
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.