Streaming Batch Gradient Tracking for Neural Network Training (student Abstract).

Siyuan Huang,Brian D. Hoskins,Matthew W. Daniels,Mark D. Stiles,Gina C. Adam
DOI: https://doi.org/10.1609/aaai.v34i10.7178
2020-01-01
Proceedings of the AAAI Conference on Artificial Intelligence
Abstract:Faster and more energy efficient hardware accelerators are critical for machine learning on very large datasets. The energy cost of performing vector-matrix multiplication and repeatedly moving neural network models in and out of memory motivates a search for alternative hardware and algorithms. We propose to use streaming batch principal component analysis (SBPCA) to compress batch data during training by using a rank-k approximation of the total batch update. This approach yields comparable training performance to minibatch gradient descent (MBGD) at the same batch size while reducing overall memory and compute requirements.
What problem does this paper attempt to address?