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

Download:

Current browse context:

cs.AR

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 ScienceWISE logo

Computer Science > Hardware Architecture

Title: Spartus: A 9.4 TOp/s FPGA-based LSTM Accelerator Exploiting Spatio-Temporal Sparsity

Abstract: Long Short-Term Memory (LSTM) recurrent networks are frequently used for tasks involving time-sequential data such as speech recognition. Unlike previous LSTM accelerators that either exploit spatial weight sparsity or temporal activation sparsity, this paper proposes a new accelerator called "Spartus" that exploits spatio-temporal sparsity to achieve ultra-low latency inference. Spatial sparsity is induced using a new Column-Balanced Targeted Dropout (CBTD) structured pruning method, producing structured sparse weight matrices for a balanced workload. The pruned networks running on Spartus hardware achieve weight sparsity levels of up to 96% and 94% with negligible accuracy loss on the TIMIT and the Librispeech datasets. To induce temporal sparsity in LSTM, we extend the previous DeltaGRU method to the DeltaLSTM method. Combining spatio-temporal sparsity with CBTD and DeltaLSTM saves on weight memory access and associated arithmetic operations. The Spartus architecture is scalable and supports real-time online speech recognition when implemented on small and large FPGAs. Spartus per-sample latency for a single DeltaLSTM layer of 1024 neurons averages 1 us. Exploiting spatio-temporal sparsity on our test LSTM network using the TIMIT dataset leads to 46X speedup of Spartus over its theoretical hardware performance to achieve 9.4 TOp/s effective batch-1 throughput and 1.1 TOp/s/W power efficiency.
Comments: Accepted for publication in IEEE Transactions on Neural Networks and Learning Systems, 2022
Subjects: Hardware Architecture (cs.AR); Artificial Intelligence (cs.AI); Computer Vision and Pattern Recognition (cs.CV); Machine Learning (cs.LG)
Journal reference: IEEE Transactions on Neural Networks and Learning Systems, 2022
DOI: 10.1109/TNNLS.2022.3180209
Cite as: arXiv:2108.02297 [cs.AR]
  (or arXiv:2108.02297v5 [cs.AR] for this version)

Submission history

From: Chang Gao [view email]
[v1] Wed, 4 Aug 2021 22:02:14 GMT (3158kb,D)
[v2] Wed, 11 Aug 2021 17:09:44 GMT (3244kb,D)
[v3] Fri, 20 Aug 2021 14:29:37 GMT (2256kb,D)
[v4] Tue, 29 Mar 2022 08:43:37 GMT (3837kb,D)
[v5] Mon, 13 Jun 2022 14:02:11 GMT (1532kb,D)

Link back to: arXiv, form interface, contact.