Training method, apparatus, and device for federated neural network model, computer program product, and computer-readable storage medium

Abstract

Embodiments of this application provide a training method, apparatus, and device for a federated neural network model, a computer program product, and a computer-readable storage medium. The method includes: performing forward computation processing of sample data through a first model, to obtain an output vector of the first model; performing forward computation processing through the interaction layer of the output vector of the first bottom model, at last one model parameter of an interaction layer, and at least one encrypted model to obtain an output vector of the interaction layer; performing forward computation processing on the output vector of the interaction layer through the second model, to obtain an output vector of the federated neural network model; back propagating the federated neural network model according to a loss result corresponding to the output vector of the federated neural network model, and updating the at least one model parameter of the interaction layer, a parameter of the first bottom model, and a parameter of the top model according to a back propagation processing result; and obtaining a trained federated neural network model based on the updated model parameters of the interaction layer, the first bottom model, and the second model.