Dynamic sparse attention
Окно заранее знает расстояние до токена, но ничего не знает о его смысле. Dynamic sparse attention (динамическое разрежённое внимание) добавляет дешёвый поиск: запрос сначала выбирает несколько релевантных блоков, и лишь затем внутри них вычисляется полноценный attention.
Позиция не говорит о релевантности
Локальное окно всегда читает последние блоки. Это надёжно для ближайшего контекста, но важное определение, число или условие может находиться гораздо раньше. Статический паттерн пропустит его независимо от содержания запроса.
Выберите один запрос и сначала найдите в документе нужный ему блок. Сравните последнее окно с динамическим выбором при том же запросе, затем смените вопрос. Какой способ меняет набор читаемых блоков вслед за содержанием запроса?
Переключайте вопросы к одному документу. Окно остаётся на месте, а dynamic selection меняет набор блоков вместе с запросом.
Сначала дешёвый поиск, потом дорогой attention
Каждый блок получает компактное представление. Indexer (индексатор) сравнивает запрос с этими представлениями и оставляет top-k кандидатов. Точные попарные оценки токенов вычисляются только внутри выбранного подмножества.
- Indexerбыстрые оценки блоков
- Top-kнесколько кандидатов
- Attentionточная работа с токенами
Indexer ранжирует
Поиск дополняют страховочными маршрутами
Один обучаемый indexer может ошибиться. Поэтому динамическую выборку полезно сочетать с маршрутами, которые сохраняют общий обзор и ближайший контекст.
- Сжатая ветвьГрубое сжатое представление всего префикса.
- Выбранная ветвьТочные токены из наиболее релевантных блоков.
- Локальное окноПоследние токены доступны без участия indexer.
Поиск тоже требует вычислений
Чтобы выбрать лучшие блоки, indexer должен хотя бы грубо оценить весь префикс. Если он слишком тяжёлый, экономия основного attention растворяется в стоимости маршрутизации.
- Размер блокаменьше блок — точнее выбор; больше кандидатов для indexer
- Значение kбольше k — выше полнота (recall); дороже основной attention
- Качество indexerточнее поиск дальних фактов; сложнее обучение и вывод