ELROND: Exploring and decomposing intrinsic capabilities of diffusion models

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Skierś, Paweł, Trzciński, Tomasz, Deja, Kamil
Format: Preprint
Veröffentlicht: 2026
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866908827438284800
author Skierś, Paweł
Trzciński, Tomasz
Deja, Kamil
author_facet Skierś, Paweł
Trzciński, Tomasz
Deja, Kamil
contents A single text prompt passed to a diffusion model often yields a wide range of visual outputs determined solely by stochastic process, leaving users with no direct control over which specific semantic variations appear in the image. While existing unsupervised methods attempt to analyze these variations via output features, they omit the underlying generative process. In this work, we propose a framework to disentangle these semantic directions directly within the input embedding space. To that end, we collect a set of gradients obtained by backpropagating the differences between stochastic realizations of a fixed prompt that we later decompose into meaningful steering directions with either Principal Components Analysis or Sparse Autoencoder. Our approach yields three key contributions: (1) it isolates interpretable, steerable directions for precise, fine-grained control over a single concept; (2) it effectively mitigates mode collapse in distilled models by reintroducing lost diversity; and (3) it establishes a novel estimator for concept complexity under a specific model, based on the dimensionality of the discovered subspace.
format Preprint
id arxiv_https___arxiv_org_abs_2602_10216
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle ELROND: Exploring and decomposing intrinsic capabilities of diffusion models
Skierś, Paweł
Trzciński, Tomasz
Deja, Kamil
Machine Learning
A single text prompt passed to a diffusion model often yields a wide range of visual outputs determined solely by stochastic process, leaving users with no direct control over which specific semantic variations appear in the image. While existing unsupervised methods attempt to analyze these variations via output features, they omit the underlying generative process. In this work, we propose a framework to disentangle these semantic directions directly within the input embedding space. To that end, we collect a set of gradients obtained by backpropagating the differences between stochastic realizations of a fixed prompt that we later decompose into meaningful steering directions with either Principal Components Analysis or Sparse Autoencoder. Our approach yields three key contributions: (1) it isolates interpretable, steerable directions for precise, fine-grained control over a single concept; (2) it effectively mitigates mode collapse in distilled models by reintroducing lost diversity; and (3) it establishes a novel estimator for concept complexity under a specific model, based on the dimensionality of the discovered subspace.
title ELROND: Exploring and decomposing intrinsic capabilities of diffusion models
topic Machine Learning
url https://arxiv.org/abs/2602.10216