Measuring Learned Equivariance: An Information Theoretic Approach
Learning symmetries through data augmentation offers an alternative to architectural equivariance, but does not guarantee that a network respects those symmetries.
ProofPaper ↗
Key points
- Measuring learned equivariance is especially difficult for intermediate representations, whose group actions are not specified.
- In a finite discrete setting, we show that equivariance requires representations of original and transformed inputs to determine one another exactly, yielding a test based on normalized mutual information (NMI).
- For high-dimensional activations, we use centered kernel alignment (CKA) as a scalable test, with an exact guarantee for orthogonal feature transformations.
- Across architectures and tasks, augmentation usually increases transformation consistency, showing that learned equivariance can be probed directly from representations without recovering the hidden transformation rules that relate them.
Sources (1)
- [1]Measuring Learned Equivariance: An Information Theoretic ApproacharXiv (AI, ML, NLP, CV, robotics, multi-agent) · Oct 12, 06:24 PM
Learning symmetries through data augmentation offers an alternative to architectural equivariance, but does not guarantee that a network respects those symmetries.
Measuring learned equivariance is especially difficult for intermediate representations, whose group actions are not specified.
Extractive summary: sentences quoted from the sources.