Event date:
May 25 2021 5:00 pm

Neural Network Pruning via Constrained Reinforcement Learning

Supervisor
Dr. Murtaza Taj
Student
Shehryar Malik
Venue
Zoom Meetings (Online)
Event
MS Thesis defense
Abstract
Deep neural networks have proved to be extremely useful for several tasks such as those in computer vision and natural language processing. However, the problem with these networks is that they typically have large memory and compute requirements which makes it difficult to deploy them on small devices such as mobiles and tablets. Pruning reduces the size of neural networks by removing (‘pruning’) neurons such that the performance drop is minimal. Traditional pruning approaches focus on designing metrics to quantify the 'usefulness' of a neuron. However, designing such metrics is often quite tedious and suboptimal. Some recent approaches have instead focused on training auxiliary neural networks to automatically learn how useful each neuron is in the network we wish to prune. In many cases, we are constrained in the amount of computational power that we can use. However, these auxiliary networks often do not take this into account. In this work, we focus on using constrained reinforcement learning algorithms to train these auxiliary neural networks to respect pre-defined computational budgets. We also carry out experiments to demonstrate the effectiveness of our approach.

Zoom Link:     https://lums-edu-pk.zoom.us/j/96074049125

Meeting ID:     9607 4049 125