买专利卖专利找龙图腾，真高效！ 查专利查商标用IPTOP,全免费！专利年费监控用IP管家,真方便！

申请/专利权人：贵州师范大学

摘要：本申请的实施例提供一种基于度量学习的小样本命名实体识别方法。应用于人工智能技术领域，包括：获取语料数据分为训练集和测试集，根据训练集构建实体类型到实体的映射；将训练集划分为支持集和查询集，并分别构建实体引导掩码提示，得到支持集句子输入增强序列和查询集句子输入增强序列，并采用预训练语言模型进行编码处理，得到支持集token编码和查询集token编码；计算两种编码的相似度，并构造相似度矩阵得到相似度分数，根据相似度分数计算损失函数，得到训练后的预训练语言模型；将测试集划分为支持集和查询集，输入到预训练语言模型中进行处理，进行实体类型预测。以此方式，本申请能够有效提升在小样本条件下文本分类的准确性及可靠性。

主权项：1.一种基于度量学习的小样本命名实体识别方法，其特征在于，包括以下步骤：获取语料数据，构建语料数据训练集和测试集，根据所述训练集构建实体类型到实体的映射；将所述训练集划分为训练支持集和训练查询集，基于所述映射分别在所述训练支持集和所述训练查询集上构建实体引导掩码提示，得到训练支持集增强序列和训练查询集增强序列；采用预训练语言模型对构建有实体引导掩码提示的所述训练支持集增强序列及所述训练查询集增强序列进行编码处理，得到训练支持集token编码和训练查询集token编码；其中，所述训练支持集token编码和所述训练查询集token编码的获取过程为：将所述训练支持集实体类型整合提示增强序列及所述训练支持集类型实例指导提示增强序列输入到所述模型中进行处理，得到训练支持集输出结果，采用掩码向量对所述训练支持集输出结果进行编码并进行加权处理得到训练支持集token编码；将所述训练查询集增强序列输入到所述模型中进行处理，得到训练查询集输出结果，采用掩码向量对所述训练查询集输出结果进行编码，得到训练查询集token编码；计算所述训练支持集token编码及所述训练查询集token编码的相似度，并构造相似度矩阵进行相似度分析，得到相似度分数，根据相似度分数计算损失函数，得到训练后的所述预训练语言模型；将所述测试集划分为测试支持集和测试查询集，将所述测试支持集和所述测试查询集数据输入到训练后的所述预训练语言模型中进行处理，进行实体类型预测。

全文数据：

权利要求：

百度查询： 贵州师范大学 一种基于度量学习的小样本命名实体识别方法