Augmentation-based self-supervised learning can recover the content shared across views without separating its underlying factors into individual coordinates.
We ask whether augmentations can be designed to make those factors identifiable.
Our Approach
Our approach allows content factors to fluctuate across augmented views while preserving their mean for each source instance.
Each instance receives a fixed, recorded set of augmentation strengths that controls how much its factors fluctuate.
When factors respond differently to changes in these strengths, their patterns of variation provide information that distinguishes them.
Under suitable decoder and support assumptions, we prove that these differences identify individual factors in a conditional Gaussian model, up to permutation, scaling, and translation.
Learner Development
We then develop a learner that uses the recorded strengths to estimate these patterns jointly across instances, enabling learning from few views per instance.
Experiments
Synthetic experiments demonstrate near-perfect factor recovery through nonlinear mixing.
Controlled dSprites experiments show improved coordinate alignment using augmentations that target known factors, even without telling the learner which augmentation corresponds to which representation coordinate.
Conclusion
These results suggest a route to component identification in self-supervised learning: augmentation design can supply identifying information beyond shared content alone.