摘要
本发明提出了一种基于信息补偿与知识蒸馏二值神经网络的图像分类方法,实现步骤为:获取训练样本集和验证样本集;对教师模型进行迭代训练;构建基于信息补偿与知识蒸馏二值神经网络的图像分类模型;对图像分类模型进行迭代训练;获取图像分类结果。本发明信息补偿模块通过补偿二值深度可分离卷积模块的输出特征图,显著提升了输出特征图的信息量;通过预训练教师模型和学生模型分别计算每个训练样本在各目标类别上的预测概率,并据此计算分布损失,在反向传播过程中利用该分布损失更新学生模型的参数,并且教师模型在训练过程中引导学生模型输出更精确的概率分布,帮助学生模型更好地学习复杂特征,从而显著提升了分类性能。
技术关键词
图像分类模型
二值神经网络
图像分类方法
训练样本集
学生
教师
轻量级卷积神经网络
蒸馏
卷积模块
输出特征
代表
随机梯度下降
动态
层叠
参数
因子
表达式
系统为您推荐了相关专利信息
定位管理方法
校园
车辆
ARIMA模型
非机动车管理系统
细胞图像分类方法
图像分类系统
宫颈癌疾病
网络模型训练
注意力
智慧农业灌溉
可视化控制方法
AI语音
动作建议
环境数据监测装置
无标签数据
医学影像分割方法
半监督训练
医学影像数据
分阶段