基于知识蒸馏的短文本分类方法

• 作者: 孙红 黄瓯严
• 来源期刊: 软件导刊
• 关键词: 知识蒸馏 文本分类 双向模型 自然语言处理
• 年，卷（期）: 2021,（6）
• 所属期刊栏目: 人工智能
• 页码范围: 23-27
• 页数: 5页
• 分类号: TP301
• 语种: 中文
• DOI: 10.11907/rjdk.202006

摘要

近年来,随着深度学习技术在自然语言处理任务中的广泛应用,语言模型规模越来越大.然而,大规模模型推理速度慢、资源消耗成本高,难以在工业上进行应用,小规模模型又难以达到大规模模型效果.因此提出一种基于教师—学生框架的知识蒸馏模型,以预训练模型BERT作为教师模型,以长短时记忆网络(BiLSTM)等小模型作为学生模型,通过教师指导学生学习的方式将教师模型学习到的知识迁移至学生模型中.实验结果表明,蒸馏模型将运算推理时间缩短至教师模型的1/725,将学生模型短文本分类准确率提升3.16％.