Angular Dispersion Accelerates $k$-Nearest Neighbors Machine Translation
2509.16729v1
cs.CL, cs.LG
2025-09-24
Авторы:
Evgeniia Tokarchuk, Sergey Troshin, Vlad Niculae
Резюме на русском
#### Контекст
Angular Dispersion Accelerates $k$-Nearest Neighbors Machine Translation (k-NN MT) — это исследование в области машинного перевода, нацеленное на улучшение производительности $k$-NN MT. Данный подход к переводу использует внешнюю память, где хранятся представления языковых контекстов в виде векторов. Несмотря на его высокую эффективность, $k$-NN MT страдает от высокой сложности вычислений и большими потреблением памяти, что ограничивает его применение в реальном времени. Основной проблемой является то, что необходимость выполнять быстрый поиск в больших многомерных пространствах, что приводит к узким местам в производительности. Поэтому целью данного исследования является развитие методов, которые улучшают производительность $k$-NN MT, уменьшая или устраняя зависимость от таких ограничений.
#### Метод
Работа сосредотачивается на изменении свойств внешней памяти $k$-NN MT, а именно, на повышении "angular dispersion" — разброса углов между векторами внешней памяти. Основная идея заключается в том, что распределение токенов по угловым меркам повышает баланс между различными векторами, что в свою очередь ускоряет поиск и улучшает результаты перевода. Алгоритм включает в себя техники, нацеленные на изменение функциональности $k$-NN MT, в частности, увеличивая различия в углах между контекстными векторами. Это достигается за счет изменения характера обучения моделей перевода, чтобы повысить разреженность внешней памяти.
#### Результаты
Использовались данные из различных машинно-переводных задач, включая различные языковые пары и текстовые корпусы. Были проведены эксперименты с различными модификациями $k$-NN MT, включая изменение размера внешней памяти и других параметров. Наблюдалось улучшение в скорости поиска во внешней памяти при увеличении разброса углов между векторами. Также были подтверждены положительные эффекты на качество перевода, которые были связаны с улучшением распределения контекстных векторов.
#### Значимость
Результаты данного исследования имеют большое значение для развития $k$-NN MT, так как позволяют увеличить скорость работы и улучшить качество перевода, не прибегая к сложностям, связанным с уменьшением размера внешней памяти или увеличением количества токенов. Данный подход может быть применен в различных областях нотронного перевода, где необходима высокая скорость работы и качество перевода.
#### Выводы
В ходе работы были доказаны преимущества повышения angular dispersion внешней памяти в $k$-NN MT. Было показано, что данный подход может быть использован для улучшения производительности без ущерба качеству перевода. Будущими направлениями исследований являются расширение применения данной техники к различным моде
Abstract
Augmenting neural machine translation with external memory at decoding time,
in the form of k-nearest neighbors machine translation ($k$-NN MT), is a
well-established strategy for increasing translation performance. $k$-NN MT
retrieves a set of tokens that occurred in the most similar contexts recorded
in a prepared data store, using hidden state representations of translation
contexts as vector lookup keys. One of the main disadvantages of this method is
the high computational cost and memory requirements. Since an exhaustive search
is not feasible in large data stores, practitioners commonly use approximate
$k$-NN MT lookup, yet even such algorithms are a bottleneck. In contrast to
research directions seeking to accelerate $k$-NN MT by reducing data store size
or the number of lookup calls, we pursue an orthogonal direction based on the
performance properties of approximate $k$-NN MT lookup data structures. In
particular, we propose to encourage angular dispersion of the neural hidden
representations of contexts. We show that improving dispersion leads to better
balance in the retrieval data structures, accelerating retrieval and slightly
improving translations.
Ссылки и действия
Дополнительные ресурсы: