Efficient Long-Text Understanding with Short-Text Models

Maor Ivgi, Uri Shaham, Jonathan Berant

Research output: Contribution to journalArticlepeer-review

20 Scopus citations

Abstract

Transformer-based pretrained language models (LMs) are ubiquitous across natural language understanding, but cannot be applied to long sequences such as stories, scien-tific articles, and long documents due to their quadratic complexity. While a myriad of efficient transformer variants have been proposed, they are typically based on cus-tom implementations that require expensive pretraining from scratch. In this work, we pro-pose SLED: SLiding-Encoder and Decoder, a simple approach for processing long sequences that re-uses and leverages battle-tested short-text pretrained LMs. Specifically, we partition the input into overlapping chunks, encode each with a short-text LM encoder and use the pretrained decoder to fuse information across chunks (fusion-in-decoder). We illustrate through controlled experiments that SLED offers a viable strategy for long text understanding and evaluate our approach on SCROLLS, a benchmark with seven datasets across a wide range of language understanding tasks. We find that SLED is competitive with specialized models that are up to 50x larger and require a dedicated and expensive pretraining step.

Original languageEnglish
Pages (from-to)284-299
Number of pages16
JournalTransactions of the Association for Computational Linguistics
Volume11
DOIs
StatePublished - 2023

Bibliographical note

Publisher Copyright:
© 2023 Association for Computational Linguistics.

Funding

This research was partially supported by The Yan-dex Initiative for Machine Learning, the Shashua Fellowship, the Len Blavatnik and the Blavatnik Family Foundation, and the European Research Council (ERC) under the European Union Horizons 2020 research and innovation programme (grant ERC DELPHI 802800). We would also like to thank our action editor and the anonymous reviewers for their insightful suggestions and feedback. This work was completed in partial fulfillment for the Ph.D. degree of the first author.

FundersFunder number
European Union Horizons 2020 research and innovation programmeDELPHI 802800
Yan-dex Initiative for Machine Learning
Blavatnik Family Foundation
European Commission

    Fingerprint

    Dive into the research topics of 'Efficient Long-Text Understanding with Short-Text Models'. Together they form a unique fingerprint.

    Cite this