基于伪标签及交叉验证的小样本实体识别方法及系统

本发明提供了一种基于伪标签及交叉验证的小样本实体识别方法，包括：基于改进型的K折交叉验证算法对已标注数据进行处理，获得训练集以及测试集对基线模型UIE进行训练以及验证获得微调模型；基于预设规则在未标注数据中进行随机批量选取并进行处理后，获得被选取伪数据；基于被选取伪数据以及已标注数据，使用半监督算法和改进型的K折交叉验证算法对基线模型UIE进行训练获得最终模型；基于最终模型对输入文本进行处理，获得实体识别结果。采用未标注数据小样本数据对模型进行训练的目的同时采用改进型的K折交叉验证算法实现了在一个模型上进行微调，提高小样本实体识别模型的泛化能力以及指标性能，完成了小样本情况下实体的精准识别。