使用机器学习模型来对图像进行分类

摘要

用于使用机器学习模型来对图像进行分类的系统和方法。所述方法中的一种包括：获得用于训练所述机器学习模型的训练数据，其中，所述机器学习模型被配置成处理输入图像以针对每个输入图像生成嵌入空间中的预测点；从针对所述训练数据中的训练图像的标签数据，确定所述对象类别中的每一个对象类别的相应的数字嵌入，其中，任何两个对象类别的数字嵌入之间的所述嵌入空间中的距离反映所述两个对象类别的视觉共现的程度；以及在所述训练数据上训练所述机器学习模型。本说明书中描述的所述系统可有效地执行多标签、大规模多类别图像分类，其中类的数目大(数千或数万)并且其中每个图像通常属于应该全部被适当地识别的多个类别。