Stability and Generalization of Adversarial Diffusion Training
Abstract
Algorithmic stability is an established tool for analyzing generalization. While adversarial training enhances model robustness, it often suffers from robust overfitting and an enlarged generalization gap. Recent work has established the convergence of adversarial training in decentralized networks, but its generalization properties remain unexplored. This work presents a stability-based generalization analysis of adversarial training under the diffusion strategy for convex losses. We derive a bound showing that the generalization error grows with both the adversarial perturbation strength and the number of training steps, a finding consistent with the single-agent case but novel for decentralized settings. Numerical experiments on logistic regression validate these theoretical predictions.
Fields of Study & Subjects
Stochastic Gradient Optimization Techniques
Adversarial Robustness in Machine Learning
Distributed Control Multi-Agent Systems
Publication details
- Journal / source
- ICASSP 2026 - 2026 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP)
- Publisher
- IEEE
- Pages
- 926–930
- Publication date
- 21 April 2026
- Language
- EN