FEATURE EXTRACTION VIA FEDERATED SELF-SUPERVISED LEARNING

    公开(公告)号:US20240303506A1

    公开(公告)日:2024-09-12

    申请号:US18598123

    申请日:2024-03-07

    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.