arXiv:2606.02671cs.LGcs.AI2026-06被引 1

用排序指标优化生存预测,让器官分配更高效

Aligning Data-Driven Predictors with Allocation: A Decision-Focused Approach to Survival Analysis

论文配图:Aligning Data-Driven Predictors with Allocation: A Decision-Focused Approach to Survival Analysis
图 1 · 摘自论文原文
  • 以信息检索的NDCG指标替代传统精度,直接优化分配决策效果
  • 在真实心脏移植数据上提升模型排序能力50%-100%,每年多救数万生命年
  • 解决右删失问题,适合医疗资源分配等高风险决策场景

机器学习预测模型虽广泛用于自动化决策,但常与实际任务脱节。本文以器官分配为例,指出仅优化标准指标(如C-index)的生存预测模型,在实际分配中可能表现极差,甚至不如随机选择。为此,提出一种面向决策的学习方法,基于归一化折扣累计收益(NDCG)优化生存预测。理论证明,NDCG可转化为分配性能保证。实验中,通过自助法优化现有模型,在美国心脏移植历史数据上使基准模型的NDCG提升50%-100%。部署后预计每年可多挽救数万生命年。该框架可推广至其他依赖预测的决策场景。

原文摘要 · Abstract (English)

Machine learning predictors have become essential tools for guiding automated decision making. However, a major misalignment persists: predictive models are typically optimized in terms of standard statistical metrics in isolation from the algorithmic tasks they inform. We highlight this incongruity in the high-stakes domain of organ allocation by demonstrating that any algorithm relying on (even highly accurate) survival predictors optimized for standard metrics -- such as the Concordance index (C-index) -- can yield arbitrarily poor outcomes when used for allocation, failing to guarantee utility better than a uniform random selection. To bridge the gap between survival analysis and policy optimization, we introduce a decision-focused learning approach based on optimizing normalized discounted cumulative gain (NDCG), a mainstay metric in information retrieval. We establish the utility of NDCG in survival analysis by proving that it translates to guarantees on the performance of allocation. Empirically, we propose a bootstrapping approach to optimize the NDCG of existing survival models. Unlike prior work, we also address the challenge of right censorship when evaluating ranking. On historical heart transplant data from the US, our method dramatically boosts the NDCG of baseline models by 50-100%, which translates to tens of thousands of additional life years gained annually when deployed for transplant allocation. We anticipate that our framework will find broader applications in decision making with predictions.

生存分析决策优化器官分配NDCG

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。