面向红外图像对抗攻击的多任务防御模型构建方法

摘要

本发明提供面向红外图像对抗攻击的多任务防御模型构建方法，包括：根据获取到的红外图像数据集确定训练数据集和测试数据集；建立目标网络模型和附加网络模型，对目标网络模型和附加网络模型进行训练，根据训练获得的最优目标网络模型和最优附加网络模型的参数构建多任务防御模型并进行训练，训练完成获得最优多任务防御模型。本发明利用目标网络和附加网络之间进行参数共享，使目标网络在不牺牲原本分类精度的基础上，拥有附加网络的降噪和抗扰动能力；附加网络和目标分类网络会通过联合训练的方式进行信息交互，而不是仅对附加网络进行单独训练，网络应用泛化性高，提升目标网络的鲁棒性。