arXiv · 2410.01697
Enhancing Adversarial Robustness through Multi-Objective Representation Learning
Abstract
Deep neural networks (DNNs) are vulnerable to small adversarial perturbations, which are tiny changes to the input data that appear insignificant but cause the model to produce drastically different outputs. Many defense methods require modifying model architectures during evaluation or performing test-time data purification. This not only introduces additional complexity but is often architecture-dependent. We show, however, that robust feature learning during training can significantly enhance DNN robustness. We propose MOREL, a multi-objective approach that aligns natural and adversarial features using cosine similarity and multi-positive contrastive losses to encourage similar features for same-class inputs. Extensive experiments demonstrate that MOREL significantly improves robustness against both white-box and black-box attacks. Our code is available at https://github.com/salomonhotegni/MOREL
Explore related subjects
Keep this discovery
Sedjro Salomon Hotegni, Sebastian Peitz. 2024-10-02. Enhancing Adversarial Robustness through Multi-Objective Representation Learning. https://arxiv.org/abs/2410.01697
Cite the original work for its findings. Save a collection to share your selection of sources.