Neural network simplification using a progressive barrier based approach
Name
1051458923-MIT.pdf
Description
Full printable version
Size
6.48 MB
Format
Adobe PDF
Checksum (MD5)
7826070be028266b703d9e67f6705276
Author(s)
Yang, Tien-Ju
Advisor(s)
Vivienne Sze.
Date Issued
2018
Publisher
Massachusetts Institute of Technology
Abstract
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.
Description
Thesis: S.M., Massachusetts Institute of Technology, Department of Electrical Engineering and Computer Science, 2018.
Cataloged from PDF version of thesis.
Includes bibliographical references (pages 49-57).
Subjects
Electrical Engineering and Computer Science.
MIT Department
Massachusetts Institute of Technology. Department of Electrical Engineering and Computer Science
Terms of Use
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.
Persistent DSpace Link
