买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
申请/专利权人:南京航空航天大学
摘要:本发明公开了一种基于中间层对齐的异构师生模型知识蒸馏方法,属于自然语言处理领域,提取教师模型与学生模型的中间层输出结果并进行中间层蒸馏对象的选择;设计中间层映射规则和维度转换模块以实现异构模型间基于中间层的知识对齐;根据不同的蒸馏部位设计相应的蒸馏损失函数以融合师生模型间各部位的损失值;最后调节各部分蒸馏的损失权重配比以达到蒸馏效果的优化与提升。本发明通过引入对师生模型中间层蒸馏结果的探索,克服了目前主流的生成式大语言模型蒸馏算法仅考虑到师生模型间的最后输出层损失的问题,提出了更普适于异构生成式大语言模型的中间层蒸馏算法,在多个相关数据集上的蒸馏效果均获得了显著提升。
主权项:1.一种基于中间层对齐的异构师生模型知识蒸馏方法,其特征在于,包括以下步骤:中间层对象选择:提取教师模型与学生模型的中间层输出结果,并进行中间层蒸馏对象的选择;中间层结果对齐:设计中间层映射规则和维度转换模块进行异构模型间的基于中间层的知识对齐;中间层损失计算:根据教师模型与学生模型不同的蒸馏部位设计相应的蒸馏损失函数,保证引入的中间层蒸馏损失值与标准知识蒸馏方法中的仅输出层的蒸馏损失、语言模型损失在数值量级上保持对齐;蒸馏损失融合优化:调节各部分蒸馏的损失权重配比后进行模型的蒸馏训练。
全文数据:
权利要求:
百度查询: 南京航空航天大学 基于中间层对齐的异构师生模型知识蒸馏方法
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。