Geometry-Preserving Encoder/Decoder in Latent Generative Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Lee, Wonjun, O'Neill, Riley C. W., Zou, Dongmian, Calder, Jeff, Lerman, Gilad
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914078282219520
author Lee, Wonjun
O'Neill, Riley C. W.
Zou, Dongmian
Calder, Jeff
Lerman, Gilad
author_facet Lee, Wonjun
O'Neill, Riley C. W.
Zou, Dongmian
Calder, Jeff
Lerman, Gilad
contents Generative modeling aims to generate new data samples that resemble a given dataset, with diffusion models recently becoming the most popular generative model. One of the main challenges of diffusion models is solving the problem in the input space, which tends to be very high-dimensional. Recently, solving diffusion models in the latent space through an encoder that maps from the data space to a lower-dimensional latent space has been considered to make the training process more efficient and has shown state-of-the-art results. The variational autoencoder (VAE) is the most commonly used encoder/decoder framework in this domain, known for its ability to learn latent representations and generate data samples. In this paper, we introduce a novel encoder/decoder framework with theoretical properties distinct from those of the VAE, specifically designed to preserve the geometric structure of the data distribution. We demonstrate the significant advantages of this geometry-preserving encoder in the training process of both the encoder and decoder. Additionally, we provide theoretical results proving convergence of the training process, including convergence guarantees for encoder training, and results showing faster convergence of decoder training when using the geometry-preserving encoder.
format Preprint
id arxiv_https___arxiv_org_abs_2501_09876
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Geometry-Preserving Encoder/Decoder in Latent Generative Models
Lee, Wonjun
O'Neill, Riley C. W.
Zou, Dongmian
Calder, Jeff
Lerman, Gilad
Numerical Analysis
Machine Learning
Generative modeling aims to generate new data samples that resemble a given dataset, with diffusion models recently becoming the most popular generative model. One of the main challenges of diffusion models is solving the problem in the input space, which tends to be very high-dimensional. Recently, solving diffusion models in the latent space through an encoder that maps from the data space to a lower-dimensional latent space has been considered to make the training process more efficient and has shown state-of-the-art results. The variational autoencoder (VAE) is the most commonly used encoder/decoder framework in this domain, known for its ability to learn latent representations and generate data samples. In this paper, we introduce a novel encoder/decoder framework with theoretical properties distinct from those of the VAE, specifically designed to preserve the geometric structure of the data distribution. We demonstrate the significant advantages of this geometry-preserving encoder in the training process of both the encoder and decoder. Additionally, we provide theoretical results proving convergence of the training process, including convergence guarantees for encoder training, and results showing faster convergence of decoder training when using the geometry-preserving encoder.
title Geometry-Preserving Encoder/Decoder in Latent Generative Models
topic Numerical Analysis
Machine Learning
url https://arxiv.org/abs/2501.09876