Как реализовать пользовательский поиск луча в TensorFlow?
В оригинальной статье Google SmartReply:
Наш поиск ведется следующим образом. Во-первых, элементы R (множество всех возможных ответов) организованы в три. Затем мы проводим поиск луча слева направо, но сохраняем только те гипотезы, которые появляются в дереве. Этот процесс поиска имеет сложность O(bl) для размера луча b и максимальной длины отклика l. Как b, так и l обычно находятся в диапазоне 10-30, поэтому этот метод значительно сокращает время нахождения ответов верхнего уровня и является критическим элементом обеспечения возможности развертывания этой системы.
Я хочу реализовать нечто подобное. Некоторые вопросы:
- Как представить набор всех возможных ответов (R) в структуре данных Trie в TF?
- Как изменить текущий поиск луча, чтобы сохранить только те гипотезы, которые доступны в R?