Skip to content

Feat/sofa deterioration lab#1030

Open
kevintrancs wants to merge 2 commits intosunlabuiuc:masterfrom
kevintrancs:feat/sofa-deterioration-lab
Open

Feat/sofa deterioration lab#1030
kevintrancs wants to merge 2 commits intosunlabuiuc:masterfrom
kevintrancs:feat/sofa-deterioration-lab

Conversation

@kevintrancs
Copy link
Copy Markdown

Contributor Info

NetID: kevint6
Email: kevint6@illinois.edu

Type of Contribution

Standalone Task (Option 3)

Original Paper Reference

Paper: Early Prediction of Causes (not Effects) in Healthcare by Long-Term Clinical Time Series Forecasting (Staniek et al., 2024)

Implementation Description

This PR implements the SofaLabForecastingMIMIC3 task. Following the "prediction of causes" paradigm from the referenced paper, this task forecasts three specific clinical variables (Bilirubin, Creatinine, and Platelets) and then deterministically derives a binary SOFA-deterioration label based on the 24h delta in SOFA scores.
Note: Because the paper's purpose to produce a vectorized forecasting as a the output, like the input, I couldn't use pyhealth native models to be correct with the paper other than opt to just train on sofa labels, but that would defeat the purpose of the uniqueness of the paper. So I opted to use sklearn LinearRegression as a baseline model to test.

File Guide

pyhealth/tasks/sofa_lab_forecasting_mimic3.py: Core task implementation.
tests/core/test_mimic3_sofa_lab.py: Comprehensive test suite using synthetic data.
examples/mimic3_sofa_lab_forecasting_linear.py: Ablation study comparing 12h vs. 24h lookback performance.
docs/api/tasks/pyhealth.tasks.sofa_lab_forecasting_mimic3.rst: API documentation.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant