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

Download:

Current browse context:

cs.CL

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 > Computation and Language

Title: FastSeq: Make Sequence Generation Faster

Abstract: Transformer-based models have made tremendous impacts in natural language generation. However the inference speed is a bottleneck due to large model size and intensive computing involved in auto-regressive decoding process. We develop FastSeq framework to accelerate sequence generation without accuracy loss. The proposed optimization techniques include an attention cache optimization, an efficient algorithm for detecting repeated n-grams, and an asynchronous generation pipeline with parallel I/O. These optimizations are general enough to be applicable to Transformer-based models (e.g., T5, GPT2, and UniLM). Our benchmark results on a set of widely used and diverse models demonstrate 4-9x inference speed gain. Additionally, FastSeq is easy to use with a simple one-line code change. The source code is available at this https URL
Comments: ACL 2021 Demo Track
Subjects: Computation and Language (cs.CL); Machine Learning (cs.LG)
Cite as: arXiv:2106.04718 [cs.CL]
  (or arXiv:2106.04718v2 [cs.CL] for this version)

Submission history

From: Yu Yan [view email]
[v1] Tue, 8 Jun 2021 22:25:28 GMT (2828kb,D)
[v2] Mon, 12 Jul 2021 22:24:37 GMT (2828kb,D)

Link back to: arXiv, form interface, contact.