用可解释的KAN模型替代黑箱神经网络,提升核能数据建模的可信度
Opening the Black-Box: Symbolic Regression with Kolmogorov-Arnold Networks for Energy Applications
- KAN通过训练转化为符号方程,实现模型架构可读
- 在8个核能数据集上,KAN与传统神经网络精度相当
- SHAP分析显示KAN捕捉真实物理规律,适合高安全场景
尽管现代机器学习方法在速度和精度上表现优异,但很少能提供可解释性或可理解性——这在医疗、金融和工程等高敏感行业至关重要。本文基于代表核能领域的八个数据集,对比了传统的前馈神经网络(FNN)与科莫戈罗夫-阿诺德网络(KAN)。不仅评估模型性能与准确性,还通过模型结构和事后SHAP分析考察可解释性与可解释性。结果显示,在输出维度受限的情况下,KAN与FNN在所有数据集上的精度相当;而训练后,KAN可转化为符号表达式,实现完全可读,反观FNN仍为黑箱。借助核型SHAP的事后可解释性分析发现,KAN从实验数据中学习到真实的物理关系,而FNN仅生成统计上准确的结果。总体而言,该研究证明了KAN是传统机器学习方法的有前景替代方案,尤其适用于同时要求高精度与可理解性的应用。
原文摘要 · Abstract (English)
While most modern machine learning methods offer speed and accuracy, few promise interpretability or explainability -- two key features necessary for highly sensitive industries, like medicine, finance, and engineering. Using eight datasets representative of one especially sensitive industry, nuclear power, this work compares a traditional feedforward neural network (FNN) to a Kolmogorov-Arnold Network (KAN). We consider not only model performance and accuracy, but also interpretability through model architecture and explainability through a post-hoc SHAP analysis. In terms of accuracy, we find KANs and FNNs comparable across all datasets, when output dimensionality is limited. KANs, which transform into symbolic equations after training, yield perfectly interpretable models while FNNs remain black-boxes. Finally, using the post-hoc explainability results from Kernel SHAP, we find that KANs learn real, physical relations from experimental data, while FNNs simply produce statistically accurate results. Overall, this analysis finds KANs a promising alternative to traditional machine learning methods, particularly in applications requiring both accuracy and comprehensibility.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。