arXiv · 2302.10633
Generalization Bounds for Adversarial Contrastive Learning
Abstract
Deep networks are well-known to be fragile to adversarial attacks, and adversarial training is one of the most popular methods used to train a robust model. To take advantage of unlabeled data, recent works have applied adversarial training to contrastive learning (Adversarial Contrastive Learning; ACL for short) and obtain promising robust performance. However, the theory of ACL is not well understood. To fill this gap, we leverage the Rademacher complexity to analyze the generalization performance of ACL, with a particular focus on linear models and multi-layer neural networks under $\ell_p$ attack ($p \ge 1$). Our theory shows that the average adversarial risk of the downstream tasks can be upper bounded by the adversarial unsupervised risk of the upstream task. The experimental results validate our theory.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Xin Zou, Weiwei Liu. 2023-02-21. Generalization Bounds for Adversarial Contrastive Learning. https://arxiv.org/abs/2302.10633
Cite the original work for its findings. Save a collection to share your selection of sources.