一种基于知识对比增强提示的小样本文本分类方法

    公开(公告)号:CN116881457A

    公开(公告)日:2023-10-13

    申请号:CN202310864622.8

    申请日:2023-07-13

    Abstract: 本发明公开了一种基于知识对比增强提示的小样本文本分类方法,属于自然语言处理技术领域,包括以下步骤:S1:初始化连续提示;S2:知识模板生成;S3:联合训练优化;S4:掩码预测。本发明首先利用BiLSTM初始化一个可学习且具有关联性的连续提示向量,然后将预训练语言模型作为知识库,自动生成一组正、负提示模板,在此基础上结合对比学习和掩码语言模型进行联合训练,最终得到有效的连续提示嵌入进行准确的掩码预测,并提供了一种可以自动构造连续提示模板的有效方法,并在14个数据集上进行广泛的实验,结果表明该方法的准确率较最优对比模型平均提升了3.5%以上,解决小样本环境下连续提示对初始参数敏感且容易过拟合的两大问题。

Patent Agency Ranking