AI

Revealing the Utilized Rank of Subspaces of Learning in Neural Networks

1 Mins read

In this work, we study how well the learned weights of a neural network utilize the space available to them. This notion is related to capacity, but additionally incorporates the interaction of the network architecture with the dataset. Most learned weights appear to be full rank, and are therefore not amenable to low rank decomposition. This deceptively implies that the weights are utilizing the entire space available to them. We propose a simple data-driven transformation that projects the weights onto the subspace where the data and the weight interact. This preserves the functional mapping of the layer and reveals its low rank structure. In our findings, we conclude that most models utilize a fraction of the available space. For instance, for ViTB-16 and ViTL-16 trained on ImageNet, the mean layer utilization is 35% and 20% respectively. Our transformation results in reducing the parameters to 50% and 25% respectively, while resulting in less than 0.2% accuracy drop after fine-tuning. We also show that self-supervised pre-training drives this utilization up to 70%, justifying its suitability for downstream tasks.


Source link

Related posts
AI

Alibaba Just Released Marco-o1: Advancing Open-Ended Reasoning in AI

3 Mins read
The field of AI is progressing rapidly, particularly in areas requiring deep reasoning capabilities. However, many existing large models are narrowly focused,…
AI

MIT researchers develop an efficient way to train more reliable AI agents | MIT News

4 Mins read
Fields ranging from robotics to medicine to political science are attempting to train AI systems to make meaningful decisions of all kinds….
AI

Meet Arch 0.1.3: Open-Source Intelligent Proxy for AI Agents

3 Mins read
The integration of AI agents into various workflows has increased the need for intelligent coordination, data routing, and enhanced security among systems….

 

 

Leave a Reply

Your email address will not be published. Required fields are marked *