CoLT5: Faster Long-Range Transformers with Conditional Computation
Not on the map. Why?
Abstract (arXiv)
Many natural language processing tasks benefit from long inputs, but processing long documents with Transformers is expensive -- not only due to quadratic attention complexity but also from applying feedforward and projection layers to every token. However, not all tokens are equally important, especially for longer documents. We propose CoLT5, a long-input Transformer model that builds on this intuition by employing conditional computation, devoting more resources to important tokens in both feedforward and attention layers. We show that CoLT5 achieves stronger performance than LongT5 with much faster training and inference, achieving SOTA on the long-input SCROLLS benchmark. Moreover, CoLT5 can effectively and tractably make use of extremely long inputs, showing strong gains up to 64k input length.
Key facts. Cited by 84 papers in the map (data through 7 Oct 2026). Builds on 36 of its 49 references that are in the map. First posted on arXiv 17 Mar 2023.
Builds on
36 of 49 references.
References · 36
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
- 2020Longformer: The Long-Document Transformercited by 3,362
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
- 2017TriviaQA: A Large Scale Distantly Supervised Challenge Dataset for Reading Comprehensioncited by 2,895
- 2020Big Bird: Transformers for Longer Sequencescited by 1,673
- 2019Generating Long Sequences with Sparse Transformerscited by 1,523
- 2020Linformer: Self-Attention with Linear Complexitycited by 1,462
- 2019PEGASUS: Pre-training with Extracted Gap-sentences for Abstractive Summarizationcited by 1,100
- 2018Adafactor: Adaptive Learning Rates with Sublinear Memory Costcited by 896
24 more
- 2019Fast Transformer Decoding: One Write-Head is All You Needcited by 700
- 2020Long Range Arena: A Benchmark for Efficient Transformerscited by 696
- 2017The NarrativeQA Reading Comprehension Challengecited by 590
- 2022Efficiently Scaling Transformer Inferencecited by 529
- 2018A Discourse-Aware Attention Model for Abstractive Summarization of Long Documentscited by 477
- 2021A Dataset of Information-Seeking Questions and Answers Anchored in Research Paperscited by 475
- 2022ST-MoE: Designing Stable and Transferable Sparse Expert Modelscited by 391
- 2021QMSum: A New Benchmark for Query-based Multi-domain Meeting Summarizationcited by 363
- 2021Efficient Attentions for Long Document Summarizationcited by 342
- 2022Confident Adaptive Language Modelingcited by 297
- 2022UL2: Unifying Language Learning Paradigmscited by 292
- 2021LongT5: Efficient Text-To-Text Transformer for Long Sequencescited by 232
- 2021QuALITY: Question Answering with Long Input Texts, Yes!cited by 204
- 2022Scaling Up Models and Data with textttt5x and textttseqiocited by 187
- 2022SCROLLS: Standardized CompaRison Over Long Language Sequencescited by 170
- 2021ContractNLI: A Dataset for Document-level Natural Language Inference for Contractscited by 145
- 2021SummScreen: A Dataset for Abstractive Screenplay Summarizationcited by 80
- 2021KG-FiD: Infusing Knowledge Graph in Fusion-in-Decoder for Open-Domain Question Answeringcited by 66
- 2022FiDO: Fusion-in-Decoder optimized for stronger performance and faster inferencecited by 31
- 2021Rider: Reader-Guided Passage Reranking for Open-Domain Question Answeringcited by 28
- 2018Adaptive Document Retrieval for Deep Question Answeringcited by 16
- 2021ReadTwice: Reading Very Large Documents with Memoriescited by 14
- 2022Multi-Vector Retrieval as Sparse Alignmentcited by 12
- 2022Can Open-Domain QA Reader Utilize External Knowledge Efficiently like Humans?cited by 10
Built on by
Citing papers per year: 2023: 32; 2024: 21; 2025: 21; 2026: 10.
84 papers.
Sort:
By area:
- 2023The Rise and Potential of Large Language Model Based Agents: A Surveycited by 1,422
- 2023LongBench: A Bilingual, Multitask Benchmark for Long Context Understandingcited by 1,378
- 2023H2O: Heavy-Hitter Oracle for Efficient Generative Inference of Large Language Modelscited by 805
- 2023Extending Context Window of Large Language Models via Positional Interpolationcited by 721
- 2023A Comprehensive Overview of Large Language Modelscited by 639
- 2024RAPTOR: Recursive Abstractive Processing for Tree-Organized Retrievalcited by 519
- 2023Large Language Models Are Zero-Shot Time Series Forecasterscited by 491
- 2023Adapting Language Models to Compress Contextscited by 350
- 2023A Real-World WebAgent with Planning, Long Context Understanding, and Program Synthesiscited by 350
- 2023RepoBench: Benchmarking Repository-Level Code Auto-Completion Systemscited by 346
- 2023Challenges and Applications of Large Language Modelscited by 251
- 2023LongNet: Scaling Transformers to 1,000,000,000 Tokenscited by 201
38 more
- 2025MoBA: Mixture of Block Attention for Long-Context LLMscited by 188
- 2023Do All Languages Cost the Same? Tokenization in the Era of Commercial Language Modelscited by 186
- 2023ZeroSCROLLS: A Zero-Shot Benchmark for Long Text Understandingcited by 175
- 2023Embers of Autoregression: Understanding Large Language Models Through the Problem They are Trained to Solvecited by 175
- 2025Kimi Linear: An Expressive, Efficient Attention Architecturecited by 169
- 2023Focused Transformer: Contrastive Training for Context Scalingcited by 133
- 2023Walking Down the Memory Maze: Beyond Context Limit through Interactive Readingcited by 129
- 2024Sequoia: Scalable, Robust, and Hardware-aware Speculative Decodingcited by 103
- 2023Scaling Transformer to 1M tokens and beyond with RMTcited by 97
- 2023Conditional Adapters: Parameter-efficient Transfer Learning with Fast Inferencecited by 69
- 2024Sparser is Faster and Less is More: Efficient Sparse Attention for Long-Range Transformerscited by 51
- 2024Challenges in Deploying Long-Context Transformers: A Theoretical Peak Performance Analysiscited by 44
- 2023No Train No Gain: Revisiting Efficient Training Algorithms For Transformer-based Language Modelscited by 41
- 2024The What, Why, and How of Context Length Extension Techniques in Large Language Models -- A Detailed Surveycited by 32
- 2024DAPE: Data-Adaptive Positional Encoding for Length Extrapolationcited by 30
- 2025Adaptive Layer-skipping in Pre-trained LLMscited by 24
- 2023Dodo: Dynamic Contextual Compression for Decoder-only LMscited by 23
- 2024GRAM: Global Reasoning for Multi-Page VQAcited by 23
- 2024Two Stones Hit One Bird: Bilevel Positional Encoding for Better Length Extrapolationcited by 23
- 2025System-1.5 Reasoning: Traversal in Language and Latent Spaces with Dynamic Shortcutscited by 23
- 2025CAM: A Constructivist View of Agentic Memory for LLM-Based Reading Comprehensioncited by 23
- 2023ByteSized32: A Corpus and Challenge Task for Generating Task-Specific World Models Expressed as Text Gamescited by 22
- 2023Learning to Skip for Language Modelingcited by 19
- 2025KV-Distill: Nearly Lossless Learnable Context Compression for LLMscited by 19
- 2023Functional Interpolation for Relative Positions Improves Long Context Transformerscited by 18
- 2025Instruction-Following Pruning for Large Language Modelscited by 14
- 2024Routing Experts: Learning to Route Dynamic Experts in Multi-modal Large Language Modelscited by 13
- 2023Long-Horizon Dialogue Understanding for Role Identification in the Game of Avalon with Large Language Modelscited by 12
- 2024Mixture-of-Modules: Reinventing Transformers as Dynamic Assemblies of Modulescited by 11
- 2023Optimizing Retrieval-augmented Reader Models via Token Eliminationcited by 10
- 2024Conditional computation in neural networks: principles and research trendscited by 10
- 2024DAPE V2: Process Attention Score as Feature Map for Length Extrapolationcited by 10
- 2024MDCure: A Scalable Pipeline for Multi-Document Instruction-Followingcited by 10
- 2025HoPE: Hyperbolic Rotary Positional Encoding for Stable Long-Range Dependency Modeling in Large Language Modelscited by 10
- 2023Chunk, Align, Select: A Simple Long-sequence Processing Method for Transformerscited by 9
- 2024DLO: Dynamic Layer Operation for Efficient Vertical Scaling of LLMscited by 9
- 2023Drilling Down into the Discourse Structure with LLMs for Long Document Question Answeringcited by 8
- 2023GLIMMER: generalized late-interaction memory rerankercited by 7
50 most cited of 84 All 84 (JSON)
- 2026KV2: A Self-Refining KV Cache2 Oct 2026
- 2026On-Demand Attention: Language Models Know When to Recall17 Sep 2026
- 2026LoGo: Token-Level Dynamic Local-Global Attention30 Aug 2026
- 2026From Attention Masks to Inert Zero-Vector Tokens: OAttention and O-Closure for Token Dynamics21 Aug 2026
- 2026TriRoute: Unified Learned Routing for Joint Adaptive Attention, Experts, and KV-Cache Allocation7 Jul 2026
- 2026Persona-Pruner: Sculpting Lightweight Models for Role-Playing12 Jun 2026
- 2026LoopMoE: Unifying Iterative Computation with Mixture-of-Experts for Language Modeling3 Jun 2026
- 2026DTCRS: Dynamic Tree Construction for Recursive Summarization8 Apr 2026
- 2026Expert Threshold Routing for Autoregressive Language Modeling with Dynamic Computation Allocation and Load Balancing12 Mar 2026
- 2026LoRA-Drop: Temporal LoRA Decoding for Efficient LLM Inference5 Jan 2026
- 2025Learning When Not to Attend Globally27 Dec 2025
- 2025Kimi Linear: An Expressive, Efficient Attention Architecture30 Oct 2025
38 more
- 2025Alleviating Forgetfulness of Linear Attention by Hybrid Sparse Attention and Contextualized Learnable Token Eviction23 Oct 2025
- 2025Mirror Speculative Decoding: Breaking the Serial Barrier in LLM Inference15 Oct 2025
- 2025CAM: A Constructivist View of Agentic Memory for LLM-Based Reading Comprehension7 Oct 2025
- 2025ReSSFormer: A Recursive Sparse Structured Transformer for Scalable and Long-Context Reasoning2 Oct 2025
- 2025Collaborative Compression for Large-Scale MoE Deployment on Edge30 Sep 2025
- 2025ResFormer: All-Time Reservoir Memory for Long Sequence Classification28 Sep 2025
- 2025HoPE: Hyperbolic Rotary Positional Encoding for Stable Long-Range Dependency Modeling in Large Language Models5 Sep 2025
- 2025SAS: Simulated Attention Score10 Jul 2025
- 2025Spark Transformer: Reactivating Sparsity in FFN and Attention7 Jun 2025
- 2025MiniLongBench: The Low-cost Long Context Understanding Benchmark for Large Language Models26 May 2025
- 2025System-1.5 Reasoning: Traversal in Language and Latent Spaces with Dynamic Shortcuts25 May 2025
- 2025S'MoRE: Structural Mixture of Residual Experts for Parameter-Efficient LLM Fine-tuning8 Apr 2025
- 2025Adaptive Layer-skipping in Pre-trained LLMs31 Mar 2025
- 2025CASE -- Condition-Aware Sentence Embeddings for Conditional Semantic Textual Similarity Measurement21 Mar 2025
- 2025KV-Distill: Nearly Lossless Learnable Context Compression for LLMs13 Mar 2025
- 2025Self-Adjust Softmax25 Feb 2025
- 2025MoBA: Mixture of Block Attention for Long-Context LLMs18 Feb 2025
- 2025M2R2: Mixture of Multi-Rate Residuals for Efficient Transformer Inference4 Feb 2025
- 2025Instruction-Following Pruning for Large Language Models3 Jan 2025
- 2024MDCure: A Scalable Pipeline for Multi-Document Instruction-Following30 Oct 2024
- 2024Skipping Computations in Multimodal LLMs12 Oct 2024
- 2024DAPE V2: Process Attention Score as Feature Map for Length Extrapolation7 Oct 2024
- 2024Enhancing and Accelerating Large Language Models via Instruction-Aware Contextual Compression28 Aug 2024
- 2024Routing Experts: Learning to Route Dynamic Experts in Multi-modal Large Language Models19 Jul 2024
- 2024HDT: Hierarchical Document Transformer11 Jul 2024
- 2024Mixture-of-Modules: Reinventing Transformers as Dynamic Assemblies of Modules9 Jul 2024
- 2024DLO: Dynamic Layer Operation for Efficient Vertical Scaling of LLMs3 Jul 2024
- 2024Sparser is Faster and Less is More: Efficient Sparse Attention for Long-Range Transformers24 Jun 2024
- 2024QRMeM: Unleash the Length Limitation through Question then Reflection Memory Mechanism19 Jun 2024
- 2024Taking a Deep Breath: Enhancing Language Modeling of Large Language Models with Sentinel Tokens16 Jun 2024
- 2024SUBLLM: A Novel Efficient Architecture with Token Sequence Subsampling for LLM3 Jun 2024
- 2024DAPE: Data-Adaptive Positional Encoding for Length Extrapolation23 May 2024
- 2024Challenges in Deploying Long-Context Transformers: A Theoretical Peak Performance Analysis14 May 2024
- 2024Conditional computation in neural networks: principles and research trends12 Mar 2024
- 2024Sequoia: Scalable, Robust, and Hardware-aware Speculative Decoding19 Feb 2024
- 2024Investigating Recurrent Transformers with Dynamic Halt1 Feb 2024
- 2024RAPTOR: Recursive Abstractive Processing for Tree-Organized Retrieval31 Jan 2024
- 2024Two Stones Hit One Bird: Bilevel Positional Encoding for Better Length Extrapolation29 Jan 2024
50 newest of 84 All 84 (JSON)
- 2025MoBA: Mixture of Block Attention for Long-Context LLMscited by 188
- 2024Conditional computation in neural networks: principles and research trendscited by 10
- 2024DLO: Dynamic Layer Operation for Efficient Vertical Scaling of LLMscited by 9
- 2026LoopMoE: Unifying Iterative Computation with Mixture-of-Experts for Language Modelingcited by 7
- 2025S'MoRE: Structural Mixture of Residual Experts for Parameter-Efficient LLM Fine-tuningcited by 3
- 2025Collaborative Compression for Large-Scale MoE Deployment on Edgecited by 3
- 2026Expert Threshold Routing for Autoregressive Language Modeling with Dynamic Computation Allocation and Load Balancingcited by 3
- 2025M2R2: Mixture of Multi-Rate Residuals for Efficient Transformer Inferencecited by 1
- 2026TriRoute: Unified Learned Routing for Joint Adaptive Attention, Experts, and KV-Cache Allocationcited by 0
- 2026TriRoute: Unified Learned Routing for Joint Adaptive Attention, Experts, and KV-Cache Allocation7 Jul 2026
- 2026LoopMoE: Unifying Iterative Computation with Mixture-of-Experts for Language Modeling3 Jun 2026
- 2026Expert Threshold Routing for Autoregressive Language Modeling with Dynamic Computation Allocation and Load Balancing12 Mar 2026
- 2025Collaborative Compression for Large-Scale MoE Deployment on Edge30 Sep 2025
- 2025S'MoRE: Structural Mixture of Residual Experts for Parameter-Efficient LLM Fine-tuning8 Apr 2025
- 2025MoBA: Mixture of Block Attention for Long-Context LLMs18 Feb 2025
- 2025M2R2: Mixture of Multi-Rate Residuals for Efficient Transformer Inference4 Feb 2025
- 2024DLO: Dynamic Layer Operation for Efficient Vertical Scaling of LLMs3 Jul 2024
- 2024Conditional computation in neural networks: principles and research trends12 Mar 2024
- 2023H2O: Heavy-Hitter Oracle for Efficient Generative Inference of Large Language Modelscited by 805
- 2024Sequoia: Scalable, Robust, and Hardware-aware Speculative Decodingcited by 103
- 2025KV-Distill: Nearly Lossless Learnable Context Compression for LLMscited by 19
- 2025Alleviating Forgetfulness of Linear Attention by Hybrid Sparse Attention and Contextualized Learnable Token Evictioncited by 3
- 2025Mirror Speculative Decoding: Breaking the Serial Barrier in LLM Inferencecited by 1
- 2026LoRA-Drop: Temporal LoRA Decoding for Efficient LLM Inferencecited by 1
- 2026KV2: A Self-Refining KV Cachecited by 1
- 2026KV2: A Self-Refining KV Cache2 Oct 2026
- 2026LoRA-Drop: Temporal LoRA Decoding for Efficient LLM Inference5 Jan 2026
- 2025Alleviating Forgetfulness of Linear Attention by Hybrid Sparse Attention and Contextualized Learnable Token Eviction23 Oct 2025
- 2025Mirror Speculative Decoding: Breaking the Serial Barrier in LLM Inference15 Oct 2025
- 2025KV-Distill: Nearly Lossless Learnable Context Compression for LLMs13 Mar 2025
- 2024Sequoia: Scalable, Robust, and Hardware-aware Speculative Decoding19 Feb 2024
- 2023H2O: Heavy-Hitter Oracle for Efficient Generative Inference of Large Language Models24 Jun 2023
- 2023The Rise and Potential of Large Language Model Based Agents: A Surveycited by 1,422
- 2024Challenges in Deploying Long-Context Transformers: A Theoretical Peak Performance Analysiscited by 44
- 2025CAM: A Constructivist View of Agentic Memory for LLM-Based Reading Comprehensioncited by 23
- 2026On-Demand Attention: Language Models Know When to Recallcited by 0
- 2026On-Demand Attention: Language Models Know When to Recall17 Sep 2026
- 2025CAM: A Constructivist View of Agentic Memory for LLM-Based Reading Comprehension7 Oct 2025
- 2024Challenges in Deploying Long-Context Transformers: A Theoretical Peak Performance Analysis14 May 2024
- 2023The Rise and Potential of Large Language Model Based Agents: A Survey14 Sep 2023
- 2023LongBench: A Bilingual, Multitask Benchmark for Long Context Understandingcited by 1,378
- 2023Extending Context Window of Large Language Models via Positional Interpolationcited by 721
- 2023A Comprehensive Overview of Large Language Modelscited by 639
- 2023Large Language Models Are Zero-Shot Time Series Forecasterscited by 491
- 2023Adapting Language Models to Compress Contextscited by 350
- 2023A Real-World WebAgent with Planning, Long Context Understanding, and Program Synthesiscited by 350
- 2023RepoBench: Benchmarking Repository-Level Code Auto-Completion Systemscited by 346
- 2023Challenges and Applications of Large Language Modelscited by 251
- 2023LongNet: Scaling Transformers to 1,000,000,000 Tokenscited by 201
- 2023Do All Languages Cost the Same? Tokenization in the Era of Commercial Language Modelscited by 186
- 2023ZeroSCROLLS: A Zero-Shot Benchmark for Long Text Understandingcited by 175
- 2023Embers of Autoregression: Understanding Large Language Models Through the Problem They are Trained to Solvecited by 175
12 most cited of 55 All 55 (JSON)
- 2026LoGo: Token-Level Dynamic Local-Global Attention30 Aug 2026
- 2026From Attention Masks to Inert Zero-Vector Tokens: OAttention and O-Closure for Token Dynamics21 Aug 2026
- 2026Persona-Pruner: Sculpting Lightweight Models for Role-Playing12 Jun 2026
- 2025Learning When Not to Attend Globally27 Dec 2025
- 2025Kimi Linear: An Expressive, Efficient Attention Architecture30 Oct 2025
- 2025ResFormer: All-Time Reservoir Memory for Long Sequence Classification28 Sep 2025
- 2025HoPE: Hyperbolic Rotary Positional Encoding for Stable Long-Range Dependency Modeling in Large Language Models5 Sep 2025
- 2025SAS: Simulated Attention Score10 Jul 2025
- 2025Spark Transformer: Reactivating Sparsity in FFN and Attention7 Jun 2025
- 2025MiniLongBench: The Low-cost Long Context Understanding Benchmark for Large Language Models26 May 2025
- 2025Adaptive Layer-skipping in Pre-trained LLMs31 Mar 2025
- 2025CASE -- Condition-Aware Sentence Embeddings for Conditional Semantic Textual Similarity Measurement21 Mar 2025
12 newest of 55 All 55 (JSON)