search

tensorflow / similarity

TensorFlow Similarity is a python package focused on making similarity learning quick and easy.
Apache License 2.0
1.01k stars 104 forks source link

Look into adding soft KNN for monitoring during training #212

Open owenvallis opened 2 years ago

owenvallis commented 2 years ago

https://arxiv.org/abs/1902.01889

owenvallis commented 2 years ago

"As an analytical tool, it provides insights into the evolution of class similarity structures during learning"

igor-susic commented 2 years ago

Hi @owenvallis if this one is still for grabs I would take some time to implement it.

innat commented 2 years ago

ref. https://github.com/gaudelbijay/SNNL-Loss/blob/main/soft_nearest_neighbor_loss.py