1. Fast neural network training on a cluster of GPUs for action recognition with high accuracy
- Author
-
Fan Zhou, Barry Chen, Guojing Cong, Joshua Shapiro, Giacomo Domeniconi, and Chih-Chieh Yang
- Subjects
Artificial neural network ,Computer Networks and Communications ,Computer science ,business.industry ,Training (meteorology) ,020206 networking & telecommunications ,02 engineering and technology ,Machine learning ,computer.software_genre ,Residual neural network ,Theoretical Computer Science ,Artificial Intelligence ,Hardware and Architecture ,Distributed algorithm ,0202 electrical engineering, electronic engineering, information engineering ,Cluster (physics) ,Action recognition ,020201 artificial intelligence & image processing ,Artificial intelligence ,business ,computer ,Software - Abstract
We propose algorithms and techniques to accelerate training of deep neural networks for action recognition on a cluster of GPUs. The convergence analysis of our algorithm shows it is possible to reduce communication cost and at the same time minimize the number of iterations needed for convergence. We customize the Adam optimizer for our distributed algorithm to improve efficiency. In addition, we employ transfer-learning to further reduce training time while improving validation accuracy. For the UCF101 and HMDB51 datasets, the validation accuracies achieved are 93.1% and 67.9% respectively. With an additional end-to-end trained temporal stream, the validation accuracies achieved for UCF101 and HMDB51 are 93.47% and 81.24% respectively. As far as we know, these are the highest accuracies achieved with the two-stream approach using ResNet that does not involve computationally expensive 3D convolutions or pretraining on much larger datasets.
- Published
- 2019
- Full Text
- View/download PDF