BR-MTFL: A Novel Byzantine Resilience-Enhanced Multitask Federated Learning Framework for High-Speed Train Fault Diagnosis

Junxian You, Rui Yang*, Yifan Zhan, Baoye Song, Yong Zhang, Zidong Wang

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

Abstract

In high-speed train systems deployed across diverse geographical regions, robust fault diagnosis techniques are essential for ensuring operational safety. This article proposes the Byzantine resilience-enhanced multitask federated learning (BR-MTFL) framework, a novel framework tailored for the complexities of fault diagnosis in traction asynchronous motors under varying operational conditions. This framework innovatively introduces multitask federated learning (MTFL) to accommodate regional law restrictions and varying fault diagnosis requirements. In addition, the Byzantine resilience of our proposed framework is specifically enhanced to address the challenges posed by inconsistent and potentially misleading feature distributions across different train networks. BR-MTFL is practically validated through experiments conducted across nine clients, each representing a distinct set of fault types and operational conditions typical of high-speed trains. The experiments demonstrate the ability of BR-MTFL to outperform conventional federated learning frameworks in terms of accuracy and resilience to Byzantine threats. BR-MTFL establishes a new standard for federated learning applications in high-speed train fault diagnosis, particularly where data diversity and privacy dominate.

Original languageEnglish
Article number3518713
JournalIEEE Transactions on Instrumentation and Measurement
Volume74
DOIs
Publication statusPublished - 12 Mar 2025

Keywords

  • Byzantine resilience
  • fault diagnosis
  • federated learning
  • high-speed train
  • multitask learning

Fingerprint

Dive into the research topics of 'BR-MTFL: A Novel Byzantine Resilience-Enhanced Multitask Federated Learning Framework for High-Speed Train Fault Diagnosis'. Together they form a unique fingerprint.

Cite this