Health Condition Assessment of Braking System Based on Multi-agent Federated Learning

Abstract The brake system has the characteristics of multi-component, multi working conditions and complex degradation process, which brings great challenges to its health condition assessment. As it is difficult for a single brake agent to make a comprehensive and accurate health condition assessment, a health condition assessment model based on multi-agent federated learning is proposed in this paper. The different agents train their brake health data under different working conditions and states, which ensures the accuracy of health condition assessment and the safety of data of each agent. Aiming at the problem that it is difficult to determine the credibility of agent data in the process of federated learning, a credibility of agent data scheme based on evidence theory is proposed, which not only reduces the cost of calculation and communication, but also further improves the accuracy of health condition assessment. Simulation results show that the scheme not only has higher prediction accuracy, but also can ensure the security of agent data compared with the conventional centralized training scheme.