Fairness in Survival Outcome Prediction for Medical Treatments

Yunqi Li, Hanxiong Chen, Lanjing Zhang, Yongfeng Zhang

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

Cancer is a life-threatening disease and affects millions of people globally. Predicting the survival outcome of patients after receiving a cancer diagnosis is of great importance for personalized treatment planning. While predictive machine learning models have shown promising results in predicting survival outcomes, they may result in unfair treatments against certain groups, such as those based on age, gender, race and socioeconomic status. In this paper, we analyze and identify what kind of fairness criterion should be developed in cancer survival predictions. Considering the high stakes involved, we believe the basis for developing fairness in this field should be assisting to improve the model performance of diverse patients, rather than hurting the performance of certain patients in exchange for better performance of other patients. We suggest min-max fairness as an appropriate concept that seeks to minimize the group error of the worst-off group, instead of pursuing a strict equal error rate across various patient groups which would reduce the utility of the advantaged groups. We investigate the application of min-max fairness through active sampling in predicting the survival outcome of cancer patients using machine learning models, and evaluate the fairness-aware method on several publicly available cancer datasets. The experimental results demonstrate that the fairness-aware predictive method can help improve the model performance of the disadvantaged patient group without hurting the advantaged patient group, and thus increase the overall model performance for all patients.

Original languageEnglish (US)
Title of host publication2024 58th Annual Conference on Information Sciences and Systems, CISS 2024
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9798350369298
DOIs
StatePublished - 2024
Externally publishedYes
Event58th Annual Conference on Information Sciences and Systems, CISS 2024 - Princeton, United States
Duration: Mar 13 2024Mar 15 2024

Publication series

Name2024 58th Annual Conference on Information Sciences and Systems, CISS 2024

Conference

Conference58th Annual Conference on Information Sciences and Systems, CISS 2024
Country/TerritoryUnited States
CityPrinceton
Period3/13/243/15/24

ASJC Scopus subject areas

  • Computer Networks and Communications
  • Information Systems
  • Safety, Risk, Reliability and Quality
  • Control and Optimization
  • Modeling and Simulation
  • Computational Theory and Mathematics

Keywords

  • AI for Health
  • Algorithmic Bias
  • Trustworthy AI

Cite this