CNIPA.AI
검색으로 돌아가기
기록

一种基于静态与动态词表协同的投机解码方法及系统

발명유효
1조회수
8청구항 · 2 독립항
§ Ⅰ

개요

발명자

余祥; 李石坚; 张犁; 潘纲

IPC 분류

G06N 5/04 (2023.01)G06F 40/279 (2020.01)

本发明公开了一种基于静态与动态词表协同的投机解码方法及系统,包括:初始化静态词表与动态词表;基于推理请求,草稿模型在词表上生成草稿;利用目标模型在其完整词表上对草稿验证,输出草稿各位置在目标模型完整词表上的概率分布;利用概率分布,从每个草稿位置各自选取概率最高的前几个词组成候选词集合,移除已包含于的词,得到剩余候选词集合;利用剩余候选词集合对及其维护数据进行更新;利用概率分布确定验证通过的令牌,将其追加到输出序列;重复上述步骤,直到生成序列终止令牌,得到推理结果。本发明可有效提升词表裁剪范式下的草稿质量,提高投机解码的推理加速性能。