Salvato in:
Dettagli Bibliografici
Autori principali: Wang, Archer, Anand, Emile, Du, Yilun, Soljačić, Marin
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:https://arxiv.org/abs/2601.22057
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866918394980204544
author Wang, Archer
Anand, Emile
Du, Yilun
Soljačić, Marin
author_facet Wang, Archer
Anand, Emile
Du, Yilun
Soljačić, Marin
contents Decomposing complex data into factorized representations can reveal reusable components and enable synthesizing new samples via component recombination. We investigate this in the context of diffusion-based models that learn factorized latent spaces without factor-level supervision. In images, factors can capture background, illumination, and object attributes; in robotic videos, they can capture reusable motion components. To improve both latent factor discovery and quality of compositional generation, we introduce an adversarial training signal via a discriminator trained to distinguish between single-source samples and those generated by recombining factors across sources. By optimizing the generator to fool this discriminator, we encourage physical and semantic consistency in the resulting recombinations. Our method outperforms implementations of prior baselines on CelebA-HQ, Virtual KITTI, CLEVR, and Falcor3D, achieving lower FID scores and better disentanglement as measured by MIG and MCC. Furthermore, we demonstrate a novel application to robotic video trajectories: by recombining learned action components, we generate diverse sequences that significantly increase state-space coverage for exploration on the LIBERO benchmark.
format Preprint
id arxiv_https___arxiv_org_abs_2601_22057
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Unsupervised Decomposition and Recombination with Discriminator-Driven Diffusion Models
Wang, Archer
Anand, Emile
Du, Yilun
Soljačić, Marin
Computer Vision and Pattern Recognition
Artificial Intelligence
I.2.10; I.2.9
Decomposing complex data into factorized representations can reveal reusable components and enable synthesizing new samples via component recombination. We investigate this in the context of diffusion-based models that learn factorized latent spaces without factor-level supervision. In images, factors can capture background, illumination, and object attributes; in robotic videos, they can capture reusable motion components. To improve both latent factor discovery and quality of compositional generation, we introduce an adversarial training signal via a discriminator trained to distinguish between single-source samples and those generated by recombining factors across sources. By optimizing the generator to fool this discriminator, we encourage physical and semantic consistency in the resulting recombinations. Our method outperforms implementations of prior baselines on CelebA-HQ, Virtual KITTI, CLEVR, and Falcor3D, achieving lower FID scores and better disentanglement as measured by MIG and MCC. Furthermore, we demonstrate a novel application to robotic video trajectories: by recombining learned action components, we generate diverse sequences that significantly increase state-space coverage for exploration on the LIBERO benchmark.
title Unsupervised Decomposition and Recombination with Discriminator-Driven Diffusion Models
topic Computer Vision and Pattern Recognition
Artificial Intelligence
I.2.10; I.2.9
url https://arxiv.org/abs/2601.22057