Stability and Generalization of Adversarial Diffusion Training

Read full text
Authors: H.M.M. Hosseini, Ying Shan Cao, Ali H. Sayed
Cited by: 0
References: 15

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