Neural network simplification using a progressive barrier based approach

Neural networks are indispensable to state-of-the-art artificial intelligence algorithms. However, its high accuracy comes at the cost of high computational complexity. This leads to the high operating cost of data centers and also hinders its deployment on mobile devices. In this thesis, we propose an algorithm to address this problem. The proposed algorithm uses progressive barriers to automatically and progressively simplify a pre-trained neural network until the target complexity is met while maximizing the accuracy. Along with the neural network that meets the target complexity, the algorithm also generates a family of simplified networks with different accuracy-complexity trade-offs, which allows for dynamic network selection and further study. Experiment results show that the algorithm achieves better accuracy-complexity trade-offs on a highly compact MobileNet architecture, compared with state-of-the-art automated network simplification approaches. For image classification on the ImageNet dataset, the algorithm reduces the number of multiply-accumulate operations by 1.68x while achieving 0.9% higher accuracy.

Department dc:contributor.department

Massachusetts Institute of Technology. Department of Electrical Engineering and Computer Science.

Grantor dc:publisher

Massachusetts Institute of Technology

Year dc:date.issued

2018

Author dc:creator

• Yang, Tien-Ju

Advisor dc:contributor.advisor

• Vivienne Sze.

• Electrical Engineering and Computer Science.

Statement dc:rights

• MIT theses are protected by copyright. They may be viewed, downloaded, or printed from this source but further reproduction or distribution in any format is prohibited without written permission.

Licence dc:rights.uri

• http://dspace.mit.edu/handle/1721.1/7582

Language dc:language.iso

eng

Handle dc:identifier.uri

http://hdl.handle.net/1721.1/118034

OAI identifier oai:identifier

oai:dspace.mit.edu:1721.1/118034