Spectral Norm of Convolutional Layers with Circular and Zero Paddings
Blaise Delattre, Quentin Barthélemy, Alexandre Allauzen
TL;DR
The paper develops a generalized Gram iteration framework to compute guaranteed upper bounds on the spectral norm of convolutional layers, extending from circular padding to zero padding using Toeplitz structures and proving quadratic convergence. By exploiting block-diagonalization in the Fourier domain for circular convolutions and structured Gram iterates for Toeplitz forms, it provides tight bounds with practical computational cost. It introduces Spectral Rescaling (SR), a differentiable procedure that yields true 1-Lipschitz layers and bridges the gap between AOL and spectral normalization, improving robustness while maintaining training stability. Empirical results show improved accuracy and scalability of spectral-norm estimation and demonstrate certified robustness gains on CIFAR datasets, with code available for reproducibility.
Abstract
This paper leverages the use of \emph{Gram iteration} an efficient, deterministic, and differentiable method for computing spectral norm with an upper bound guarantee. Designed for circular convolutional layers, we generalize the use of the Gram iteration to zero padding convolutional layers and prove its quadratic convergence. We also provide theorems for bridging the gap between circular and zero padding convolution's spectral norm. We design a \emph{spectral rescaling} that can be used as a competitive $1$-Lipschitz layer that enhances network robustness. Demonstrated through experiments, our method outperforms state-of-the-art techniques in precision, computational cost, and scalability. The code of experiments is available at https://github.com/blaisedelattre/lip4conv.
