训练关键词识别模型的方法、识别关键词的方法及装置


专利名称

训练关键词识别模型的方法、识别关键词的方法及装置

所属单位

智能科学与技术学院

通讯地址

北京市西城区展览路1号

联系人

科学技术发展研究院

电话

010-68322482

专利发明人

周若华

专利号

CN202410742658.3

专利类型

发明授权

专利状态

授权且有效

专利简介

本申请提供一种训练关键词识别模型的方法、识别关键词的方法及装置,该方法包括,通过预设的标签数据在全部基类上对基础全分类模型进行训练,得到分类器和编码器,其中,标签数据包括多个带有关键词标签的文本;结合所述编码器对分类器进行训练,得到全分类模型;在元训练阶段运用元学习的方法训练模型时,确定支持样本的类别质心,并结合余弦相似度方法预测查询样本属于类别质心对应类别的相似概率;通过相似概率计算训练任务的损失,并根据损失调整全分类模型的参数,得到少样本关键词识别模型。过该方法可以在少量训练数据的情况下,准确的训练关键词识别模型并通过少样本关键词识别模型识别语音关键词的效果。
Baidu
map