用多任务元学习预测手术效果,提升个性化决策准确性。
Causal Machine Learning for Surgical Interventions
- 将每种手术选择建模为独立任务,共享跨任务特征表示。
- 在脊柱融合数据上平均AUC达0.84,治疗效应估计误差最低(0.2778)。
- 适合需要精准评估手术收益的临床医生与医疗AI研究者。
外科决策复杂,需理解患者特征、干预措施与结果之间的因果关系。在脊柱融合或脊柱侧弯矫正等高风险场景中,传统统计方法难以处理异质性数据,限制了个体化治疗效应(ITE)的准确估计。本文提出X-MultiTask多任务元学习框架,将不同手术决策(如前路与后路入路、手术与否)视为独立任务,同时学习任务间共享表征,并引入逆概率加权(IPW)增强因果有效性。在两个数据集上评估:(1) 公开脊柱融合数据集(1,017名患者),分析前路与后路入路对并发症严重程度的影响;(2) 私有AIS数据集(368名患者),比较后路脊柱融合(PSF)与非手术管理对患者报告结局(PROs)的影响。模型在前路组平均AUC达0.84,在后路组保持0.77的竞争力。在治疗效应估计上,整体NN-PEHE为0.2778,ATE为0.0763,优于基线。在预测AIS患者PROs时,各维度表现均更优,NN-PEHE为0.2551,ATE为0.0902。该模型提供稳健的个体化因果估计,助力精准外科决策与患者预后改善。代码已开源。
原文摘要 · Abstract (English)
Surgical decision-making is complex and requires understanding causal relationships between patient characteristics, interventions, and outcomes. In high-stakes settings like spinal fusion or scoliosis correction, accurate estimation of individualized treatment effects (ITEs) remains limited due to the reliance on traditional statistical methods that struggle with complex, heterogeneous data. In this study, we develop a multi-task meta-learning framework, X-MultiTask, for ITE estimation that models each surgical decision (e.g., anterior vs. posterior approach, surgery vs. no surgery) as a distinct task while learning shared representations across tasks. To strengthen causal validity, we incorporate the inverse probability weighting (IPW) into the training objective. We evaluate our approach on two datasets: (1) a public spinal fusion dataset (1,017 patients) to assess the effect of anterior vs. posterior approaches on complication severity; and (2) a private AIS dataset (368 patients) to analyze the impact of posterior spinal fusion (PSF) vs. non-surgical management on patient-reported outcomes (PROs). Our model achieves the highest average AUC (0.84) in the anterior group and maintains competitive performance in the posterior group (0.77). It outperforms baselines in treatment effect estimation with the lowest overall $ε_{\text{NN-PEHE}}$ (0.2778) and $ε_{\text{ATE}}$ (0.0763). Similarly, when predicting PROs in AIS, X-MultiTask consistently shows superior performance across all domains, with $ε_{\text{NN-PEHE}}$ = 0.2551 and $ε_{\text{ATE}}$ = 0.0902. By providing robust, patient-specific causal estimates, X-MultiTask offers a powerful tool to advance personalized surgical care and improve patient outcomes. The code is available at https://github.com/Wizaaard/X-MultiTask.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。