Downsampling leads to Image Memorization in Convolutional Autoencoders

Adityanarayanan Radhakrishnan, Caroline Uhler, Mikhail Belkin

Abstract:

Memorization of data in deep neural networks has become a subject of significant research interest.
In this paper, we link memorization of images in deep convolutional autoencoders to downsampling through strided convolution. To analyze this mechanism in a simpler setting, we train linear convolutional autoencoders and show that linear combinations of training data are stored as eigenvectors in the linear operator corresponding to the network when downsampling is used. On the other hand, networks without downsampling do not memorize training data. We provide further evidence that the same effect happens in nonlinear networks. Moreover, downsampling in nonlinear networks causes the model to not only memorize just linear combinations of images, but individual training images. Since convolutional autoencoder components are building blocks of deep convolutional networks, we envision that our findings will shed light on the important phenomenon of memorization in over-parameterized deep networks.

Keywords: Memorization in Deep Learning, Convolutional Autoencoders

TL;DR: We identify downsampling as a mechansim for memorization in convolutional autoencoders.

Data: [CIFAR-10](https://paperswithcode.com/dataset/cifar-10)