Sandeep Nagar

Parallel Backpropagation for Inverse of a Convolution with Application to Normalizing Flows

The inverse of an invertible convolution is an important operation that comes up in Normalizing Flows, Image Deblurring, etc. The naive algorithm for backpropagation of this operation using Gaussian elimination has running time $O(n^3)$ where $n$ is the number of pixels in the image. We give a fast…