Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Disentangled Representation Learning via Flow Matching

About

Disentangled representation learning aims to capture the underlying explanatory factors of observed data, enabling a principled understanding of the data-generating process. Recent advances in generative modeling have introduced new paradigms for learning such representations. However, existing diffusion-based methods encourage factor independence via inductive biases, yet frequently lack strong semantic alignment. In this work, we propose a flow matching-based framework for disentangled representation learning, which casts disentanglement as learning factor-conditioned flows in a compact latent space. To enforce explicit semantic alignment, we introduce a non-overlap (orthogonality) regularizer that suppresses cross-factor interference and reduces information leakage between factors. Extensive experiments across multiple datasets demonstrate consistent improvements over representative baselines, yielding higher disentanglement scores as well as improved controllability and sample fidelity.

Jinjin Chi, Taoping Liu, Mengtao Yin, Ximing Li, Yongcheng Jing, Jialie Shen, Leszek Rutkowski, Dacheng Tao• 2026

Related benchmarks

TaskDatasetResultRank
Disentangled Representation LearningCars3D
FactorVAE0.976
43
DisentanglementShapes3D
FactorVAE Score0.999
26
Showing 2 of 2 rows

Other info

Follow for update