Skip to content

Exclude pediatric patients (younger than 18) from clinical prediction tasks#1150

Open
DarylOkeke wants to merge 3 commits into
sunlabuiuc:masterfrom
DarylOkeke:fix/exclude-pediatric-patients
Open

Exclude pediatric patients (younger than 18) from clinical prediction tasks#1150
DarylOkeke wants to merge 3 commits into
sunlabuiuc:masterfrom
DarylOkeke:fix/exclude-pediatric-patients

Conversation

@DarylOkeke
Copy link
Copy Markdown

Multiple clinical prediction tasks had TODO comments saying that patients under 18 should be excluded, but the filtering was never implemented.

This adds age filtering to the mortality prediction, length-of-stay, and drug recommendation tasks across MIMIC-III, MIMIC-IV, eICU, and OMOP datasets. Adult prediction benchmarks shouldn't mix in pediatric cohorts b/c the clinical patterns are too different.

Some notes:

  • eICU stores age as a string and uses "> 89" for patients over 89, so that case is handled explicitly
  • MIMIC-III requires deriving age from dob and admission time since it doesn't store a precomputed age field
  • All checks fail-open: if age data is missing or unparseable, the patient is included instead of dropped

Copy link
Copy Markdown
Collaborator

@jhnwu3 jhnwu3 left a comment

Choose a reason for hiding this comment

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

Hey nice first PR!

Is it possible to add a Boolean arg for this filtering? I think it'd be nice to be able to quickly ablation study this difference in patient demographics later.

@DarylOkeke
Copy link
Copy Markdown
Author

Thanks! I added exclude_minors to all the affected classes and legacy functions in the 3 task files. Defaults to true so the existing behavior stays the same

Another thing though, I saw that LengthOfStayPredictioneICU doesn't have an age check at all. The original TODOs didn't cover it but the other 3 LOS classes do. Want me to add one for consistency, or no?

@jhnwu3
Copy link
Copy Markdown
Collaborator

jhnwu3 commented May 25, 2026

Yeah that would be nice

@DarylOkeke
Copy link
Copy Markdown
Author

Finished!

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.

2 participants