Винни-Пух в 768 измерениях: семантический поиск для Codex Pets на YDB
Разработчик создал семантический поиск по каталогу Codex Pets с использованием эмбеддингов, хранящихся в YDB. Система объединяет лексическое, текстовое семантическое и визуальное семантическое ранжирование. Тесты показывают улучшение метрик NDCG и Recall по сравнению с чисто лексическим поиском.
Каталог Codex Pets содержит 153 одобренных анимированных питомца, которые сопровождают ИИ-агентов. По мере роста каталога поиск на основе тегов стал непрактичным, не находя питомцев, описанных по внешнему виду, а не по точным тегам. Автор внедрил семантический поиск: текстовые поля (имя, описание, теги) кодируются в векторы размерностью 768, а мультимодальная модель генерирует подписи для четырех
Показать ещё ↓
Источник: Habr — хаб ML —
оригинал