基于改进的ViT损失分布差异的对抗样本检测方法及系统

本发明公开了一种基于改进的ViT损失分布差异的对抗样本检测方法及系统，所述方法包括：获取干净样本，所述干净样本为无类别标签的图像样本；构建检测模型，所述检测模型为采用分层注意力机制和注意力池化机制改进的ViT模型；将所获取的干净样本进行总变差降噪，然后将降噪前后的干净样本输入检测模型，以均方误差作为损失函数训练模型；对待检测样本进行总变差降噪，然后将降噪前后的待检测样本分别输入训练好的检测模型，计算待检测样本降噪前后损失的均方误差；最后，计算待检测样本的K‑S值、Z‑score标准分，并由此判定待检测样本是否为对抗样本。本发明能够有效提高检测模型的性能。