一种模型蒸馏方法、装置及介质

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
一种模型蒸馏方法、装置及介质
申请号:CN202510205236
申请日期:2025-02-24
公开号:CN119692430B
公开日期:2025-06-13
类型:发明专利
摘要
本申请公开了一种模型蒸馏方法、装置及介质,该方法包括:获取教师模型的教师相似度矩阵,学生模型的学生相似度矩阵,以及表征教师模型与学生模型相似度排序差异的排序差异矩阵;教师模型的计算精度高于学生模型。根据教师相似度矩阵构造对比学习样本对,并根据排序差异矩阵构造排序差异样本对;基于对比学习样本对和排序差异样本对构建蒸馏损失函数;通过蒸馏损失函数对学生模型进行迭代训练,得到目标学生模型。由此,基于教师模型构造对比学习样本对和排序差异样本对,构建蒸馏损失函数,并通过该蒸馏损失函数对学生模型进行迭代训练,使得学生模型可以学习教师模型的知识,提升学生模型准确性,从而实现同时兼顾计算效率和计算准确性。
技术关键词
模型蒸馏方法 教师 学生 样本 矩阵 蒸馏装置 残差网络 模型训练模块 标签 处理器 对象 可读存储介质 存储器 程序 图像 计算机