摘要
本发明公开了一种干扰生成对抗网络驱动的知识蒸馏方法,该方法首先获取由三通道RGB格式图像构成的图像分类数据集。其次构建干扰生成对抗网络AD‑GAN,包括教师模型、学生模型和干扰生成模型,三个模型输入均为图像分类数据集中的图像,教师和学生模型输出为图像类别预测的置信度向量,干扰生成模型的输出为带干扰的图像。最后对三个模型进行对抗训练,一个训练批次分为干扰生成模型训练阶段和学生模型蒸馏训练阶段,两者的训练分别使用独立的优化器,干扰生成模型作为对抗训练中的生成器,教师和学生模型共同作为判别器。本发明增强了学生模型的泛化能力和抗噪性能,为资源受限场景提供实用且高效的解决方案。
技术关键词
知识蒸馏方法
生成对抗网络
学生
样本
教师
特征提取模块
特征提取器
阶段
图像类别
预测类别
参数
优化器
神经网络模型
通道
动态更新
模式
强度
系统为您推荐了相关专利信息
掩码矩阵
大语言模型
文本
可读存储介质
主题模型
控制分析系统
马尔科夫链模型
数据处理模块
牵引机
数据传输模块