Federal learning model training method and device, electronic equipment and storage medium

Abstract

The invention provides a federal learning model training method and device, electronic equipment and a storage medium, and the training method comprises the steps: carrying out the sample alignment with a data provider server; generating a feature code set, and sending the feature number and the public key of the data provider server to the data provider server; dividing a current sample into a training set and a verification set, and obtaining a parameter set of the federated learning model; performing M times of iterative training on the federal learning model according to the training set, the verification set, the parameter set and the feature coding set; and in the process of each iteration training in the M iteration training, if it is judged that the federated learning model meets the early stop condition, controlling the M iteration training to stop early, and obtaining target parameters of the federated learning model obtained by the last iteration training in the M iteration training. Therefore, training can be more efficient, meanwhile, the modeling effect is improved, an early stop strategy is adopted, overfitting of the model is avoided, and meanwhile complexity is reduced.

Description

technical field [0001] The present application relates to the technical field of data processing, and in particular to a training method, device, electronic device and storage medium of a federated learning model. Background technique [0002] With the development of machine learning, more and more machine learning techniques are being applied in various industries. The quantity and quality of data often determine the upper limit of the effectiveness of machine learning models. However, as regulations and supervision become more and more stringent, and people pay more and more attention to data security and privacy protection, the phenomenon of data islands is formed. In such a scenario, federated learning emerges as the times require, which allows participants to jointly train on the basis of not sharing data to solve the problem of data islands. [0003] Among related technologies, federated learning is an encrypted distributed machine learning technology that integrates...

AI Technical Summary

Problems solved by technology

However, as regulations and supervision become more and more stringent, and people pay more and more attention to data security and privacy protection, the phenomenon of data islands is formed.

Images