System.ArgumentOutOfRangeException: 索引和长度必须引用该字符串内的位置。 参数名: length 在 System.String.Substring(Int32 startIndex, Int32 length) 在 zhuanliShow.Bind() 基于知识蒸馏的人机交互方法、设备及存储介质技术_技高网

基于知识蒸馏的人机交互方法、设备及存储介质技术

技术编号:42541299 阅读:4 留言:0更新日期:2024-08-27 19:45
本发明专利技术涉及数据处理领域,公开了一种基于知识蒸馏的人机交互方法、设备及存储介质。该方法包括:在预先训练的教师模型输出的软标签集合中提取预设数量的软标签,得到知识信息;根据预设的质量标准参数,对知识信息执行筛选操作,得到目标知识信息;根据目标知识信息,对预设的学生模型执行蒸馏训练,并判断软标签集合是否还存在待提取软标签;若软标签集合不存在待提取软标签,将学生模型作为目标学生模型;检测到人机交互请求时,调用教师模型和目标学生模型处理人机交互请求。在本发明专利技术实施例中,减轻了对大型模型的依赖,从而降低了人机交互所需的硬件资源的压力。

【技术实现步骤摘要】

本专利技术涉及数据处理领域,尤其涉及一种基于知识蒸馏的人机交互方法、设备及存储介质


技术介绍

1、知识蒸馏是一种模型压缩技术,旨在将一个复杂模型(教师模型)学到的知识迁移到一个轻量级模型(学生模型)上,以实现模型压缩和加速。

2、当前的人机对话场景中,通常采用生成式大语言模型(gllms)作为对话生成的核心,以实现与用户的多轮对话。在专业领域的对话场景中,为解决大模型对专业知识的缺失问题,通常会构建结构化的专业知识库。整个对话系统流程中,会设计多次调用生成式大模型和知识库检索的流程。然而,大型语言模型通常包含数十亿甚至数百亿个参数,这使得它们的模型文件大小高达几十gb。加载和存储这些庞大的模型需要大量内存和存储资源,给人机交互所需的硬件环境带来了很大压力。需要一种新的技术手段解决上述技术问题。


技术实现思路

1、本专利技术的主要目的在于解决人机交互所需的硬件成本高的技术问题。

2、本专利技术第一方面提供了一种基于知识蒸馏的人机交互方法,所述基于知识蒸馏的人机交互方法包括:

3、在预先训练的教师模型输出的软标签集合中提取预设数量的软标签,得到知识信息;

4、根据预设的质量标准参数,对所述知识信息执行筛选操作,得到目标知识信息;

5、根据所述目标知识信息,对预设的学生模型执行蒸馏训练,并判断所述软标签集合是否还存在待提取软标签;

6、若所述软标签集合不存在所述待提取软标签,将所述学生模型作为目标学生模型;

7、检测到人机交互请求时,调用所述教师模型和所述目标学生模型处理所述人机交互请求。

8、可选的,在本专利技术第一方面的第一种实现方式中,所述判断所述软标签集合是否还存在待提取软标签的步骤之后,所述方法还包括:

9、若所述软标签集合存在所述待提取软标签,返回执行在预先训练的教师模型输出的软标签集合中提取预设数量的软标签,得到知识信息的步骤。

10、可选的,在本专利技术第一方面的第二种实现方式中,所述根据所述目标知识信息,对预设的学生模型执行蒸馏训练,并判断所述软标签集合是否还存在待提取软标签的步骤包括:

11、判断所述目标知识信息是否为空;

12、若所述目标知识信息不为空,根据所述目标知识信息,对预设的学生模型执行蒸馏训练,并判断所述软标签集合是否还存在待提取软标签;

13、若所述目标知识信息为空,直接执行判断所述软标签集合是否还存在待提取软标签的步骤。

14、可选的,在本专利技术第一方面的第三种实现方式中,所述根据预设的质量标准参数,对所述知识信息执行筛选操作,得到目标知识信息的步骤之前,所述方法还包括:

15、对准确度、置信度、相关性、多样性、鲁棒性、泛化能力和/或计算效率执行数值设定操作,得到所述质量标准参数。

16、可选的,在本专利技术第一方面的第四种实现方式中,所述根据所述目标知识信息,对预设的学生模型执行蒸馏训练,并判断所述软标签集合是否还存在待提取软标签的步骤包括:

17、将所述目标知识信息作为监督信号,更新所述学生模型的参数,并定期获取所述学生模型的性能参数;

18、若所述性能参数满足收敛条件,判断所述软标签集合是否还存在待提取软标签。

19、可选的,在本专利技术第一方面的第五种实现方式中,所述定期获取所述学生模型的性能参数的步骤包括:

20、获取所述学生模型对应的准确度、召回率、f1分数、损失函数值,以得到所述学生模型的性能参数。

21、可选的,在本专利技术第一方面的第六种实现方式中,所述若所述软标签集合不存在所述待提取软标签,将所述学生模型作为目标学生模型的步骤包括:

22、若所述软标签集合不存在所述待提取软标签,对所述学生模型执行模型压缩操作,得到目标学生模型。

23、可选的,在本专利技术第一方面的第七种实现方式中,所述在预先训练的教师模型输出的软标签集合中提取预设数量的软标签,得到知识信息的步骤包括:

24、将预设的输入数据输入所述教师模型;

25、接收所述师模型输出的概率分布信息,得到软标签集合。

26、本专利技术第二方面提供了一种基于知识蒸馏的人机交互设备,包括:存储器和至少一个处理器,所述存储器中存储有指令,所述存储器和所述至少一个处理器通过线路互连;所述至少一个处理器调用所述存储器中的所述指令,以使得所述基于知识蒸馏的人机交互设备执行上述的基于知识蒸馏的人机交互方法。

27、本专利技术的第三方面提供了一种计算机可读存储介质,所述计算机可读存储介质中存储有指令,当其在计算机上运行时,使得计算机执行上述的基于知识蒸馏的人机交互方法。

28、在本专利技术实施例中,将教师模型学到的知识信息提取出来,根据预设的质量标准参数,对提取的知识信息进行筛选操作,从而得到目标知识信息,该步骤确保了所提取的知识信息质量高,对学生模型的训练有利。基于目标知识信息,对预设的学生模型执行蒸馏训练。可将教师模型的知识迁移至学生模型,使学生模型能够以更轻量级的形式表现出教师模型的知识。在此过程中,系统会判断软标签集合是否还存在待提取软标签,以确定是否需要进一步的蒸馏训练。如果软标签集合不存在待提取软标签,即学生模型已经从教师模型中获取了足够的知识,那么将学生模型作为目标学生模型。当系统检测到人机交互请求时,会同时调用教师模型和目标学生模型来处理该请求,可提供更加全面和准确的响应,同时减轻了对大型模型的依赖,从而降低了人机交互所需的硬件资源的压力。

本文档来自技高网
...

【技术保护点】

1.一种基于知识蒸馏的人机交互方法,其特征在于,所述基于知识蒸馏的人机交互方法包括:

2.根据权利要求1所述的基于知识蒸馏的人机交互方法,其特征在于,所述判断所述软标签集合是否还存在待提取软标签的步骤之后,所述方法还包括:

3.根据权利要求1所述的基于知识蒸馏的人机交互方法,其特征在于,所述根据所述目标知识信息,对预设的学生模型执行蒸馏训练,并判断所述软标签集合是否还存在待提取软标签的步骤包括:

4.根据权利要求1所述的基于知识蒸馏的人机交互方法,其特征在于,所述根据预设的质量标准参数,对所述知识信息执行筛选操作,得到目标知识信息的步骤之前,所述方法还包括:

5.根据权利要求1所述的基于知识蒸馏的人机交互方法,其特征在于,所述根据所述目标知识信息,对预设的学生模型执行蒸馏训练,并判断所述软标签集合是否还存在待提取软标签的步骤包括:

6.根据权利要求5所述的基于知识蒸馏的人机交互方法,其特征在于,所述定期获取所述学生模型的性能参数的步骤包括:

7.根据权利要求1所述的基于知识蒸馏的人机交互方法,其特征在于,所述若所述软标签集合不存在所述待提取软标签,将所述学生模型作为目标学生模型的步骤包括:

8.根据权利要求1所述的基于知识蒸馏的人机交互方法,其特征在于,所述在预先训练的教师模型输出的软标签集合中提取预设数量的软标签,得到知识信息的步骤包括:

9.一种基于知识蒸馏的人机交互设备,其特征在于,所述基于知识蒸馏的人机交互设备包括:存储器和至少一个处理器,所述存储器中存储有指令,所述存储器和所述至少一个处理器通过线路互连;

10.一种计算机可读存储介质,所述计算机可读存储介质上存储有计算机程序,其特征在于,所述计算机程序被处理器执行时实现如权利要求1-8中任一项所述的基于知识蒸馏的人机交互方法。

...

【技术特征摘要】

1.一种基于知识蒸馏的人机交互方法,其特征在于,所述基于知识蒸馏的人机交互方法包括:

2.根据权利要求1所述的基于知识蒸馏的人机交互方法,其特征在于,所述判断所述软标签集合是否还存在待提取软标签的步骤之后,所述方法还包括:

3.根据权利要求1所述的基于知识蒸馏的人机交互方法,其特征在于,所述根据所述目标知识信息,对预设的学生模型执行蒸馏训练,并判断所述软标签集合是否还存在待提取软标签的步骤包括:

4.根据权利要求1所述的基于知识蒸馏的人机交互方法,其特征在于,所述根据预设的质量标准参数,对所述知识信息执行筛选操作,得到目标知识信息的步骤之前,所述方法还包括:

5.根据权利要求1所述的基于知识蒸馏的人机交互方法,其特征在于,所述根据所述目标知识信息,对预设的学生模型执行蒸馏训练,并判断所述软标签集合是否还存在待提取软标签的步骤包括:

6.根据...

【专利技术属性】
技术研发人员:曾昊罗傲文博刘云峰朱明仁孟家邑
申请(专利权)人:深圳追一科技有限公司
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1