Risk of Training Diagnostic Algorithms on Data with Demographic Bias

Samaneh Abbasi-Sureshjani,Ralf Raumanns,Britt E. J. Michels,Gerard Schouten,Veronika Cheplygina
DOI: https://doi.org/10.48550/arXiv.2005.10050
2020-06-17
Abstract:One of the critical challenges in machine learning applications is to have fair predictions. There are numerous recent examples in various domains that convincingly show that algorithms trained with biased datasets can easily lead to erroneous or discriminatory conclusions. This is even more crucial in clinical applications where the predictive algorithms are designed mainly based on a limited or given set of medical images and demographic variables such as age, sex and race are not taken into account. In this work, we conduct a survey of the MICCAI 2018 proceedings to investigate the common practice in medical image analysis applications. Surprisingly, we found that papers focusing on diagnosis rarely describe the demographics of the datasets used, and the diagnosis is purely based on images. In order to highlight the importance of considering the demographics in diagnosis tasks, we used a publicly available dataset of skin lesions. We then demonstrate that a classifier with an overall area under the curve (AUC) of 0.83 has variable performance between 0.76 and 0.91 on subgroups based on age and sex, even though the training set was relatively balanced. Moreover, we show that it is possible to learn unbiased features by explicitly using demographic variables in an adversarial training setup, which leads to balanced scores per subgroups. Finally, we discuss the implications of these results and provide recommendations for further research.
Machine Learning,Artificial Intelligence
What problem does this paper attempt to address?
The problem that this paper attempts to solve is the unfair prediction problem in medical image diagnosis algorithms due to the demographic bias of the data set. Specifically, the author is concerned that when training a diagnosis algorithm, if the data set used does not fully represent groups with different demographic characteristics such as age and gender, it may lead to poor performance of the algorithm on certain subgroups or produce discriminatory conclusions. This bias is particularly important in clinical applications because medical image diagnosis algorithms are usually designed mainly based on a given set of medical images, ignoring the influence of demographic variables such as age, gender, and race. To quantify and solve this problem, the author carried out the following several tasks: 1. **Literature survey**: A survey was conducted on the papers of the MICCAI 2018 conference to evaluate whether these papers described the demographic characteristics in the data set when using medical images for diagnosis and whether they discussed demographic bias. 2. **Experimental analysis**: Using a public skin lesion data set, it was shown that even when the training set is relatively balanced, there are significant differences in the performance of the classifier on different age and gender subgroups. 3. **Adversarial training**: An adversarial training method was proposed. By explicitly using demographic variables to learn unbiased features, more balanced performance on different subgroups can be achieved. Through these tasks, the author aims to emphasize the importance of considering demographic characteristics in medical image diagnosis tasks and provide methods and suggestions for reducing bias.