A data-driven framework for identifying patient subgroups on which an AI/machine learning model may underperform

A fundamental goal of evaluating the performance of a clinical model is to ensure it performs well across a diverse intended patient population. A primary challenge is that the data used in model development and testing often consist of many overlapping, heterogeneous patient subgroups that may not be explicitly defined or labeled. While a model's average performance on a dataset may be high, the model can have significantly lower performance for certain subgroups, which may be hard to detect. We describe an algorithmic framework for identifying subgroups with potential performance disparities (AFISP), which produces a set of interpretable phenotypes corresponding to subgroups for which the model's performance may be relatively lower. This could allow model evaluators, including developers and users, to identify possible failure modes prior to wide-scale deployment. We illustrate the application of AFISP by applying it to a patient deterioration model to detect significant subgroup performance disparities, and show that AFISP is significantly more scalable than existing algorithmic approaches.

Authors

Institutions

Publication Details

Journal
npj Digital Medicine
Published
2024-11-21
DOI
https://doi.org/10.1038/s41746-024-01275-6
Citations
10
Primary Topic
Machine Learning in Healthcare
Type
article
Field-Weighted Citation Impact
3.47

Funders

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

A data-driven framework for identifying patient subgroups on which an AI/machine learning model may underperform

Matthew C. Diamond, Nicholas Petrick, Suchi Saria, Adarsh Subbaswamy et al.
10 citations
npj Digital Medicine
Machine Learning in Healthcare
3.47
article

A data-driven framework for identifying patient subgroups on which an AI/machine learning model may underperform

Matthew C. Diamond, Nicholas Petrick, Suchi Saria, Adarsh Subbaswamy, Berkman Sahiner, Roy J. Adams, Vinay Pai
article en
10 citations

Abstract

A fundamental goal of evaluating the performance of a clinical model is to ensure it performs well across a diverse intended patient population. A primary challenge is that the data used in model development and testing often consist of many overlapping, heterogeneous patient subgroups that may not be explicitly defined or labeled. While a model's average performance on a dataset may be high, the model can have significantly lower performance for certain subgroups, which may be hard to detect. We describe an algorithmic framework for identifying subgroups with potential performance disparities (AFISP), which produces a set of interpretable phenotypes corresponding to subgroups for which the model's performance may be relatively lower. This could allow model evaluators, including developers and users, to identify possible failure modes prior to wide-scale deployment. We illustrate the application of AFISP by applying it to a patient deterioration model to detect significant subgroup performance disparities, and show that AFISP is significantly more scalable than existing algorithmic approaches.

npj Digital MedicineVol. 7(1)
United States Food and Drug Administration (US), Center for Devices and Radiological Health (US), Johns Hopkins University (US), Bayer (United States) (US), Johns Hopkins Medicine (US)
U.S. Department of Energy, U.S. Food and Drug Administration, Oak Ridge Institute for Science and Education, Center of Excellence in Regulatory Science and Innovation
Openalex Percentile: Top 6%
Machine Learning in Healthcare
3.47
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.