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
- Matthew C. Diamond
- Nicholas Petrick (ORCID: https://orcid.org/0000-0001-5167-8899)
- Suchi Saria (ORCID: https://orcid.org/0000-0002-7667-5210)
- Adarsh Subbaswamy (ORCID: https://orcid.org/0000-0003-2020-1806)
- Berkman Sahiner (ORCID: https://orcid.org/0000-0003-2804-2264)
- Roy J. Adams (ORCID: https://orcid.org/0000-0001-6859-3007)
- Vinay Pai
Institutions
- 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)
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
- 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