将SHAP拓展到循环谱域,提升机械故障诊断的可解释性
CS-SHAP: Extending SHAP to Cyclic-Spectral Domain for Better Interpretability of Intelligent Fault Diagnosis
- 提出循环谱域的SHAP方法,同时分析载波与调制频率贡献
- 在三个数据集上验证,解释结果更符合故障机理
- 开源代码,适合高可靠性场景下的模型可解释性需求
神经网络凭借强大的非线性映射和端到端能力,广泛应用于机械智能故障诊断(IFD)。然而,作为典型的黑箱模型,其决策依据难以理解,限制了在高可靠性场景中的应用。为此,研究者提出了多种可解释性方法。其中,后处理方法无需修改模型结构,保持灵活性与可扩展性。但现有方法常因需预处理破坏端到端特性,或忽略故障机理,导致解释效果不佳。为此,本文推导出循环谱(CS)变换,将沙普利加法解释(SHAP)拓展至循环谱域,提出CS-SHAP。该方法能同时评估载波与调制频率的贡献,更贴近故障机制,提供更清晰、准确的解释。通过三个数据集验证,CS-SHAP展现出优越的可解释性,确保正确性、可复现性和实际性能。开源代码已发布于https://github.com/ChenQian0618/CS-SHAP,具备成为IFD领域后处理可解释性基准的潜力,甚至适用于其他分类任务。
原文摘要 · Abstract (English)
Neural networks (NNs), with their powerful nonlinear mapping and end-to-end capabilities, are widely applied in mechanical intelligent fault diagnosis (IFD). However, as typical black-box models, they pose challenges in understanding their decision basis and logic, limiting their deployment in high-reliability scenarios. Hence, various methods have been proposed to enhance the interpretability of IFD. Among these, post-hoc approaches can provide explanations without changing model architecture, preserving its flexibility and scalability. However, existing post-hoc methods often suffer from limitations in explanation forms. They either require preprocessing that disrupts the end-to-end nature or overlook fault mechanisms, leading to suboptimal explanations. To address these issues, we derived the cyclic-spectral (CS) transform and proposed the CS-SHAP by extending Shapley additive explanations (SHAP) to the CS domain. CS-SHAP can evaluate contributions from both carrier and modulation frequencies, aligning more closely with fault mechanisms and delivering clearer and more accurate explanations. Three datasets are utilized to validate the superior interpretability of CS-SHAP, ensuring its correctness, reproducibility, and practical performance. With open-source code and outstanding interpretability, CS-SHAP has the potential to be widely adopted and become the post-hoc interpretability benchmark in IFD, even in other classification tasks. The code is available on https://github.com/ChenQian0618/CS-SHAP.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。