arXiv:2410.07446cs.LG2024-10被引 16

融合量子与经典网络,提升心脏病诊断准确率与可解释性

KACQ-DCNN: Uncertainty-Aware Interpretable Kolmogorov-Arnold Classical-Quantum Dual-Channel Neural Network for Heart Disease Detection

  • 用可学习的一维激活函数替代传统神经元,构建双通道混合模型
  • 在4量子比特下达92.03%准确率,优于37个基准模型,ROC-AUC达94.77%
  • 结合解释工具与不确定性量化,适合临床决策支持场景

心力衰竭是全球主要死亡原因之一,亟需更优的诊断方法。传统机器学习面临高维数据、类别不平衡、特征表示差及可解释性不足等问题;而现有量子机器学习混合模型尚未充分发挥量子优势。本文提出科莫戈罗夫-阿诺德经典-量子双通道神经网络(KACQ-DCNN),以可学习一维激活函数的科莫戈罗夫-阿诺德网络(KAN)替代传统多层感知机,构建新型混合架构。4量子比特、1层的KACQ-DCNN模型在多项指标上超越37个基准模型(含16个经典与12个量子神经网络),达到92.03%准确率,宏平均精确率、召回率与F1分数均为92.00%,ROC-AUC达94.77%,经配对t检验(校正后显著性阈值0.0056)验证显著领先。消融实验证明经典-量子融合带来约2%性能提升。结合LIME与SHAP解释技术增强特征可解释性,通过置信区间预测实现稳健的不确定性量化。结果表明,KACQ-DCNN在提升心血管疾病诊断精度的同时,兼具可解释性与不确定性评估能力。

原文摘要 · Abstract (English)

Heart failure is a leading cause of global mortality, necessitating improved diagnostic strategies. Classical machine learning models struggle with challenges such as high-dimensional data, class imbalances, poor feature representations, and a lack of interpretability. While quantum machine learning holds promise, current hybrid models have not fully exploited quantum advantages. In this paper, we propose the Kolmogorov-Arnold Classical-Quantum Dual-Channel Neural Network (KACQ-DCNN), a novel hybrid architecture that replaces traditional multilayer perceptrons with Kolmogorov-Arnold Networks (KANs), enabling learnable univariate activation functions. Our KACQ-DCNN 4-qubit, 1-layer model outperforms 37 benchmark models, including 16 classical and 12 quantum neural networks, achieving an accuracy of 92.03%, with macro-average precision, recall, and F1 scores of 92.00%. It also achieved a ROC-AUC of 94.77%, surpassing other models by significant margins, as validated by paired t-tests with a significance threshold of 0.0056 (after Bonferroni correction). Ablation studies highlight the synergistic effect of classical-quantum integration, improving performance by about 2% over MLP variants. Additionally, LIME and SHAP explainability techniques enhance feature interpretability, while conformal prediction provides robust uncertainty quantification. Our results demonstrate that KACQ-DCNN improves cardiovascular diagnostics by combining high accuracy with interpretability and uncertainty quantification.

心脏病诊断量子神经网络可解释性不确定性量化

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