METHOD, APPARATUS, DEVICE AND READABLE MEDIUM FOR TRANSFER LEARNING IN MACHINE LEARNING

    公开(公告)号:US20210065058A1

    公开(公告)日:2021-03-04

    申请号:US16998616

    申请日:2020-08-20

    Abstract: A method, apparatus, device and readable medium for transfer learning in machine learning are provided. The method includes: constructing a target model according to the number of classes to be achieved by a target task and a duly-trained source model; obtaining a value of a regularized loss function of the corresponding target model and a value of a cross-entropy loss function of the target model, based on sets of training data in a training dataset of the target task; according to the value of the regularized loss function and the value of the cross-entropy loss function corresponding to each set of training data, updating parameters in the target model by a gradient descent method to implement the training of the target model. The above technical solution avoids excessive constraints on parameters in the prior art, thereby refraining from damaging the training effect of the source model on the target task.

Patent Agency Ranking