Biology & medicine

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.

Highest risk → lowest risk5 places / 100 patients
Recorded case No recorded case Top 5%

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.

CONTENT BRANCHAll retained history events

Diagnoses, measurements, medicines and missing information

Summarize the content of the history
TEMPORAL BRANCHA sequence of earlier and recent events

Read event content together with the time between events

Update a memory as the history unfolds
DEMOGRAPHICSAge + sex

Two input features pass through a learned demographic projection.

Add demographic context
Combine content, timing and demographics

Learn a separate trajectory for each outcome

Accumulate risk across time

Thirteen 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.

Fraction of future cases found within the highest-risk 5%
SystemValidationHidden test
Constant-risk control5.0%5.0%
Risk-group control14.6%18.5%
Sequence model21.1%23.6%
Tree-based reference23.5%26.8%

Compute and resources

Preparation and fitting

Reported measurement

≈8 min
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

<1 min
Time
<1 min
Hardware
Not reported
GPU allocation
Not separately reported

Reported prediction time after caching, training and calibration.

No proportional breakdown: only a range, bound or per-decision timing is reported.

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.

Task description and code

Code links require access to the project’s GitHub repository.

Leaderboard

Simulated

Design preview · leaderboard values and research histories are simulated.

Agent search and reference methods ranked together by Recall at 5% alerts.
#MethodTypeRecall at 5% alerts ↑AUROC ↑AUPRC ↑Research timeRun
1Agent search0.62470.52400.685276 h
2Agent search0.62290.52240.683260 h
3Multimodal temporal GRUReference0.62000.52000.6800——
4Diagnosis-only temporal GRUReference0.50840.42640.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.