Using Under-Represented Subgroup Fine Tuning to Improve Fairness for Disease Prediction

Yanchen Wang, Rex Bone, Will Fleisher, Carole Roan Gresenz, Jean Mitchell, Wilbert van der Klaauw, Crystal Wang, Lisa Singh

Abstract: The role of artificial intelligence is growing in healthcare and disease prediction. Because of its potential impact and demographic disparities that have been identified in machine learning models for disease prediction, there are growing concerns about transparency, accountability and fairness of these predictive models. However, very little research has investigated methods for improving model fairness in disease prediction, particularly when the sensitive attribute is multivariate and when the distribution of sensitive attribute groups is highly skewed. In this work, we explore algorithmic fairness when predicting heart disease and Alzheimer’s Disease and Related Dementias (ADRD). We propose a fine tuning approach to improve model fairness that takes advantage of observations from the majority groups to build a pre-trained model and uses observations from each underrepresented subgroup to fine tune the pre-trained model, thereby incorporating additional specific knowledge about e

External IDs:dblp:conf/biostec/WangBFGMKWS25