ДалееСледующее руководство
Внимание к множественным запросам
Языковой ИИ
РУКОВОДСТВО ПО ЯЗЫКУ ИИ
Grouped-Query Attention (GQA) shares key/value heads across groups of query heads.
Прослушайте это руководствоОзвучено ИИ · ~1 min
The original GQA paper studies this as a tradeoff between multi-head and multi-query attention for decoder inference.
The GQA paper proposes grouping query heads so multiple query heads share key/value heads. MHA has one key/value head per query head; MQA shares a single key/value head across all queries; GQA lies between them. The paper evaluates GQA on language modeling and machine translation tasks and reports quality closer to MHA with inference speed closer to MQA in its experiments. It also reports uptraining existing MHA models using about 5% of original pretraining compute. These claims are limited to the paper’s models, tasks and setup.
Comparing MHA, GQA and MQA head sharing.
Estimating KV-cache size from the number of query heads and key/value groups.
Reading the paper’s reported quality/speed tradeoff on evaluated language-model tasks.
Reviewing the paper’s uptraining recipe for converting an MHA checkpoint to GQA.
Benchmark head-sharing choices on the target model and hardware. The original paper’s quality/speed results do not establish universal serving speedups or negligible quality loss across all systems.
Бесплатная рассылка
Три проверенных AI-истории каждое утро буднего дня, написанные простым английским. Бесплатно навсегда, без рекламы.
Одно электронное письмо каждый будний день. Отписаться в один клик. Мы никогда не продаем и не передаем ваш адрес.
Проверьте себя
Мгновенная обратная связь по каждому ответу и общий сертификат с проверяемым идентификатором после прохождения курса.
Поддержите бесплатное образование в области искусственного интеллекта. AI Understanding — это некоммерческая организация 501(c)(3). Никакой рекламы и платного доступа. Сделать пожертвование
Grouped-Query Attention (GQA) shares key/value heads across groups of query heads. The original GQA paper studies this as a tradeoff between multi-head and multi-query attention for decoder inference.
GQA хранит отдельные заголовки запросов, но позволяет каждой группе из них использовать один набор заголовков ключей и значений, сокращая кеш KV.
Мультиголовка дает каждой головке свой КВ; мультизапрос разделяет один КВ на все головки; GQA находится между несколькими группами КВ.
Экономия проявляется во время генерации, когда чтение KV-кэша является основной затратой на пропускную способность памяти.
32 головки запросов, разделенные на 8 общих групп KV, дают примерно четырехкратное сокращение кэшируемых ключей и значений.
The GQA paper reports quality closer to multi-head attention (MHA) with inference speed closer to multi-query attention (MQA) in its evaluated tasks. This is an empirical result, not a universal guarantee.
Продолжайте учиться
Другие руководства, выбранные по этой теме
ДалееСледующее руководство
Внимание к множественным запросам
Языковой ИИ