Search papers, labs, and topics across Lattice.
This paper introduces an intervention-aware clinical world model designed to enhance post-operative outcome forecasting in cardiology, specifically for atrial fibrillation ablation. By structuring patient data into a 3D spatial latent state that evolves with time-ordered clinical events, the model captures the irregular trajectory of recovery and updates risk assessments dynamically. The approach demonstrates significant predictive performance with an AUROC of 0.756 and AUPRC of 0.777 for recurrence prediction, highlighting its potential for improving clinical decision-making during the recovery period.
Capturing the dynamic nature of post-operative recovery, this model predicts long-term recurrence risk with impressive accuracy, outperforming traditional static approaches.
Many clinical prediction models treat post-intervention outcomes as a one-step mapping from baseline measurements to a future endpoint. However, recovery after a procedure often unfolds as an irregular trajectory: clinical observations, medication changes, repeat interventions, and physiological measurements are recorded asynchronously and can change risk assessment over time. We propose an intervention-aware clinical world model that represents each patient with a structured latent state and evolves it through time-ordered post-intervention events. The model first encodes baseline imaging into a 3D spatial latent state. It then updates this state using procedural context, static covariates, elapsed time, and peri-event physiological embeddings. Follow-up imaging provides training-only supervision through a latent forecasting objective. We apply the framework to atrial fibrillation ablation. During the 90-day recovery window, irregular post-procedure records provide clinically meaningful evidence for long-term recurrence risk. In repeated internal cross-validation on DECAAF-II, our model achieves AUROC 0.756 and AUPRC 0.777 for recurrence prediction. It also achieves a scar-extent MAE of 2.971 percentage points without requiring follow-up MRI intensities at inference. The learned state supports recurrence-risk queries at different horizons and retrospective input editing of blanking-period records.
