TensorFlow Similarity度量学习
TensorFlow Similarity 是一个用于相似性学习的 TensorFLow 库,也被称为度量学习和对比性学习。目前仍处于测试阶段。
Tensorflow Similarity 提供了先进的度量学习算法和所有必要的组件来研究、训练、评估和服务基于相似性的模型。
通过TensorFlow的相似性,你可以训练并提供模型,在大量的例子语料库中找到类似的项目(如图像)。例如,如上图所示,你可以训练一个相似性模型,从 Oxford IIIT 宠物数据集中寻找和聚类看起来相似的猫和狗的图像,只需在几个类上进行训练。要训练你自己的相似性模型,可参阅此 notebook。
度量学习不同于传统分类,因为它的目标不同。该模型学习以监督或自我监督的方式最小化相似示例之间的距离并最大化不同示例之间的距离。无论哪种方式,TensorFlow Similarity 都提供了必要的损失、指标、采样器、可视化器和索引子系统,以使其快速简便。
目前,TensorFlow Similarity 支持监督训练。在未来的版本中,它将支持半监督和自监督训练。
评论