Statistics and Learning Theory
Learning Theory I

Stochastic Bottleneck: Rateless Auto-Encoder for Flexible Dimensionality Reduction

Toshiaki Koike-Akino, Ye Wang

Date & Time

01:00 am – 01:00 am


We propose a new concept of rateless auto-encoders (RL-AEs) that enable a flexible latent dimensionality, which can be seamlessly adjusted for varying distortion. In the proposed RL-AEs, instead of a deterministic bottleneck architecture, we use an over-complete representation that is stochastically regularized with weighted dropouts. Our RL-AEs employ monotonically increasing dropout rates across the latent representation nodes such that the latent variables become sorted by importance like in principal component analysis (PCA). This is motivated by the rateless property of conventional PCA, where the least important principal components can be discarded to realize variable rate dimensionality reduction that gracefully degrades the distortion. Our proposed stochastic bottleneck framework enables seamless rate adaptation with high reconstruction performance, without requiring predetermined latent dimensionality at training. We experimentally demonstrate that the proposed RL-AEs can achieve variable dimensionality reduction while retaining nearly optimal distortion compared to conventional AEs.


Toshiaki Koike-Akino

Mitsubishi Electric Research Laboratories (MERL)

Ye Wang

Mitsubishi Electric Research Laboratories (MERL)

Session Chair

Lizhong Zheng

Massachusetts Institute of Technology