Depthwise Separable Convolutions Allow for Fast and Memory-Efficient Spectral Normalization

Christina Runkel,Christian Etmann,Michael Möller,Carola-Bibiane Schönlieb
DOI: https://doi.org/10.48550/arXiv.2102.06496
2021-02-12
Abstract:An increasing number of models require the control of the spectral norm of convolutional layers of a neural network. While there is an abundance of methods for estimating and enforcing upper bounds on those during training, they are typically costly in either memory or time. In this work, we introduce a very simple method for spectral normalization of depthwise separable convolutions, which introduces negligible computational and memory overhead. We demonstrate the effectiveness of our method on image classification tasks using standard architectures like MobileNetV2.
Machine Learning,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?