一种基于层级递减KV缓存压缩的级联投机推理方法及系统

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
一种基于层级递减KV缓存压缩的级联投机推理方法及系统
申请号:CN202510839998
申请日期:2025-06-23
公开号:CN120764673A
公开日期:2025-10-10
类型:发明专利
摘要
本发明公开了一种基于层级递减KV缓存压缩的级联投机推理方法及系统,该方法首先将上下文提示文本输入目标模型编码,生成KV缓存,计算出token之间的注意力分数。其次基于最后输入token的注意力分数降序排序,选择前k个注意力分数对应的KV缓存块作为级联中间层缓存,加载轻量级大语言模型作为草稿模型,并采用层级递减KV缓存压缩策略维护草稿模型缓存。然后基于所有缓存,构建双层级联投机推理架构,获取目标推理路径,并更新缓存。最后重复上述操作,直至按照目标推理路径输出上下文提示文本对应的目标响应数据。本发明降低KV缓存占比的同时提高全KV缓存的目标模型的草稿token接受率,减少精度的下降。
技术关键词
级联 中间层 推理方法 推理架构 大语言模型 注意力 层级 序列 滑动窗口 推理系统 文本 策略 输入模块 输出模块 数据 编码 表达式 阶段 精度