Apple Inc.
Model and ensemble compression for metric learning

Abstract:

Systems and processes for metric learning distillation are disclosed herein. In accordance with one example, a method includes, at an electronic device, at an electronic device having one or more processors and memory, receiving a first plurality of vectors from a first model, receiving a second plurality of vectors from a second model, determining a first plurality of vector distances based on the first plurality of vectors, generating a first matrix based on the first plurality of vector distances, determining a second plurality of vector distances based on the second plurality of vectors, generating a second matrix based on the second plurality of vector distances, comparing the first matrix with the second matrix, and adjusting the second model based on the comparison of the first matrix and the second matrix.