Parallelizing Legendre Memory Unit Training
dc.contributor.author | Chilkuri, Narsimha Reddy | |
dc.date.accessioned | 2021-07-14T19:12:58Z | |
dc.date.available | 2021-07-14T19:12:58Z | |
dc.date.issued | 2021-07-14 | |
dc.date.submitted | 2021-07-07 | |
dc.description.abstract | Recently, a new recurrent neural network (RNN) named the Legendre Memory Unit (LMU) was proposed and shown to achieve state-of-the-art performance on several benchmark datasets. Here we leverage the linear time-invariant (LTI) memory component of the LMU to construct a simplified variant that can be parallelized during training (and yet executed as an RNN during inference), resulting in up to 200 times faster training. We note that our efficient parallelizing scheme is general and is applicable to any deep network whose recurrent components are LTI systems. We demonstrate the improved accuracy and decreased parameter count of our new architecture compared to the original LMU and a variety of published LSTM and transformer networks across seven benchmarks. For instance, our LMU sets a new state-of-the-art result on psMNIST, and uses half the parameters while outperforming DistilBERT and LSTM models on IMDB sentiment analysis. | en |
dc.identifier.uri | http://hdl.handle.net/10012/17142 | |
dc.language.iso | en | en |
dc.pending | false | |
dc.publisher | University of Waterloo | en |
dc.subject | Deep Learning | en |
dc.subject | Recurrent Neural Network | en |
dc.subject | Legendre Memory Unit | en |
dc.title | Parallelizing Legendre Memory Unit Training | en |
dc.type | Master Thesis | en |
uws-etd.degree | Master of Applied Science | en |
uws-etd.degree.department | Systems Design Engineering | en |
uws-etd.degree.discipline | System Design Engineering | en |
uws-etd.degree.grantor | University of Waterloo | en |
uws-etd.embargo.terms | 0 | en |
uws.contributor.advisor | Eliasmith, Chris | |
uws.contributor.affiliation1 | Faculty of Engineering | en |
uws.peerReviewStatus | Unreviewed | en |
uws.published.city | Waterloo | en |
uws.published.country | Canada | en |
uws.published.province | Ontario | en |
uws.scholarLevel | Graduate | en |
uws.typeOfResource | Text | en |