龙图腾网&IPTOP
- 微信扫码登录
- 账号密码登录
- 短信登录/注册
买专利卖专利找龙图腾，真高效！ 查专利查商标用IPTOP,全免费！专利年费监控用IP管家,真方便！
申请/专利权人：上海壁仞科技股份有限公司
摘要：本申请实施例提供了一种注意力机制算子的优化方法、设备、存储介质及程序产品，涉及人工智能技术领域，该方法包括：在模型推理过程中，采用第一量化缩放因子对第一键表示进行在线量化获得第二键表示；以及，采用第二量化缩放因子对第一值表示进行在线量化获得第二值表示。然后将第二键表示和第一量化缩放因子保存在显存中的键信息集合中，以及，将第二值表示和第二量化缩放因子保存在显存中的值信息集合中。再基于第一查询表示、显存中的键信息集合和值信息集合进行自注意力机制计算，获取输出张量。这样，在大语言模型推理过程中，显存只需要保存量化后的键值数据，大大减少了键值缓存带来的额外显存占用，也减少注意力机制推理的显存占用。
主权项：1.一种注意力机制算子的优化方法，其特征在于，包括：将输入张量映射成第一键表示、第一值表示和第一查询表示；采用第一量化缩放因子对所述第一键表示进行量化，获得第二键表示；以及，采用第二量化缩放因子对所述第一值表示进行量化，获得第二值表示；将所述第二键表示和所述第一量化缩放因子保存在显存中的键信息集合中，以及，将所述第二值表示和所述第二量化缩放因子保存在显存中的值信息集合中；基于所述第一查询表示、所述键信息集合和所述值信息集合执行自注意力机制计算，获取输出张量。
全文数据：
权利要求：
百度查询： 上海壁仞科技股份有限公司 注意力机制算子的优化方法、设备、存储介质及程序产品
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息，力求客观、公正，但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解，仅供参考使用，不能作为本公司承担任何法律责任的依据或者凭证。
主营中国专利交易买卖与专利转让许可,高校科技成果推广与科技成果转化,知识产权运营管理与平台开发,科技人才专家库与科技猎头等科技服务
开放平台
担保交易
惠及多方
会员特惠
专属平台
适合多方
数据共享
功能齐全
皖公网安备 34010402703815号