Risk-aware feature-token attention with subgroup-calibrated fairness regularization for survival
摘要
Survival analysis in healthcare can be plagued by demographic imbalances due to fairness-oblivious model assumptions or training data biases, especially in clinically consequential domains, such as allogeneic hematopoietic cell transplantation (HCT). To address this problem, we introduce Fair Survival Analysis Transformer (FairSurvTrans), a transformer model for survival forecasting that draws ideas from the design of a large language model (LLM). The model applies positional encoding and multi-head self-attention mechanisms to structured clinical data while incorporating a dynamic fairness-conscious loss function that penalizes performance inequality across the racial subgroups. When trained using synthetically balanced HCT data, FairSurvTrans demonstrated superior predictive performance with an overall C-index value of