察元AI教程库 · 文章 · 部署

MoSKA:共享KV注意力混合机制用于高效长序列LLM推理

大语言模型(LLMs)上下文长度的快速增长导致了Key-Value(KV)缓存的严重性能瓶颈,其内存密集特性导致GPU利用率显著下降。本文介绍了共享KV注意力的混合(MoSKA)架构,该架构解决了这一问题

作者:Myunghyun Rhee, Sookyung Choi, Euiseok Kim · 语言:en

前往原址学习

简介

大语言模型(LLMs)上下文长度的快速增长导致了Key-Value(KV)缓存的严重性能瓶颈,其内存密集特性导致GPU利用率显著下降。本文介绍了共享KV注意力的混合(MoSKA)架构,该架构解决了这一问题

返回教程库 · 察元AI产品家族