Orthonormal Convolutions for the Rotation Based Iterative Gaussianization

TL;DR

This paper introduces Convolutional RBIG, an extension of rotation-based iterative Gaussianization that employs learned orthonormal convolutions, enabling scalable image Gaussianization and extraction of statistical data properties.

Contribution

It proposes a novel convolutional rotation method using orthonormal convolutions optimized via reconstruction loss, extending RBIG to high-dimensional image data.

Findings

Enables Gaussianization of larger image datasets.

Allows extraction of multivariate mutual information from images.

Demonstrates texture synthesis and feature visualization.

Abstract

In this paper we elaborate an extension of rotation-based iterative Gaussianization, RBIG, which makes image Gaussianization possible. Although RBIG has been successfully applied to many tasks, it is limited to medium dimensionality data (on the order of a thousand dimensions). In images its application has been restricted to small image patches or isolated pixels, because rotation in RBIG is based on principal or independent component analysis and these transformations are difficult to learn and scale. Here we present the \emph{Convolutional RBIG}: an extension that alleviates this issue by imposing that the rotation in RBIG is a convolution. We propose to learn convolutional rotations (i.e. orthonormal convolutions) by optimising for the reconstruction loss between the input and an approximate inverse of the transformation using the transposed convolution operation. Additionally, we suggest different regularizers in learning these orthonormal convolutions. For example, imposing sparsity in the activations leads to a transformation that extends convolutional independent component analysis to multilayer architectures. We also highlight how statistical properties of the data, such as multivariate mutual information, can be obtained from \emph{Convolutional RBIG}. We illustrate the behavior of the transform with a simple example of texture synthesis, and analyze its properties by visualizing the stimuli that maximize the response in certain feature and layer.