Allan Zhou

Universal Neural Functionals

2024 · arXiv (Cornell University)

A challenging problem in many modern machine learning tasks is to process weight-space features, i.e., to transform or extract information from the weights and gradients of a neural network. Recent works have developed promising weight-space …