有效

基于置信度引导的文本分类方法、装置和计算机设备

周鋆、杨昊、朱先强、朱承、张维明
中国人民解放军国防科技大学

摘要

本申请涉及一种基于置信度引导的文本分类方法、装置和计算机设备,包括:首先将待分类的目标文本输入预训练的文本分类模型,分别得到目标文本被分到各个文本类别的置信度,其中置信度是根据目标文本对应的softmax函数值取对数得到的,接着根据置信度以及文本类别数量和输入的目标文本的批量大小的乘积构建损失函数,通过优化损失函数更新文本分类模型,最后采用更新后的文本分类模型进行目标文本的分类。采用本发明可以大大提高文本分类的鲁棒性。