-
公开(公告)号:US20240303506A1
公开(公告)日:2024-09-12
申请号:US18598123
申请日:2024-03-07
发明人: Ziheng WANG , Conor PERREAULT , Xi LIU , Anthony M. JARC
IPC分类号: G06N3/098 , G06N3/0895
CPC分类号: G06N3/098 , G06N3/0895
摘要: One embodiment of the present invention sets forth a technique for training a machine learning model to perform feature extraction. The technique includes executing a student version of the machine learning model to generate a first set of features from a first set of image crops and executing a teacher version of the machine learning model to generate a second set of features from a second set of image crops. The technique also includes training the student version of the machine learning model based on one or more losses computed between the first and second sets of features. The technique further includes transmitting the trained student version of the machine learning model to a server, wherein the trained student version can be aggregated by the server with additional trained student versions of the machine learning model to generate a global version of the machine learning model.