mlsanityData to 7 Oct 2026

Understanding Trajectory Heterogeneity in Federated World Model Learning

arXiv 2610.02957 ↗First posted 2 Oct 2026cs.LG cs.CLcited by 0

On the map: World models → Medical and surgical world models

Abstract (arXiv)

World models learn state evolution from trajectories, making access to temporal context a central training requirement. Federated learning can use distributed records, while ownership boundaries within a trajectory restrict the examples each client can construct. Our study benchmarks this cross-time setting through hourly action-conditioned clinical prediction on eight MIMIC-IV disease cohorts, comprising 40.87 million transition memberships. We specify severity-based client ownership, patient-separated construction, local history and future-window rules, and paired rollout evaluation from one to 32 hours. A matrix of ten federated algorithms covers 32 disease--partition configurations under five rounds of ten-percent participation. Three findings emerge from existing results and training logs. First, client ownership and participation jointly restrict long-window coverage: only 7.55%--21.36% of pooled-available 32-step windows have a locally complete anchor visited during training, averaged across diseases. Second, finer severity partitions accompany higher FedAvg error in 15 of 16 paired comparisons, while algorithm gains are small and horizon-dependent: FedProx reduces mean error by 0.56%, with no consistent improvement at 32 steps. Third, algorithm labels conceal distinct update behavior, including inactive extrapolation and orders-of-magnitude differences in update scale. Cached-update performance also varies strongly across trajectory partitions under the same benchmark protocol. These results establish temporal access, participation coverage, optimization behavior, and horizon-resolved prediction as complementary dimensions for evaluating federated clinical world models.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Medical and surgical world models, one of 11 directions in World models. First posted on arXiv 2 Oct 2026.

Builds on

No reference list yet.

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026From Static Risk to Dynamic Trajectories: Toward World-Model-Inspired Clinical Prediction
  2. 2026ChronoMedicalWorld: A Medical World Model for Learning Patient Trajectories from Longitudinal Care Data
  3. 2026Medical world models in healthcare: foundations, applications, and challenges for trustworthy clinical translation
  4. 2022Federated Offline Reinforcement Learning
  5. 2026LifeLong Digital Twin: A Unified Modeling Paradigm and Agent Harness for Event-Driven Lifelong Health State Trajectories
  6. 2026Medical world models: representing medical states, modelling clinical dynamics and guiding intervention policies
  7. 2026Learning Continuous Patient Trajectories from Electronic Health Records
  8. 2025FedAgentBench: Towards Automating Real-world Federated Medical Image Analysis with Server-Client LLM Agents