[发明专利]基于不确定性估计知识蒸馏的语言模型压缩方法在审
申请号: | 202210540113.5 | 申请日: | 2022-05-17 |
公开(公告)号: | CN114819148A | 公开(公告)日: | 2022-07-29 |
发明(设计)人: | 董伟生;黄天瑜;毋芳芳;石光明 | 申请(专利权)人: | 西安电子科技大学 |
主分类号: | G06N3/08 | 分类号: | G06N3/08;G06N3/04 |
代理公司: | 陕西电子工业专利中心 61205 | 代理人: | 王品华 |
地址: | 710071*** | 国省代码: | 陕西;61 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本发明公开了一种基于不确定性估计知识蒸馏的语言模型压缩方法,主要解决现有网络压缩技术存在的训练成本高、速度慢和知识蒸馏过程存在噪声干扰的问题。其实现方案是:1)对原始语言模型进行对半压缩得到压缩后的神经网络;2)利用原始语言模型合理初始化压缩后神经网络的参数;3)添加前馈网络结构的参数蒸馏损失函数,设计不确定性估计损失函数及自然语言处理任务的交叉熵损失函数;4)利用所设计的损失函数训练压缩后的神经网络模型。本发明降低了网络压缩训练过程的计算量,提高了网络压缩率,加速了网络推理速度,可广泛应用于模型部署、模型压缩任务,为硬件资源紧缺的应用场景提供新的模型压缩解决方案。 | ||
搜索关键词: | 基于 不确定性 估计 知识 蒸馏 语言 模型 压缩 方法 | ||
【主权项】:
暂无信息
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于西安电子科技大学,未经西安电子科技大学许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/202210540113.5/,转载请声明来源钻瓜专利网。