English

Harmonics of Learning: Universal Fourier Features Emerge in Invariant Networks

Machine Learning 2024-06-17 v3 Artificial Intelligence Signal Processing

Abstract

In this work, we formally prove that, under certain conditions, if a neural network is invariant to a finite group then its weights recover the Fourier transform on that group. This provides a mathematical explanation for the emergence of Fourier features -- a ubiquitous phenomenon in both biological and artificial learning systems. The results hold even for non-commutative groups, in which case the Fourier transform encodes all the irreducible unitary group representations. Our findings have consequences for the problem of symmetry discovery. Specifically, we demonstrate that the algebraic structure of an unknown group can be recovered from the weights of a network that is at least approximately invariant within certain bounds. Overall, this work contributes to a foundation for an algebraic learning theory of invariant neural network representations.

Keywords

Cite

@article{arxiv.2312.08550,
  title  = {Harmonics of Learning: Universal Fourier Features Emerge in Invariant Networks},
  author = {Giovanni Luca Marchetti and Christopher Hillar and Danica Kragic and Sophia Sanborn},
  journal= {arXiv preprint arXiv:2312.08550},
  year   = {2024}
}

Comments

Accepted at the Conference on Learning Theory (COLT) 2024