一种基于鉴别器思想的文本蒸馏方法、系统和存储介质

By employing a text distillation method based on the discriminator concept, teacher and student models are trained using labeled and unlabeled text datasets. The student model parameters are optimized by combining mask training. This resolves the performance-scale contradiction in model compression and enables efficient application on low-resource devices.

CN115271064BActive Publication Date: 2026-07-17HANGZHOU YIWISE INTELLIGENT TECH CO LTD

Patent Information

Authority / Receiving Office
CN · China
Patent Type
Patents(China)
Current Assignee / Owner
HANGZHOU YIWISE INTELLIGENT TECH CO LTD
Filing Date
2022-07-20
Publication Date
2026-07-17

AI Technical Summary

Technical Problem

In existing technologies, knowledge distillation methods struggle to effectively compress model size without diminishing the learning capacity of student models, making them perform similarly to teacher models on resource-constrained devices.

Method used

We employ a text distillation method based on the discriminator concept. By acquiring labeled and unlabeled text datasets, we train the teacher and student models using knowledge distillation. We then combine this with mask training to test the learning performance of the student model and update the student model parameters using knowledge distillation loss and mask training loss.

Benefits of technology

While reducing the number of parameters in the student model, its performance is improved, enabling it to perform similarly to the teacher model on low-resource devices, and the student model can self-test and improve during the learning process.

✦ Generated by Eureka AI based on patent content.

Smart Images

  • Figure CN115271064B_ABST
    Figure CN115271064B_ABST
Patent Text Reader

Abstract

本发明公开了一种基于鉴别器思想的文本蒸馏方法、系统和存储介质,属于自然语言处理领域。方法包括:获取已标注的第一文本数据集和未标注的第二文本数据集;利用所述的第一文本数据集对预训练模型进行训练,将训练好的预训练模型作为教师模型;所述的预训练模型包括若干相同的网络层;构建学生模型,利用第二文本数据集对教师模型和学生模型进行知识蒸馏训练,并采用掩码训练法测试学生模型的学习效果,结合知识蒸馏损失和掩码训练损失更新学生模型的参数;将训练好的学生模型代替教师模型。本发明对传统的知识蒸馏算法进行了改进,让学生模型在参数量尽可能小的情况下提高性能,使其在性能表现上像教师模型一样优秀。
Need to check novelty before this filing date? Find Prior Art