用可解释的神经网络实现轴承故障与严重度精准分类
Explainable fault and severity classification for rolling element bearings using Kolmogorov-Arnold networks
- 基于柯尔莫哥洛夫-阿诺德网络自动选特征并优化参数
- 在两个数据集上实现故障检测100%准确率,多数分类任务达满分F1
- 模型轻量且可解释,适合工业实时监测与研究场景
滚动轴承是旋转机械的关键部件,其性能直接影响工业系统的效率与可靠性。轴承故障是导致设备失效的主要原因,常引发昂贵停机、生产下降甚至灾难性损坏。本文提出一种基于柯尔莫哥洛夫-阿诺德网络的方法,在统一框架内实现自动特征选择、超参数调优与可解释故障分析。通过训练浅层网络并最小化所选特征数量,该框架生成轻量化模型,利用特征归因和激活函数的符号化表达实现结果可解释性。在两个主流轴承故障诊断数据集上验证,该方法在故障检测任务中达到100% F1分数,多数故障与严重度分类任务也获得满分表现。尤其能同时处理同一数据集中不平衡、不对中等多种故障类型。符号化表示增强了模型可解释性,特征归因揭示了各任务最优信号类型。结果表明该框架适用于实时机械监控等实际应用及需要高效可解释模型的科研需求。
原文摘要 · Abstract (English)
Rolling element bearings are critical components of rotating machinery, with their performance directly influencing the efficiency and reliability of industrial systems. At the same time, bearing faults are a leading cause of machinery failures, often resulting in costly downtime, reduced productivity, and, in extreme cases, catastrophic damage. This study presents a methodology that utilizes Kolmogorov-Arnold Networks to address these challenges through automatic feature selection, hyperparameter tuning and interpretable fault analysis within a unified framework. By training shallow network architectures and minimizing the number of selected features, the framework produces lightweight models that deliver explainable results through feature attribution and symbolic representations of their activation functions. Validated on two widely recognized datasets for bearing fault diagnosis, the framework achieved perfect F1-Scores for fault detection and high performance in fault and severity classification tasks, including 100% F1-Scores in most cases. Notably, it demonstrated adaptability by handling diverse fault types, such as imbalance and misalignment, within the same dataset. The symbolic representations enhanced model interpretability, while feature attribution offered insights into the optimal feature types or signals for each studied task. These results highlight the framework's potential for practical applications, such as real-time machinery monitoring, and for scientific research requiring efficient and explainable models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。