Energy-Based Models Trained with Equilibrium Propagation are Inherently Robust
Deep neural networks (DNNs) are easily fooled by adversarial perturbations that are impercep-tible to humans. Adversarial training, a process where adversarial examples are added to the training set, is the current state-of-the-art defense against adversarial attacks, but it lowers the model's accuracy on clean inputs, is computationally expensive, and offers less robustness to natural noise. In contrast, energy-based models (EBMs), which were designed for efficient im-plementation in neuromorphic hardware and physical systems, incorporate feedback connections from each layer to the previous layer, yielding a recurrent, deep-attractor architecture which we hypothesize should make them naturally robust. Our work is the first to explore the robustness of EBMs to both natural cor-ruptions and adversarial attacks, which we do using the CIFAR-IO and CIFAR-IOO datasets. We demonstrate that EBMs are more robust than transformers and display comparable robustness to adversarially-trained DNNs on gradient-based (white-box) attacks, query-based (black-box) attacks, and natural perturbations without sacrificing clean accuracy, and without the need for adversarial training or additional training techniques.
Authors
- Juston Moore (ORCID: https://orcid.org/0000-0003-2515-3647)
- Michael Teti (ORCID: https://orcid.org/0000-0002-0754-1761)
- Michal Kucer (ORCID: https://orcid.org/0000-0002-1438-4442)
- Garrett T. Kenyon (ORCID: https://orcid.org/0000-0003-4836-3938)
- Siddharth Mansingh
Institutions
- Los Alamos National Laboratory (US)
Publication Details
- Published
- 2024-12-18
- DOI
- https://doi.org/10.1109/icmla61862.2024.00105
- Primary Topic
- Opinion Dynamics and Social Influence
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- U.S. Department of Energy
- National Nuclear Security Administration