Method and device for data processing based on federated learning, equipment and medium

Abstract

The invention discloses a method and a device for data processing based on federated learning, equipment and a medium. The method comprises the following steps executed by a first terminal: determining user feature data shared by a first terminal and a second terminal; performing feature coding processing on the user feature data to obtain to-be-processed feature data; obtaining a model predictionvalue obtained by processing based on the to-be-processed feature data; obtaining a loss value obtained by processing the training label data and the model prediction value by adopting a predefined loss function; sending the external loss value to the second terminal in an encryption mode if the loss value is the external loss value; and determining a target gradient based on the internal loss value and a current model parameter corresponding to the first to-be-trained model if the loss value is an internal loss value, and performing model optimization on the first to-be-trained model according to the target gradient to obtain a target prediction model. The data processing method based on federated learning can effectively improve the model training efficiency and accuracy.

Description

technical field [0001] The present invention relates to the technical field of artificial intelligence, in particular to a data processing method, device, equipment and medium based on federated learning. Background technique [0002] With the widespread application of artificial intelligence technology, various machine learning methods have emerged. At present, in order to solve the data island and realize the interaction of information and the learning of models under the premise of not providing sensitive data to the outside world, a new learning method - federated learning is proposed. Federated Learning (Federated Learning) is an emerging basic technology of artificial intelligence. Its design goal is to ensure information security during big data exchange and ensure compliance with laws and regulations. Efficient machine learning. [0003] At present, when some financial institutions are training the model, because the user data of the financial institution itself ca...

AI Technical Summary

Problems solved by technology

[0004] Embodiments of the present invention provide a data processing method, device, device, and medium based on federated learning to solve the problem of low training efficiency and accuracy when using data provided by multiple data parties to train models.

Images