Isometric Representation Learning for Disentangled Latent Space of Diffusion Models

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Hahm, Jaehoon, Lee, Junho, Kim, Sunghyun, Lee, Joonseok
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866914872498847744
author Hahm, Jaehoon
Lee, Junho
Kim, Sunghyun
Lee, Joonseok
author_facet Hahm, Jaehoon
Lee, Junho
Kim, Sunghyun
Lee, Joonseok
contents The latent space of diffusion model mostly still remains unexplored, despite its great success and potential in the field of generative modeling. In fact, the latent space of existing diffusion models are entangled, with a distorted mapping from its latent space to image space. To tackle this problem, we present Isometric Diffusion, equipping a diffusion model with a geometric regularizer to guide the model to learn a geometrically sound latent space of the training data manifold. This approach allows diffusion models to learn a more disentangled latent space, which enables smoother interpolation, more accurate inversion, and more precise control over attributes directly in the latent space. Our extensive experiments consisting of image interpolations, image inversions, and linear editing show the effectiveness of our method.
format Preprint
id arxiv_https___arxiv_org_abs_2407_11451
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Isometric Representation Learning for Disentangled Latent Space of Diffusion Models
Hahm, Jaehoon
Lee, Junho
Kim, Sunghyun
Lee, Joonseok
Machine Learning
Computer Vision and Pattern Recognition
The latent space of diffusion model mostly still remains unexplored, despite its great success and potential in the field of generative modeling. In fact, the latent space of existing diffusion models are entangled, with a distorted mapping from its latent space to image space. To tackle this problem, we present Isometric Diffusion, equipping a diffusion model with a geometric regularizer to guide the model to learn a geometrically sound latent space of the training data manifold. This approach allows diffusion models to learn a more disentangled latent space, which enables smoother interpolation, more accurate inversion, and more precise control over attributes directly in the latent space. Our extensive experiments consisting of image interpolations, image inversions, and linear editing show the effectiveness of our method.
title Isometric Representation Learning for Disentangled Latent Space of Diffusion Models
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2407.11451