Event date:
Feb 22 2021 1:30 pm

Budget-Aware Pruning via Constrained Reinforcement Learning

Supervisor
Dr. Murtaza Taj
Student
Shehryar Malik
Venue
Zoom Meetings (Online)
Event
MS Synopsis defense
Abstract
Deep neural networks have proven to be extremely useful for many tasks. However, they typically have large memory and compute requirements and so are difficult to deploy on small devices, e.g., mobiles. 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 present some preliminary results which indicate the effectiveness of our approach.

Zoom Link:  https://zoom.us/j/97909991209?pwd=d3BUUVZ2SkVRU2t3OURnVjZ0VW9OUT09

Meeting ID:     979 0999 1209

Passcode:      696150