We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Accelerating recurrent neural network training using sequence bucketing and multi-GPU data parallelization

Authors: Viacheslav Khomenko (1), Oleg Shyshkov (1), Olga Radyvonenko (1), Kostiantyn Bokhan (1) ((1) Samsung R&D Institute Ukraine SRK)
Abstract: An efficient algorithm for recurrent neural network training is presented. The approach increases the training speed for tasks where a length of the input sequence may vary significantly. The proposed approach is based on the optimal batch bucketing by input sequence length and data parallelization on multiple graphical processing units. The baseline training performance without sequence bucketing is compared with the proposed solution for a different number of buckets. An example is given for the online handwriting recognition task using an LSTM recurrent neural network. The evaluation is performed in terms of the wall clock time, number of epochs, and validation loss value.
Comments: 4 pages, 5 figures, Comments, 2016 IEEE First International Conference on Data Stream Mining & Processing (DSMP), Lviv, 2016
Subjects: Machine Learning (cs.LG); Neural and Evolutionary Computing (cs.NE)
Journal reference: IEEE DSMP Lviv (2016) 100-103
DOI: 10.1109/DSMP.2016.7583516
Cite as: arXiv:1708.05604 [cs.LG]
  (or arXiv:1708.05604v1 [cs.LG] for this version)

Submission history

From: Viacheslav Khomenko [view email]
[v1] Fri, 18 Aug 2017 13:36:30 GMT (981kb)

Link back to: arXiv, form interface, contact.