Unlock instant, AI-driven research and patent intelligence for your innovation.

Training method and device for federated learning model

A technology of learning models and training methods, applied in the field of financial technology, can solve problems such as low model accuracy and poor results

Pending Publication Date: 2020-09-29
WEBANK (CHINA) +1
View PDF0 Cites 11 Cited by
  • Summary
  • Abstract
  • Description
  • Claims
  • Application Information

AI Technical Summary

Problems solved by technology

[0004] However, in the federated learning in the existing technology, the accuracy of the model trained using non-independent and identically distributed data is low, and the effect is not good.

Method used

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
View more

Image

Smart Image Click on the blue labels to locate them in the text.
Viewing Examples
Smart Image
  • Training method and device for federated learning model
  • Training method and device for federated learning model
  • Training method and device for federated learning model

Examples

Experimental program
Comparison scheme
Effect test

example 1

[0086] Initialize non-independent and identically distributed data as the training set for model training, including the following two methods.

[0087] 1. Sort the data according to the number tags, and then divide the data into multiple parts, such as ten parts. Each client will hold data with multiple digital tags, such as two types. For example, after dividing the data into ten parts, perform labeling Sorting, 1-10, then take 1 and 8 for model training on the first client, take 9 and 7 for model training on the second client, so that the data of each client cannot be used as a representative of the global data distribution .

[0088] 2. Using the benchmark data set, the data is divided into ten parts, so that the amount of data on each client varies greatly, so that the data of each client cannot be used as a representative of the global data distribution.

[0089] Select 10 clients to use the above data for model training, and 10 clients obtain the k-1th global model par...

example 2

[0105] Obtain the local model parameters of the kth iteration sent by 10 clients, and then sum the differences between the local model parameters of the kth iteration of all clients and the global model parameters of the k-1th iteration to obtain the sum for: Then multiply the sum and the learning rate, and then add it to the global model parameters of the k-1th iteration, and the global model parameters of the kth iteration are obtained as:

[0106] Step 303, the server broadcasts the global model parameters of the kth iteration to the multiple clients, so that the multiple clients can perform k+1th iteration training.

[0107] In the embodiment of the present invention, the server broadcasts the global model parameters of the kth iteration to multiple clients, and the client does not need to reset the regularization constraints, and directly enables multiple clients to perform the k+1th iteration training, and obtains The local training model parameters of the k+1 iter...

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
Login to View More

PUM

No PUM Login to View More

Abstract

The invention discloses a training method and device for a federated learning model. The method comprises the steps of: enabling a client to acquire global model parameters of (k-1)th iteration broadcasted by a server, wherein k is a positive integer; taking the global model parameters as local model parameters with regularization constraints; carrying out kth iterative training by using local data; obtaining a local model parameter of the kth iterative training, wherein the regularization constraints are determined according to a global model parameter of the server and the local model parameter of the client; and optimizing the gradient in the model through regularization constraint, so that the influence of extreme data on local model parameter training is reduced, the accuracy of training the non-independent same-distribution data by the local model parameters is improved, and then the local model parameters subjected to the kth iteration training are sent to the server, so that the server updates the global model parameters subjected to the kth iteration training, and the accuracy of training the non-independent same-distribution data by the global model parameters is improved.

Description

technical field [0001] The present invention relates to the field of financial technology (Fintech), in particular to a training method and device for a federated learning model. Background technique [0002] With the development of computer technology, more and more technologies (such as blockchain, cloud computing or big data) are applied in the financial field. The traditional financial industry is gradually transforming into financial technology, and big data technology is no exception. However, due to The security and real-time requirements of the financial and payment industries also put forward higher requirements for big data technology. [0003] The federated learning of the prior art communicates between nodes by transmitting parameters, and integrates the information provided by the data of each node by averaging the parameters during the training process. During the learning process of federated learning, it is necessary to train the data, randomly select a node...

Claims

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
Login to View More

Application Information

Patent Timeline
no application Login to View More
Patent Type & Authority Applications(China)
IPC IPC(8): G06N20/20
CPCG06N20/20
Inventor 刘楠王玥琪李晓丽陈川郑子彬严强李辉忠
Owner WEBANK (CHINA)