Recursive Disentanglement Network
Yixuan Chen, Yubin Shi, Dongsheng Li, Yujiang Wang, Mingzhi Dong, Yingying Zhao, Robert P. Dick, Qin Lv, Fan Yang, Li Shang
Abstract
Disentangled feature representation is essential for data-efficient learning. The feature space of deep models is inherently compositional. Existing β-VAE-based methods, which only apply disentanglement regularization to the resulting embedding space of deep models, cannot effectively regularize such compositional feature space, resulting in unsatisfactory disentangled results. In this paper, we formulate the compositional disentanglement learning problem from an informationtheoretic perspective and propose a recursive disentanglement network (RecurD) that propagates regulatory inductive bias recursively across the compositional feature space during disentangled representation learning. Experimental studies demonstrate that RecurD outperforms β-VAE and several of its state-of-the-art variants on disentangled representation learning and enables more data-efficient downstream machine learning tasks.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers1
Ask how each one uses itBuilds on8
- NVAE: A Deep Hierarchical Variational AutoencoderArash Vahdat, Jan KautzNeurIPS 2020 · 1,141 citations
- Learning Robust Representations via Multi-View Information BottleneckMarco Federici, Anjan Dutta, Patrick Forré, Nate Kushman et al.ICLR 2020 · 330 citations
- ControlVAE: Controllable Variational AutoencoderHuajie Shao, Shuochao Yao, Dachun Sun, Aston Zhang et al.ICML 2020 · 126 citations
- Demystifying Inter-Class DisentanglementAviv Gabbay, Yedid HoshenICLR 2020 · 63 citations
- Learning Optimal Representations with the Decodable Information BottleneckYann Dubois, Douwe Kiela, David J. Schwab, Ramakrishna VedantamNeurIPS 2020 · 58 citations
Related papers
- Compositional Generalization in Unsupervised Compositional Representation Learning: A Study on Disentanglement and Emergent LanguageZhenlin Xu, Marc Niethammer, Colin RaffelNeurIPS 2022 · 59 citations
- The role of Disentanglement in GeneralisationMilton Llera Montero, Casimir J. H. Ludwig, Rui Ponte Costa, Gaurav Malhotra et al.ICLR 2021 · 97 citations
- Geometric Inductive Biases for Identifiable Unsupervised Learning of Disentangled RepresentationsZiqi Pan, Li Niu, Liqing ZhangAAAI 2023 · 3 citations
- Towards Building A Group-based Unsupervised Representation Disentanglement FrameworkTao Yang, Xuanchi Ren, Yuwang Wang, Wenjun Zeng et al.ICLR 2022 · 36 citations
- Progressive Learning and Disentanglement of Hierarchical RepresentationsZhiyuan Li, Jaideep Vitthal Murkute, Prashnna Kumar Gyawali, Linwei WangICLR 2020 · 47 citations
