cs.LGJun 24, 2026

Variational Autoencoder Layer

Authors: Gananath R

Abstract

Variational Autoencoders (VAEs) belong to a family of autoencoders with probabilistic properties, making them well suited for generating data by producing a smooth and continuous latent space. Despite being introduced over a decade ago, the method continues to be widely adopted in both research and industry for diverse applications. While VAEs are typically used as standalone models, this paper introduces a novel approach to integrate them as a neural network layer. Furthermore, a new training strategy is proposed for models incorporating these layers, and their performance is thoroughly analyzed.

Explore similar work

CardsList
  1. Autoencoders in Function Space

    Aug 2, 2024Justin Bunker, Mark Girolami, Hefin Lambley +2Variational AutoencoderMasked Autoencoders