Title:
METHOD AND APPARATUS FOR TRAINING MACHINE LEARNING MODEL, DEVICE, AND MEDIUM
Document Type and Number:
WIPO Patent Application WO/2024/093573
Kind Code:
A1
Abstract:
Provided are a method and apparatus for training a machine learning model, a device, and a medium. A machine learning model comprises a first sub-model and a second sub-model, the first sub-model being located at a first computing node in a computing system, and the second sub-model being located at a second computing node in the computing system. The method comprises: in a training phase for training the machine learning model, receiving at the first computing node a first set of training data for training the machine learning model; receiving the second sub-model from the second computing node; separately inputting into the first sub-model and the acquired second sub-model the first set of training data to determine a first update parameter for updating the first sub-model and a second update parameter for updating the second sub-model; and transmitting to the second computing node the second update parameter. In this way, the sub-models to be used can be acquired in advance, and the data transmission amount during the training process can be reduced.
Inventors:
JIANG YIMIN (CN)
LIU JUNCAI (CN)
ZHU YIBO (US)
LIU JUNCAI (CN)
ZHU YIBO (US)
Application Number:
PCT/CN2023/120501
Publication Date:
May 10, 2024
Filing Date:
September 21, 2023
Export Citation:
Assignee:
DOUYIN VISION CO LTD (CN)
LEMON INC (GB)
LEMON INC (GB)
International Classes:
G06N20/00
Attorney, Agent or Firm:
SHIHUI PARTNERS (Tower C Beijing Yintai Centre, No.2 Jianguomenwai Avenue,Chaoyang District, Beijing 2, CN)
Download PDF: