Skip to content

Add DynamicSurvivalRNN model and MIMIC-IV Demo example - #1058

Open
Pinkgirl16 wants to merge 2 commits into
sunlabuiuc:masterfrom
Pinkgirl16:dynamic-survival-rnn
Open

Add DynamicSurvivalRNN model and MIMIC-IV Demo example#1058
Pinkgirl16 wants to merge 2 commits into
sunlabuiuc:masterfrom
Pinkgirl16:dynamic-survival-rnn

Conversation

@Pinkgirl16

@Pinkgirl16 Pinkgirl16 commented Apr 21, 2026

Copy link
Copy Markdown

Contributor: Nishi Uppuluri
NetID: nu4

Type of contribution: Model

Original paper:
Dynamic Survival Analysis for Early Event Prediction
https://arxiv.org/abs/2403.12818

High-level description:
This PR adds DynamicSurvivalRNN, a discrete-time survival model for early event prediction, inspired by the Dynamic Survival Analysis paper. The model predicts per-bin hazards over a fixed future horizon and converts them into event-within-horizon risk through the cumulative failure function. The contribution includes synthetic unit tests and a runnable example on MIMIC-IV Demo ICU mortality. The example uses the first 48 hours of ICU charted measurements to predict death within 7 days of ICU admission and includes a small hidden-dimension ablation (hidden_dim=32 vs 64). The example script runs successfully end to end and produces an ablation summary, though results are noisy because the demo dataset is very small.

Files to review:

  1. pyhealth/models/dynamic_survival_rnn.py
  2. tests/test_dynamic_survival_rnn.py
  3. examples/mimiciv_demo_mortality_dynamic_survival_rnn.py
  4. docs/api/models/pyhealth.models.dynamic_survival_rnn.rst
  5. pyhealth/models/__init__.py
  6. docs/api/models.rst

@Jathurshan0330 Jathurshan0330 left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Good PR

@github-actions

github-actions Bot commented Aug 4, 2026

Copy link
Copy Markdown

This PR has been automatically marked as stale because it has not had recent activity. It will be closed in 7 days if no further activity occurs.

@github-actions github-actions Bot added the stale label Aug 4, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants