In industrial equipment fault diagnosis, few-shot learning offers promise for scenarios with limited data, but traditional methods often overlook causal relationships, undermining prediction reliability. To address this, we propose the causal few-shot relation measurement network (CFRNet), which integrates causal reasoning into few-shot learning. The CFRNet architecture includes a feature extractor, a causal intervention module, a causal decomposition module, and a relationship measurement module, allowing the separation of causal and non-causal elements in data. By using a causal decomposition loss function, CFRNet reconstructs causal structures and represents them as independent features. The feature extractor and relationship measurement module work together to establish a learnable metric space, improving generalization and classification accuracy. Experimental results on public and proprietary datasets demonstrate CFRNet’s superior performance and greater interpretability, setting a new benchmark for causal reasoning in mechanical fault diagnosis.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

CFRNet: Causality Inspired Few-Shot Learning in Mechanical Fault Diagnosis

  • Haoyu He,
  • Juan Xu,
  • Qile Ren,
  • Mingguang Dai,
  • Xuan Liu

摘要

In industrial equipment fault diagnosis, few-shot learning offers promise for scenarios with limited data, but traditional methods often overlook causal relationships, undermining prediction reliability. To address this, we propose the causal few-shot relation measurement network (CFRNet), which integrates causal reasoning into few-shot learning. The CFRNet architecture includes a feature extractor, a causal intervention module, a causal decomposition module, and a relationship measurement module, allowing the separation of causal and non-causal elements in data. By using a causal decomposition loss function, CFRNet reconstructs causal structures and represents them as independent features. The feature extractor and relationship measurement module work together to establish a learnable metric space, improving generalization and classification accuracy. Experimental results on public and proprietary datasets demonstrate CFRNet’s superior performance and greater interpretability, setting a new benchmark for causal reasoning in mechanical fault diagnosis.