有效

基于混淆矩阵的文本分类方法、装置和计算机设备

孙建彬、姚雪湄、杨克巍、李自拓、蒋平、唐帅文、崔瑞靖、徐博浩、涂莉
中国人民解放军国防科技大学

摘要

本申请涉及一种基于混淆矩阵的文本分类方法、装置和计算机设备,首先获取候选分类算法受到攻击前的混淆矩阵,并在混淆矩阵的基础上,引入基于相同真实标签的预测正确和预测错误之间的质心,然后形成质心偏移四边形,并计算受到攻击前的混淆矩阵中的质心偏移四边形的总面积;接着采用攻击算法对各个候选分类算法进行攻击,并得到候选分类算法受到攻击后的混淆矩阵,同理,构建质心偏移四边形,采用本方法通过计算质心偏移四边形受到攻击前后的总面积差值能直观地看出候选分类算法对抗鲁棒性的差异,能够科学且快速地筛选出对抗鲁棒性最佳的分类算法进行文本分类,保证文本分类效果的稳定性。