1. Incremental Learning Using a Grow-and-Prune Paradigm With Efficient Neural Networks
- Author
-
Niraj K. Jha, Hongxu Yin, and Xiaoliang Dai
- Subjects
FOS: Computer and information sciences ,Computer Science - Machine Learning ,Computer science ,Inference ,02 engineering and technology ,Neural network synthesis ,Machine learning ,computer.software_genre ,Machine Learning (cs.LG) ,0202 electrical engineering, electronic engineering, information engineering ,Computer Science (miscellaneous) ,Redundancy (engineering) ,Neural and Evolutionary Computing (cs.NE) ,computer.programming_language ,Artificial neural network ,business.industry ,Computer Science - Neural and Evolutionary Computing ,020207 software engineering ,Computer Science Applications ,Human-Computer Interaction ,Scratch ,Incremental learning ,Deep neural networks ,020201 artificial intelligence & image processing ,Artificial intelligence ,business ,computer ,MNIST database ,Information Systems - Abstract
Deep neural networks (DNNs) have become a widely deployed model for numerous machine learning applications. However, their fixed architecture, substantial training cost, and significant model redundancy make it difficult to efficiently update them to accommodate previously unseen data. To solve these problems, we propose an incremental learning framework based on a grow-and-prune neural network synthesis paradigm. When new data arrive, the neural network first grows new connections based on the gradients to increase the network capacity to accommodate new data. Then, the framework iteratively prunes away connections based on the magnitude of weights to enhance network compactness, and hence recover efficiency. Finally, the model rests at a lightweight DNN that is both ready for inference and suitable for future grow-and-prune updates. The proposed framework improves accuracy, shrinks network size, and significantly reduces the additional training cost for incoming data compared to conventional approaches, such as training from scratch and network fine-tuning. For the LeNet-300-100 (LeNet-5) neural network architectures derived for the MNIST dataset, the framework reduces training cost by up to 64% (67%), 63% (63%), and 69% (73%) compared to training from scratch, network fine-tuning, and grow-and-prune from scratch, respectively. For the ResNet-18 architecture derived for the ImageNet dataset (DeepSpeech2 for the AN4 dataset), the corresponding training cost reductions against training from scratch, network fine-tunning, and grow-and-prune from scratch are 64% (67%), 60% (62%), and 72% (71%), respectively. Our derived models contain fewer network parameters but achieve higher accuracy relative to conventional baselines.
- Published
- 2022
- Full Text
- View/download PDF