Effective and Robust Adversarial Training against Data and Label Corruptions

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhang, Peng-Fei, Huang, Zi, Xu, Xin-Shun, Bai, Guangdong
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917659916894208
author Zhang, Peng-Fei
Huang, Zi
Xu, Xin-Shun
Bai, Guangdong
author_facet Zhang, Peng-Fei
Huang, Zi
Xu, Xin-Shun
Bai, Guangdong
contents Corruptions due to data perturbations and label noise are prevalent in the datasets from unreliable sources, which poses significant threats to model training. Despite existing efforts in developing robust models, current learning methods commonly overlook the possible co-existence of both corruptions, limiting the effectiveness and practicability of the model. In this paper, we develop an Effective and Robust Adversarial Training (ERAT) framework to simultaneously handle two types of corruption (i.e., data and label) without prior knowledge of their specifics. We propose a hybrid adversarial training surrounding multiple potential adversarial perturbations, alongside a semi-supervised learning based on class-rebalancing sample selection to enhance the resilience of the model for dual corruption. On the one hand, in the proposed adversarial training, the perturbation generation module learns multiple surrogate malicious data perturbations by taking a DNN model as the victim, while the model is trained to maintain semantic consistency between the original data and the hybrid perturbed data. It is expected to enable the model to cope with unpredictable perturbations in real-world data corruption. On the other hand, a class-rebalancing data selection strategy is designed to fairly differentiate clean labels from noisy labels. Semi-supervised learning is performed accordingly by discarding noisy labels. Extensive experiments demonstrate the superiority of the proposed ERAT framework.
format Preprint
id arxiv_https___arxiv_org_abs_2405_04191
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Effective and Robust Adversarial Training against Data and Label Corruptions
Zhang, Peng-Fei
Huang, Zi
Xu, Xin-Shun
Bai, Guangdong
Machine Learning
Computer Vision and Pattern Recognition
68T30
I.4.0
Corruptions due to data perturbations and label noise are prevalent in the datasets from unreliable sources, which poses significant threats to model training. Despite existing efforts in developing robust models, current learning methods commonly overlook the possible co-existence of both corruptions, limiting the effectiveness and practicability of the model. In this paper, we develop an Effective and Robust Adversarial Training (ERAT) framework to simultaneously handle two types of corruption (i.e., data and label) without prior knowledge of their specifics. We propose a hybrid adversarial training surrounding multiple potential adversarial perturbations, alongside a semi-supervised learning based on class-rebalancing sample selection to enhance the resilience of the model for dual corruption. On the one hand, in the proposed adversarial training, the perturbation generation module learns multiple surrogate malicious data perturbations by taking a DNN model as the victim, while the model is trained to maintain semantic consistency between the original data and the hybrid perturbed data. It is expected to enable the model to cope with unpredictable perturbations in real-world data corruption. On the other hand, a class-rebalancing data selection strategy is designed to fairly differentiate clean labels from noisy labels. Semi-supervised learning is performed accordingly by discarding noisy labels. Extensive experiments demonstrate the superiority of the proposed ERAT framework.
title Effective and Robust Adversarial Training against Data and Label Corruptions
topic Machine Learning
Computer Vision and Pattern Recognition
68T30
I.4.0
url https://arxiv.org/abs/2405.04191