arXiv:2605.22243cs.LGcs.AI2026-05

用可解释AI自动优化临床预测模型,提升准确性和透明度

Explainable AI for Data-Driven Design of High-Dimensional Predictive Studies

  • 通过可解释AI识别应排除的特征、非线性关系和特征交互
  • 在24.5万患者数据上使C-index从0.805提升至0.815
  • 推荐结果均获文献支持,适合临床研究与模型设计者使用

预测建模在健康数据分析和数据驱动的临床决策中至关重要。然而,当需筛选、转换或建模上百个特征时,手动设计预测研究极具挑战。尽管复杂机器学习模型性能优异,但其“黑箱”特性限制了临床信任与可解释性。我们开发并评估了一种探索性AI推荐器,为现有可解释统计模型提供数据驱动的优化建议。该框架利用灵活的AI捕捉复杂数据模式,并通过可解释AI技术将其转化为三类推荐:特征剔除、非线性项、特征交互。在245,614名患者的跌倒或相关伤害首次发生时间预测中,基准模型(无交互与非线性)的C-index为0.805(95% CI 0.798–0.812),经本方法增强后提升至0.815(95% CI 0.809–0.822),校准性也改善(截距:-0.006→0.003;斜率:1.063→0.950)。推荐包括剔除23个特征、为2个特征添加非线性项、引入221个特征交互,所有建议均有文献支持。该方法在另两个公开数据集上亦表现良好,证明其广泛适用性。结果表明,可解释AI与数据驱动研究设计结合,能有效提升高维透明预测模型的构建效率与性能。

原文摘要 · Abstract (English)

Predictive modelling is important for health data analysis and data-driven clinical decision-making. However, predictive studies are challenging to design optimally by hand when tens or even hundreds of features require selection, transformation, or interaction modelling. While complex machine learning models offer high performance, their "black-box" nature limits the clinical trust, transparency, and interpretability required for decision-making. We developed and evaluated an Exploratory AI Recommender that provides data-driven recommendations to improve predictive performance of existing interpretable statistical models. The developed framework uses flexible AI modelling to capture complex data patterns and explainable AI techniques to translate the patterns into three recommendation types: feature exclusion, non-linear terms, and feature interactions. We evaluated the framework by comparing predictive performance of a baseline (i.e., no interactions or non-linear terms) Cox Proportional Hazards (CPH) model against an augmented CPH incorporating recommendations suggested by our method. The primary analysis predicts the time to the first occurrence of a fall or related injury in 245,614 patients. Our method recommended excluding 23 features, including non-linear terms for two features, and including 221 suggested feature interactions. The C-index improved from 0.805 (95% CI 0.798-0.812) to 0.815 (95% CI 0.809-0.822), and so did calibration (intercept: -0.006 to 0.003; slope: 1.063 to 0.950). All recommendations were supported by existing literature. The method also proved effective on two additional public datasets, demonstrating wider applicability. The proposed Exploratory AI Recommender demonstrates the potential of explainable AI and data-driven study design to improve the process of developing, and the performance of high-dimensional transparent predictive models.

可解释AI临床预测特征选择生存分析

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