基于因果学习的减速机单源域泛化故障诊断方法

Single Source Domain Generalization Fault Diagnosis Method for Reducers Based on Causal Learning

  • 摘要:
    目的 针对现有神经网络在未知工况场景下故障诊断准确率低的问题,提出一种基于因果学习的单源域泛化故障诊断方法。
    方法 首先,将源域样本经过数据组合变换得到辅助域样本,用于模拟源域与目标域之间的领域偏移;然后,构建因果图用于表示输入数据、变换因子、语义特征和故障类别之间的依赖关系;接着,设计反事实推理模块,推断各变换因子对诊断结果的因果效应,并据此对辅助域特征进行校正;最后,通过最小化源域样本与辅助样本在特征空间中的距离进行领域对齐,提高模型在未知工况下的诊断性能。
    结果 结果表明,在完全无需获取目标域样本的条件下,所提方法的诊断准确率达90.34%,相比依赖目标域数据的方法,该方法表现出相媲美甚至更优的性能。
    结论 研究成果可为变工况场景下的船舶设备故障诊断提供关键技术支撑,助力智能船舶的发展。

     

    Abstract:
    Objectives Existing neural network-based fault diagnosis methods usually assume that training and testing data follow the same distribution. However, ship reducers often operate under variable speed, load, vibration, and environmental conditions, which can cause significant distribution shifts in vibration signals and reduce diagnostic accuracy under unseen working conditions. Domain adaptation methods can mitigate this issue but usually require target domain data, which are difficult to collect in practical ship operation. To overcome these challenges, this study proposes a causal learning-based single source domain generalization method, named CLSSDG, for fault diagnosis of ship reducers. The proposed method is designed to improve model generalization under unknown operating conditions by using only one source domain, thereby reducing data collection requirements and enhancing practical applicability.
    Methods First, a data transformation module is developed to generate auxiliary domain samples from source domain data using five transformations, namely noise addition, time stretching, amplitude scaling, slice rearrangement, and zero-value masking. These transformations simulate signal variations caused by interference, speed and load fluctuations, and data loss. Second, a causal graph is constructed to describe the relationships among input data, transformation factors, semantic features, and fault labels. Semantic features represent intrinsic fault-related information, while transformation factors capture condition-related variations. Since observed signals are influenced by both types of factors, models may learn interference features rather than stable fault features. To address this problem, a counterfactual inference module is designed to estimate the causal effects of different transformation factors on diagnostic predictions and identify the main contributors to domain shifts. Third, a domain alignment module corrects auxiliary features according to the estimated causal effects. Feature mappings are assigned to different transformation factors and fused through a weight mapping network. By minimizing the distance between source features and corrected auxiliary features, the model learns robust and transferable representations.
    Results Experiments are conducted on a self-built reducer test bench consisting of a driving motor, loading motor, two-stage reducer gearbox, rotor, and vibration sensors. Six health states are considered, including normal state, gear broken tooth, gear pitting, bearing outer race wear, gear pitting coupled with bearing outer race wear, and gear broken tooth coupled with bearing outer race wear. Data are collected under three speeds, namely 600 r/min, 800 r/min, and 1000 r/min, and six cross-condition diagnostic tasks are designed. The experimental results show that CLSSDG achieves an average diagnostic accuracy of 90.34% without using any target domain samples, outperforming existing single source domain generalization methods. Compared with domain adaptation methods requiring target data, CLSSDG achieves comparable or better performance, demonstrating strong practical value. Ablation studies confirm the necessity of the data transformation, counterfactual inference, and domain alignment modules. Direct domain alignment without causal guidance may lead to negative transfer, whereas causal-guided alignment improves generalization. Confusion matrix analysis shows that CLSSDG provides balanced recognition across all fault categories.
    Conclusions The proposed CLSSDG method offers an effective and interpretable solution for reducer fault diagnosis under unseen operating conditions. Its main contribution lies in integrating causal learning into single source domain generalization, where counterfactual inference is used to analyze the causes of domain shifts and guide feature correction. This strategy reduces the limitations of conventional data augmentation and improves cross-condition diagnostic robustness. Since the method requires only one source domain and no target domain samples, it is suitable for practical ship equipment monitoring. This study provides technical support for intelligent fault diagnosis, predictive maintenance, and health management of ship power transmission systems and contributes to the development of reliable smart ships.

     

/

返回文章
返回