Toward Fairness in Machine Learning Models for Predicting Treatment Retention and Premature Discontinuation in Medication for Opioid Use Disorder
6 affiliations
- 1Gary W. Rollins College of Business, The University of Tennessee at Chattanooga, Chattanooga, TN USA
- 2Department of Biomedical Engineering and Chemical Engineering, The University of Texas at San Antonio, San Antonio, TX USA
- 3Department of Biostatistics and Data Science, School of Public Health, The University of Texas Health Science Center at Houston, Houston, TX USA
- 4Center for Spatial-Temporal Modeling for Applications in Population Sciences, School of Public Health, The University of Texas Health Science Center at Houston, Houston, TX USA
- 5School of Engineering Medicine, Texas A&M University, Houston, TX USA
- 6Department of Information Systems and Cybersecurity, The University of Texas at San Antonio, San Antonio, TX USA
Abstract
Persistent low retention and completion rates in medications for opioid use disorder (MOUD) have driven the use of machine learning (ML) models to predict retention and identify patients at risk of premature discontinuation. However, the fairness of these models across patient populations remains largely unexplored, raising concerns about their application in treatment decision support. This study systematically assesses algorithmic fairness in ML models for predicting MOUD retention and premature discontinuation and investigates the effectiveness of bias mitigation techniques. Using the cross-sectional Treatment Episode Data Set–Discharges (TEDS-D), which includes treatment episodes for individuals in the U.S. discharged between 2015 and 2019, we trained four ML models to predict premature treatment discontinuation and retention beyond 180 days among individuals receiving outpatient MOUD. We evaluated overall performance and subgroup-level error rates across patient subgroups defined by race, ethnicity, age, and sex, complemented by model explanation analyses. We further assessed pre-processing, in-processing, and post-processing bias mitigation techniques and their effects on both fairness and predictive performance. The models exhibited substantial performance differences across patient subgroups, including overestimation of the likelihood of premature discontinuation for Black patients and of treatment retention beyond 180 days for older patients. Model explanation analyses further identified race and age as influential predictors, but their impacts on model predictions varied substantially across patient subgroups. Bias mitigation strategies reduced specific fairness gaps but often introduced trade-offs, such as increased error rates for other subgroups or reductions in overall predictive performance. These findings demonstrate that ML models for MOUD outcome prediction can exhibit subgroup-level performance gaps even when overall predictive performance appears acceptable and that bias mitigation can reduce, but not fully eliminate, these gaps without trade-offs. By demonstrating the importance of fairness-aware evaluation and transparent reporting of subgroup performance, this study provides practical insights for the responsible and context-sensitive use of ML models for risk stratification and care prioritization in MOUD treatment settings.
Related
- Biotech File Retention: 90-Day Review vs. Premature Archiving
- Disciform Keratitis That Improves With Poly(Adenosine Diphosphate-Ribose) Polymerase Inhibitor Treatment for Ovarian Cancer but Recurs Upon Discontinuation of Treatment
- Patient-Reported Tolerability, Treatment Initiation, and Discontinuation of Postoperative Progestin Therapy After Cesarean Scar Repair in Women Planning Pregnancy
- Adverse Effects and Treatment Discontinuation of Blood Pressure–Lowering Drugs and Combinations
- Discontinuation of GLP-1RA Treatment Reinstates Binge-Like Eating and Drives Body Weight Past Pre-Treatment Levels in Mice
BibTeX
@article{wang2026toward,
title = {{Toward Fairness in Machine Learning Models for Predicting Treatment Retention and Premature Discontinuation in Medication for Opioid Use Disorder}},
author = {Tongnian Wang and Carolina Vivas-Valencia and Cici Bauer and Yanmin Gong and Kim-Kwang Raymond Choo and Yuanxiong Guo},
journal = {Journal of healthcare informatics research},
year = {2026},
month = aug,
doi = {10.1007/s41666-026-00251-x},
url = {https://europepmc.org/article/PMC/PMC13719986},
note = {CC BY},
}
RIS
TY - JOUR
TI - Toward Fairness in Machine Learning Models for Predicting Treatment Retention and Premature Discontinuation in Medication for Opioid Use Disorder
AU - Tongnian Wang
AU - Carolina Vivas-Valencia
AU - Cici Bauer
AU - Yanmin Gong
AU - Kim-Kwang Raymond Choo
AU - Yuanxiong Guo
PY - 2026
DA - 2026/08/24
JO - Journal of healthcare informatics research
DO - 10.1007/s41666-026-00251-x
UR - https://europepmc.org/article/PMC/PMC13719986
AB - Persistent low retention and completion rates in medications for opioid use disorder (MOUD) have driven the use of machine learning (ML) models to predict retention and identify patients at risk of premature discontinuation. However, the fairness of these models across patient populations remains largely unexplored, raising concerns about their application in treatment decision support. This study systematically assesses algorithmic fairness in ML models for predicting MOUD retention and premature discontinuation and investigates the effectiveness of bias mitigation techniques. Using the cross-sectional Treatment Episode Data Set–Discharges (TEDS-D), which includes treatment episodes for individuals in the U.S. discharged between 2015 and 2019, we trained four ML models to predict premature treatment discontinuation and retention beyond 180 days among individuals receiving outpatient MOUD. We evaluated overall performance and subgroup-level error rates across patient subgroups defined by race, ethnicity, age, and sex, complemented by model explanation analyses. We further assessed pre-processing, in-processing, and post-processing bias mitigation techniques and their effects on both fairness and predictive performance. The models exhibited substantial performance differences across patient subgroups, including overestimation of the likelihood of premature discontinuation for Black patients and of treatment retention beyond 180 days for older patients. Model explanation analyses further identified race and age as influential predictors, but their impacts on model predictions varied substantially across patient subgroups. Bias mitigation strategies reduced specific fairness gaps but often introduced trade-offs, such as increased error rates for other subgroups or reductions in overall predictive performance. These findings demonstrate that ML models for MOUD outcome prediction can exhibit subgroup-level performance gaps even when overall predictive performance appears acceptable and that bias mitigation can reduce, but not fully eliminate, these gaps without trade-offs. By demonstrating the importance of fairness-aware evaluation and transparent reporting of subgroup performance, this study provides practical insights for the responsible and context-sensitive use of ML models for risk stratification and care prioritization in MOUD treatment settings.
SN - 2509-4971
SN - 2509-498X
ER -