Abstract:Deep neural networks have yielded superior performance in many applications; however, the gradient computation in a deep model with millions of instances lead to a lengthy training process even with modern GPU/TPU hardware acceleration. In this paper, we propose AutoAssist, a simple framework to accelerate training of a deep neural network. Typically, as the training procedure evolves, the amount of improvement in the current model by a stochastic gradient update on each instance varies dynamically. In AutoAssist, we utilize this fact and design a simple instance shrinking operation, which is used to filter out instances with relatively low marginal improvement to the current model; thus the computationally intensive gradient computations are performed on informative instances as much as possible. We prove that the proposed technique outperforms vanilla SGD with existing importance sampling approaches for linear SVM problems, and establish an O(1/k) convergence for strongly convex problems. In order to apply the proposed techniques to accelerate training of deep models, we propose to jointly train a very lightweight Assistant network in addition to the original deep network referred to as Boss. The Assistant network is designed to gauge the importance of a given instance with respect to the current Boss such that a shrinking operation can be applied in the batch generator. With careful design, we train the Boss and Assistant in a nonblocking and asynchronous fashion such that overhead is minimal. We demonstrate that AutoAssist reduces the number of epochs by 40% for training a ResNet to reach the same test accuracy on an image classification data set and saves 30% training time needed for a transformer model to yield the same BLEU scores on a translation dataset.

A Pseudoinverse Incremental Algorithm For Fast Training Deep Neural Networks With Application To Spectra Pattern Recognition

Fast Image Recognition with Gabor Filter and Pseudoinverse Learning AutoEncoders

Efficient Incremental Training for Deep Convolutional Neural Networks

An Efficient Learning Algorithm for Direct Training Deep Spiking Neural Networks

Incremental updating method for big data feature learning

Deep Neural Networks with Local Connectivity and Its Application to Astronomical Spectral Data.

Pseudoinverse Learning Algorithom for Fast Sparse Autoencoder Training.

Intelligent gradient amplification for deep neural networks

Incremental learning with neural networks for computer vision: a survey

Accelerated Training via Incrementally Growing Neural Networks using Variance Transfer and Learning Rate Adaptation

A Strategy for an Uncompromising Incremental Learner

Gradient amplification: An efficient way to train deep neural networks

Broad Learning System: An Effective and Efficient Incremental Learning System Without the Need for Deep Architecture

Speeding up the Training of Neural Networks with the One-Step Procedure

Reconstructing Deep Neural Networks: Unleashing the Optimization Potential of Natural Gradient Descent

AutoAssist: A Framework to Accelerate Training of Deep Neural Networks

PILAE: A Non-gradient Descent Learning Scheme for Deep Feedforward Neural Networks

Deep learning in spectral analysis: Modeling and imaging

A Gaussian Mixture Framework for Incremental Nonparametric Regression with Topology Learning Neural Networks

Exponential decay sine wave learning rate for fast deep neural network training

A New Automated Spectral Feature Extraction Method and Its Application in Spectral Classification and Defective Spectra Recovery