Natural Disease Progression
Anticipate serious new disease events from patient histories.
Task & evaluation
- Objective
- Anticipate serious new disease after hospital discharge.
- Input
- MIMIC-IV patient history available before the prediction cutoff.
- Output
- Thirteen cumulative risk probabilities per disease and for death, from 30 to 365 days.
- Data
- Controlled-access MIMIC-IV v3.1; raw records and test labels are not redistributed.
- Evaluation
- Case detection at a fixed 5% alert budget across the endpoint panel.
- Research challenge
- Learn from irregular histories and imbalanced outcomes without using future information.
Understanding the task
Explore how a medical history becomes a prediction, and why a limited follow-up budget changes the meaning of a good model.
Who might need closer follow-up?
After a hospital discharge, only a limited number of people may receive intensive follow-up. Can their existing medical histories help identify those most likely to have a serious condition recorded during the next year?
This task predicts risks for conditions such as heart failure, stroke and kidney disease at thirteen horizons from 30 days to one year. Its central question is practical: if follow-up reaches the highest-risk 5% of eligible patients, what fraction of future cases does that group contain?
From discharge history to five follow-up places
Explore two synthetic examples. First decide who can be scored; then see why the best overall ranking can miss every case in the alert budget.
1 · Freeze the history, check eligibility
Illustrative patient for one disease endpoint.
A new day-180 diagnosis with a complete year and valid label is eligible.
An event must both occur and become available by discharge to enter the model. A late report stays outside the history even when its specimen was collected earlier.
2 · Spend the fixed alert budget
Synthetic endpoint: 100 eligible patients, 20 cases.
The budget is fixed at the benchmark’s 5%. Change the ranking to compare case detection with AUROC.
3 · How history becomes a risk trajectory
The model combines what happened with when it happened. Both views stop at the discharge cutoff.
Diagnoses, measurements, medicines and missing information
Summarize the content of the historyRead event content together with the time between events
Update a memory as the history unfoldsTwo input features pass through a learned demographic projection.
Add demographic contextLearn a separate trajectory for each outcome
Accumulate risk across timeThirteen nondecreasing risks for each disease and death. The alert-budget scorer uses day 365.
The model retains a limited history. See the model implementation. Whether each branch improves prediction requires a controlled comparison.
A medical history has a cutoff
The model reads diagnoses, laboratory results, prescriptions, medication administrations and demographics available by discharge. A specimen collected earlier but reported later cannot be used: doing so would give the model information unavailable at the real decision time.
About 68,000 patients support training, with separate validation and hidden test patients. Earlier records use an older disease-coding system, creating imperfect labels when codes cannot be mapped cleanly. A missing later diagnosis can also mean that care happened elsewhere. The target is a recorded inpatient diagnosis, not the biological moment a disease began.
Remember the pattern and the order
The baseline combines two views of a history. One summarizes which events occurred; a recurrent neural network reads events in order and updates a compact memory. Age and sex provide additional context. Together these views can distinguish, for example, a persistently abnormal measurement from a recent change.
The model produces a risk trajectory for each endpoint. Risk of an event occurring by a later date cannot be lower than risk by an earlier date, so the prediction mechanism accumulates nonnegative increments over time.
Learn from what is actually observed
Training gives extra weight to rare disease events and excludes outcomes whose observation is incomplete. This avoids teaching the model that every missing follow-up record means good health.
The baseline was selected using overall ranking quality, whereas the task rewards detection within a small follow-up group. A model can rank cases reasonably well overall yet place too few in its highest-risk 5%. Aligning those goals is a concrete research opportunity.
Five places can tell a different story
Imagine 100 eligible patients, including 20 who later have the condition recorded. If the five highest-ranked patients include four of those cases, recall is 4/20, or 20%. Random selection would find about 5% on average. The real score averages this calculation across prespecified disease endpoints.
Patients with an existing endpoint or direct precursor are excluded from that endpoint’s assessment. Ranking and probability accuracy are checked separately: finding high-risk patients does not establish that a prediction of “20% risk” is numerically trustworthy.
Useful ranking, with room to improve
The temporal neural model finds about 24% of future cases within the 5% alert budget, compared with about 5% for a constant-risk control. A tree-based research reference reaches about 27%; it is a useful comparison but not an eligible final model for this neural-model task.
The neural model’s probabilities remain imperfectly calibrated. These are single-run results, and validation and test averages cover different eligible endpoints. They support further study of prioritization, not evidence that using the system would improve patient outcomes.
| System | Validation | Hidden test |
|---|---|---|
| Constant-risk control | 5.0% | 5.0% |
| Risk-group control | 14.6% | 18.5% |
| Sequence model | 21.1% | 23.6% |
| Tree-based reference | 23.5% | 26.8% |
Compute and resources
Preparation and fitting
Reported measurement
- Time
- ≈8 min
- Hardware
- 1 × H100 · 10 CPUs
- GPU allocation
- Not separately reported
A recorded baseline run from cached histories through calibrated predictions.
Elapsed time by stage
- Cache histories
- 4 min
- Training and calibration
- 4 min
Raw data extraction and final test evaluation are outside these two reported timings.
Prediction
Reported measurement
- Time
- <1 min
- Hardware
- Not reported
- GPU allocation
- Not separately reported
Reported prediction time after caching, training and calibration.
Prediction timing is separate from final metric computation; no complete test-scoring time is reported.
Separate disease signals from documentation habits
A useful comparison would remove event order while preserving event content. Another would remove laboratory and medication information. These experiments test whether the model benefits from changing physiology or mainly from how often a patient is documented. Evaluate case detection and calibrated risks together, especially across coding eras.
References and provenance
MIMIC-IV, a freely accessible electronic health record dataset
Describes the hospital records underlying the task. Access to the source patient records is controlled.
Task definition and experimental evidence
Defines the discharge cohort, disease panel and follow-up-budget score. Patient histories shown in the interactive example are synthetic.
Code links require access to the project’s GitHub repository.
Leaderboard
SimulatedDesign preview · leaderboard values and research histories are simulated.
| # | Method | Type | Recall at 5% alerts ↑ | AUROC ↑ | AUPRC ↑ | Research time | Run |
|---|---|---|---|---|---|---|---|
| 1 | Agent search | 0.6247 | 0.5240 | 0.6852 | 76 h | ||
| 2 | Agent search | 0.6229 | 0.5224 | 0.6832 | 60 h | ||
| 3 | Multimodal temporal GRU | Reference | 0.6200 | 0.5200 | 0.6800 | — | — |
| 4 | Diagnosis-only temporal GRU | Reference | 0.5084 | 0.4264 | 0.5576 | — | — |
How to read these leaderboard metrics
These leaderboard values and research histories are simulated design examples. The leaderboard is ranked by Recall at 5% alerts. The research curves show relative improvement in this metric over Multimodal temporal GRU; only accepted new-best results update the retained trajectory. A positive gain means a better result, with the sign adjusted for metrics where lower is better. The task protocol is described in Task & evaluation above. These preview columns and trajectories are not measured benchmark results and must not be compared with the recorded experiments. The leaderboard and search curves are simulated design fixtures, not v3 experiment results. The measured temporal-GRU test recall is 0.2363 in the task guide; no diagnosis-only v3 result is reported.
- Recall at 5% alertsPrimary metric · Higher is better
- The fraction of eligible disease cases detected when only the highest-scoring 5% of patients are alerted. It evaluates detection under a fixed alert budget.
- AUROCHigher is better
- Area under the ROC curve. It measures how well the method ranks positive cases above negative cases across classification thresholds.
- AUPRCHigher is better
- Area under the precision–recall curve. It captures the tradeoff between detecting positives and limiting false positive predictions.
AI4AI research progress
Research trajectory
Runtime–performance frontier
How improvement is measured
Relative improvement over Multimodal temporal GRU in the primary metric; only retained results update the best-so-far curve.
Accepted new-best experiments raise the trajectory. The frontier includes evaluated candidates; trial, partial and stopped experiments are excluded.