The invention discloses a
federated learning fault diagnosis method and a
federated learning fault diagnosis
system for multi-source
unbalanced data of a
harmonic reducer, relates to a
harmonic reducer fault diagnosis technology, and aims to solve the problem of low diagnosis accuracy caused by unbalanced sample numbers of different fault categories of the
harmonic reducer of an
industrial robot and limited single-source
signal acquisition information. The method is technically characterized by comprising the following steps of: performing
wavelet transform on multi-source signals of different users to construct a time-frequency graph
data set; carrying out
equalization processing on the
unbalanced data set by utilizing an improved data enhancement method; an effective channel attention mechanism is introduced, and the output of a residual
branch is weighted through a learnable weight, so that the adaptability of the model to different residual information and the extraction capability of the model to data key features are enhanced; the method comprises the following steps: mining complementary information among multi-source signals through an improved multi-mode variational auto-
encoder to perform
feature fusion, and constructing a multi-user personalized local model; and the
server aggregates local
model parameters and updates the model, and guarantees user island privacy data through federal learning, thereby performing fault diagnosis on the harmonic reducer under the multi-source
unbalanced data. A harmonic reducer
signal acquisition experiment platform is established for
verification, the characteristics of multi-source unbalanced data can be effectively extracted by the method,
information fusion is realized, the average fault diagnosis accuracy is 98.8%, and the performance is superior to that of the compared method.