通过对比监督实现可控制的因子解耦,无需对抗训练。
XFACTORS: Disentangled Information Bottleneck via Contrastive Supervision
- 用对比损失让相同因子值的隐变量聚在一起
- 在多个数据集上达到最优解耦分数,支持因子替换
- 无需额外标注,适合真实图像数据和大规模属性
解耦表示学习旨在将变化因素映射到独立的表示分量。纯无监督方法在合成数据上表现良好,但在真实数据上缺乏语义因子;监督方法则因依赖对抗目标或辅助分类器而难以扩展。本文提出 extsc{XFactors},一种弱监督变分自编码框架,可对选定因子进行解耦并实现显式控制。基于解耦信息瓶颈视角,将表示分解为残差子空间 $/mathcal{S}$ 与因子特异子空间 $/mathcal{T}_1, dots,/mathcal{T}_K$。每个目标因子通过对比监督(InfoNCE损失)在对应 $/mathcal{T}_i$ 中编码:相同因子值的隐变量被拉近,不同值被推远。同时,KL 正则化施加高斯结构于 $/mathcal{S}$ 及聚合因子子空间,组织几何结构且无需额外监督非目标因子,避免对抗训练与分类器。在多个数据集上, extsc{XFactors} 以固定超参数取得当前最优解耦分数,各子空间保持稳定因子对齐,支持通过潜变量替换实现可控因子交换。进一步验证其随潜空间容量增长正确扩展,并在真实数据集 CelebA 上评估。代码已公开。
原文摘要 · Abstract (English)
Disentangled representation learning aims to map independent factors of variation to independent representation components. On one hand, purely unsupervised approaches have proven successful on fully disentangled synthetic data, but fail to recover semantic factors from real data without strong inductive biases. On the other hand, supervised approaches are unstable and hard to scale to large attribute sets because they rely on adversarial objectives or auxiliary classifiers. We introduce \textsc{XFactors}, a weakly-supervised VAE framework that disentangles and provides explicit control over a chosen set of factors. Building on the Disentangled Information Bottleneck perspective, we decompose the representation into a residual subspace $\mathcal{S}$ and factor-specific subspaces $\mathcal{T}_1,\ldots,\mathcal{T}_K$ and a residual subspace $\mathcal{S}$. Each target factor is encoded in its assigned $\mathcal{T}_i$ through contrastive supervision: an InfoNCE loss pulls together latents sharing the same factor value and pushes apart mismatched pairs. In parallel, KL regularization imposes a Gaussian structure on both $\mathcal{S}$ and the aggregated factor subspaces, organizing the geometry without additional supervision for non-targeted factors and avoiding adversarial training and classifiers. Across multiple datasets, with constant hyperparameters, \textsc{XFactors} achieves state-of-the-art disentanglement scores and yields consistent qualitative factor alignment in the corresponding subspaces, enabling controlled factor swapping via latent replacement. We further demonstrate that our method scales correctly with increasing latent capacity and evaluate it on the real-world dataset CelebA. Our code is available at \href{https://github.com/ICML26-anon/XFactors}{github.com/ICML26-anon/XFactors}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。