TOWARDS NONLINEAR DISENTANGLEMENT IN NATURAL DATA WITH TEMPORAL SPARSE CODING

Klindt, D, Schott, L, Sharma, Y, Ustyuzhaninov, I, Brendel, W, Bethge, M, Paiton, DM (January 2021) TOWARDS NONLINEAR DISENTANGLEMENT IN NATURAL DATA WITH TEMPORAL SPARSE CODING. In: ICLR 2021 - 9th International Conference on Learning Representations.

Abstract

Disentangling the underlying generative factors from data has so far been limited to carefully constructed scenarios. We propose a path towards natural data by first showing that the statistics of natural data provide enough structure to enable disentanglement, both theoretically and empirically. Specifically, we provide evidence that objects in natural movies undergo transitions that are typically small in magnitude with occasional large jumps, which is characteristic of a temporally sparse distribution. Leveraging this finding we provide a novel proof that relies on a sparse prior on temporally adjacent observations to recover the true latent variables up to permutations and sign flips, providing a stronger result than previous work. We show that equipping practical estimation methods with our prior often surpasses the current state-of-the-art on several established benchmark datasets without any impractical assumptions, such as knowledge of the number of changing generative factors. Furthermore, we contribute two new benchmarks, Natural Sprites and KITTI Masks, which integrate the measured natural dynamics to enable disentanglement evaluation with more realistic datasets. We test our theory on these benchmarks and demonstrate improved performance. We also identify non-obvious challenges for current methods in scaling to more natural domains. Taken together our work addresses key issues in disentanglement research for moving towards more natural settings.

Item Type: Conference or Workshop Item (Paper)
CSHL Authors:
Communities: CSHL labs > Klindt lab
SWORD Depositor: CSHL Elements
Depositing User: CSHL Elements
Date: 1 January 2021
Date Deposited: 11 Apr 2024 15:48
Last Modified: 11 Apr 2024 15:48
Related URLs:
URI: https://repository.cshl.edu/id/eprint/41499

Actions (login required)

Administrator's edit/view item Administrator's edit/view item