WeightCLIP: Aligning Datasets and Models for Weight Space Learning

Weight space learning aims to learn representations of neural network (NN) weights, enabling different downstream tasks.

Weight space learning aims to learn representations of neural network (NN) weights, enabling different downstream tasks. Existing approaches show promising performance, but lacking a way to shape these weight-space representations using information about the datasets the models…