察元AI教程库 · 文章 · 部署
大语言模型(LLMs)上下文长度的快速增长导致了Key-Value(KV)缓存的严重性能瓶颈,其内存密集特性导致GPU利用率显著下降。本文介绍了共享KV注意力的混合(MoSKA)架构,该架构解决了这一问题
作者:Myunghyun Rhee, Sookyung Choi, Euiseok Kim · 语言:en
大语言模型(LLMs)上下文长度的快速增长导致了Key-Value(KV)缓存的严重性能瓶颈,其内存密集特性导致GPU利用率显著下降。本文介绍了共享KV注意力的混合(MoSKA)架构,该架构解决了这一问题