Недавняя статья (Вот) ввел вторичную функцию потерь, которую они назвали центром потерь. Он основан на расстоянии между вложениями в пакет и скользящим средним вложением для каждого из соответствующих классов. Там было некоторое обсуждение в группах TF Google (Вот) относительно того, как такие центры внедрения могут быть вычислены и обновлены. Я собрал некоторый код для генерации вложений среднего класса в своем ответе ниже.

Это лучший способ сделать это?