Joint training method and device for machine learning model, equipment and storage medium

Abstract

The invention provides a joint training method and device for a machine learning model, an electronic device and a computer readable storage medium, and belongs to the technical field of computers. The method comprises the following steps: a data node acquiring sample data for training; the data node processing the sample data through the first part of the target model to obtain intermediate data,and sending the intermediate data to the label node; the label node processing the intermediate data through a second part of the target model and obtaining error data based on label data corresponding to the sample data; the label node sending the error data to the data node; the data node adjusting the parameters of the first part according to the error data, and / or the label node adjusting theparameters of the second part according to the error data; wherein the target model is composed of a first part and a second part. According to the invention, the data between the data node and the label node is opaque in the joint training process, and the privacy and security of the data are improved.

Description

technical field [0001] The present disclosure relates to the field of computer technology, and in particular, to a joint training method for machine learning models, a joint training device for machine learning models, electronic equipment, and a computer-readable storage medium. Background technique [0002] As the core technology in the field of artificial intelligence, machine learning largely determines the degree of development of artificial intelligence. At present, with the combination of machine learning and big data, the machine learning model is becoming more and more complex, and the amount of data and calculation required for training the model is greatly increased. In this case, there is a method of joint training by multiple servers or terminals. . However, the existing joint training methods generally have security risks of data leakage, which limits their application in machine learning techniques. [0003] It should be noted that the information disclosed ...

AI Technical Summary

Problems solved by technology

[0004] The present disclosure provides a joint training method of a machine learning model, a joint training device of a machine learning model, an electronic device, and a computer-readable storage medium, thereby at least to a certain extent overcoming the problem of potential safety hazards in the existing joint training method

Images