arXivDaily arXiv每日学术速递 周一至周五更新
看起来今天可以划水了
arXiv 2602.17364cs.LGcs.AI
收藏

一种具有特征稳定性和可解释性的机器学习框架,用于在不完整临床数据下实现可信的决策

A feature-stable and explainable machine learning framework for trustworthy decision-making under incomplete clinical data

Computational Intelligence, Sano Centre for Computational Personalised Medicine(计算智能,Sano计算个性化医学中心);Clinical Studies Group, Randox Laboratories Ltd., Co.(临床研究组,Randox实验室有限公司)

Justyna Andrys-Olek, Paulina Tworek, Luca Gherardini, Mark W. Ruddock, Mary Jo Kurt, Peter Fitzgerald, Jose Sousa

AI总结:

CACTUS是一种针对不完整临床数据设计的可解释机器学习框架,通过提升特征稳定性与可解释性,增强模型在缺失数据下的可信决策能力。

摘要:

机器学习模型越来越多地应用于生物医学数据,但其在高风险领域中的应用仍受到较差的鲁棒性、有限的可解释性和在现实数据扰动(如缺失值)下学习特征的不稳定性限制。特别是,即使模型在预测性能方面表现优异,如果其关键特征在数据完整性变化时波动,也可能无法赢得信任,从而破坏可重复性和后续决策。在这里,我们提出了CACTUS(用于揭示结构的综合抽象和分类工具),这是一种专门设计用于解决这些挑战的可解释机器学习框架,适用于小规模、异质性和不完整临床数据集。CACTUS集成了特征抽象、可解释分类和系统性特征稳定性分析,以量化数据质量下降时信息性特征的一致性保持程度。使用一个包含568名患者的真实世界血尿队列,我们对CACTUS与广泛使用的机器学习方法(包括随机森林和梯度提升方法)在受控水平的随机引入缺失数据下进行了基准测试。我们证明,CACTUS在缺失值增加时能够保持显著更高的顶级特征稳定性,包括在性别的分层分析中。我们的结果表明,特征稳定性提供了与传统性能指标互补的信息,并且对于评估应用于生物医学数据的机器学习模型的可信度至关重要。通过明确量化对缺失数据的鲁棒性并优先考虑可解释和稳定的特征,CACTUS提供了一种可推广的框架,用于可信的数据驱动决策支持。

英文摘要:

Machine learning models are increasingly applied to biomedical data, yet their adoption in high stakes domains remains limited by poor robustness, limited interpretability, and instability of learned features under realistic data perturbations, such as missingness. In particular, models that achieve high predictive performance may still fail to inspire trust if their key features fluctuate when data completeness changes, undermining reproducibility and downstream decision-making. Here, we present CACTUS (Comprehensive Abstraction and Classification Tool for Uncovering Structures), an explainable machine learning framework explicitly designed to address these challenges in small, heterogeneous, and incomplete clinical datasets. CACTUS integrates feature abstraction, interpretable classification, and systematic feature stability analysis to quantify how consistently informative features are preserved as data quality degrades. Using a real-world haematuria cohort comprising 568 patients evaluated for bladder cancer, we benchmark CACTUS against widely used machine learning approaches, including random forests and gradient boosting methods, under controlled levels of randomly introduced missing data. We demonstrate that CACTUS achieves competitive or superior predictive performance while maintaining markedly higher stability of top-ranked features as missingness increases, including in sex-stratified analyses. Our results show that feature stability provides information complementary to conventional performance metrics and is essential for assessing the trustworthiness of machine learning models applied to biomedical data. By explicitly quantifying robustness to missing data and prioritising interpretable, stable features, CACTUS offers a generalizable framework for trustworthy data-driven decision support.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑