Graph Neural Networks for Learning Equivariant Representations of Neural Networks | OpenReview
Neural networks that process the parameters of other neural networks find applications in domains as diverse as classifying implicit neural representations, generating neural network weights, and predicting generalization errors. However, existing approaches either overlook the inherent permutation symmetry in the neural network or rely on intricate weight-sharing patterns to achieve equivariance, while ignoring the impact of the network architecture itself. In this work, we propose to represent neural networks as computational graphs of parameters, which allows us to harness powerful graph neural networks and transformers that preserve permutation symmetry. Consequently, our approach enables a single model to encode neural computational graphs with diverse architectures. We showcase the effectiveness of our method on a wide range of tasks, including classification and editing of implicit neural representations, predicting generalization performance, and learning to optimize, while con
Verifying your browser | OpenReview Verifying your browser Complete the check below to continue to OpenReview Please complete the verification above. Have an OpenReview account? Sign in to skip this check.
Explore this link on the map →related reading
- Naturally Occurring Equivariance in Neural Networksdistill.pub
- Aman's AI Journal • Primers • Ilya Sutskever's Top 30aman.ai
- 10. Equivariant Neural Networks — deep learning for molecules & materialsdmol.pub
- A Gentle Introduction to Graph Neural Networksdistill.pub
- geometricdeeplearning.com/book/graphs.htmlgeometricdeeplearning.com
- Understanding Convolutions on Graphsdistill.pub
- Neural Networks, Manifolds, and Topology -- colah's blogcolah.github.io
- Modeling Symmetries | Mark Neumannmarkneumann.xyz
- Bayesian Neural Networkscs.toronto.edu
- Scale-invariant-CNNs/docs/Report/Implementation.md at master · ruoqizzz/Scale-invariant-CNNs · GitHubgithub.com
- arxiv.org/pdf/1805.08522arxiv.org
- [1411.5908] Understanding image representations by measuring their equivariance and equivalencearxiv.org