arXiv:2509.16750cs.LG2025-09被引 3

用可解释的KAN模型做医疗分类,性能不输黑箱模型。

Interpretable Clinical Classification with Kolmogorov-Arnold Networks

  • 用KAN构建可解释的医疗分类模型,替代传统黑盒方法。
  • Logistic-KAN平均倒数排名达0.76,性能优于或相当主流模型。
  • 支持患者级可视化与近邻检索,适合临床决策透明化需求。

机器学习在临床决策支持中的应用受限于高性能模型的不透明性。临床场景中,预测结果必须可解释、可审计且可操作。本研究探索柯尔莫哥洛夫-阿诺德网络(KAN)作为表格式健康数据临床分类的内在可解释替代方案,旨在平衡预测性能与临床可理解性。提出两种KAN模型:逻辑KAN(Logistic KAN),即逻辑回归的灵活推广;以及柯尔莫哥洛夫-阿诺德加性模型(KAAM),通过特征可分解性生成透明符号表达。在多个公开医疗数据集上评估,并与标准线性、树模型和神经网络基线对比。所有数据集中,所提模型预测性能达到或超过常用基线,同时保持完全可解释性。Logistic-KAN在各项指标中总体排名第一,平均倒数排名为0.76,表现稳定。KAAM在保持竞争力准确率的同时,通过特征可分解性、患者级可视化与最近邻检索提供更强透明度,支持个体预测的直接审查。KAN模型为临床分类提供了可信的替代方案,实现预测性能与可解释性的良好平衡,助力医疗AI可信化(代码:https://github.com/Patricia-A-Apellaniz/classification_with_kans)。

原文摘要 · Abstract (English)

The increasing use of machine learning in clinical decision support has been limited by the lack of transparency of many high-performing models. In clinical settings, predictions must be interpretable, auditable, and actionable. This study investigates Kolmogorov-Arnold Networks (KANs) as intrinsically interpretable alternatives to conventional black-box models for clinical classification of tabular health data, aiming to balance predictive performance with clinically meaningful transparency. We introduce two KAN-based models: the Logistic KAN, a flexible generalization of logistic regression, and the Kolmogorov-Arnold Additive Model (KAAM), an additive variant that yields transparent symbolic representations through feature-wise decomposability. Both models are evaluated on multiple public clinical datasets and compared with standard linear, tree-based, and neural baselines. Across all datasets, the proposed models achieve predictive performance comparable to or exceeding that of commonly used baselines while remaining fully interpretable. Logistic-KAN obtains the highest overall ranking across evaluation metrics, with a mean reciprocal rank of 0.76, indicating consistently strong performance across tasks. KAAM provides competitive accuracy while offering enhanced transparency through feature-wise decomposability, patient-level visualizations, and nearest-patient retrieval, enabling direct inspection of individual predictions. KAN-based models provide a practical and trustworthy alternative to black-box models for clinical classification, offering a strong balance between predictive performance and interpretability for clinical decision support. By enabling transparent, patient-level reasoning and clinically actionable insights, the proposed models represent a promising step toward trustworthy AI in healthcare (code: https://github.com/Patricia-A-Apellaniz/classification_with_kans).

可解释性医疗AIKAN

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