一种图像二分类模型的训练方法、图像识别方法

    公开(公告)号:CN117576496A

    公开(公告)日:2024-02-20

    申请号:CN202311286454.5

    申请日:2023-10-07

    Abstract: 本发明实施例提供了一种图像二分类模型的训练方法,图像二分类模型用于判定图像为二分类中的一类的概率,该方法包括:S1、获取原始图像训练集;S2、对所述原始图像训练集按照预设的规则添加分布扰动,得到对应的图像对抗训练集;S3、利用图像对抗训练集对所述图像二分类模型进行多次迭代训练,并在每次迭代训练时基于正样本为目标图像的概率预测值以及负样本为目标图像的概率预测值计算图像二分类模型的性能评估结果;S4、根据图像二分类模型的性能评估结果,基于目标逐样本损失函数更新图像对抗训练集以及图像二分类模型中可学习的参数。通过上述方法训练图像二分类模型,提高了AUC在分布扰动下的鲁棒性,从而提高图像二分类模型的性能。

Patent Agency Ranking