Blockwise Self-Attention for Long Document Understanding
Not on the map. Why?
Abstract (arXiv)
We present BlockBERT, a lightweight and efficient BERT model for better modeling long-distance dependencies. Our model extends BERT by introducing sparse block structures into the attention matrix to reduce both memory consumption and training/inference time, which also enables attention heads to capture either short- or long-range contextual information. We conduct experiments on language model pre-training and several benchmark question answering datasets with various paragraph lengths. BlockBERT uses 18.7-36.1% less memory and 12.0-25.1% less time to learn the model. During testing, BlockBERT saves 27.8% inference time, while having comparable and sometimes better prediction accuracy, compared to an advanced BERT-based model, RoBERTa.
Key facts. Cited by 159 papers in the map (data through 14 Oct 2026). First posted on arXiv 7 Nov 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 33; 2022: 38; 2023: 33; 2024: 40; 2025: 9; 2026: 6.
159 papers.
Sort:
By area:
- 2022FEDformer: Frequency Enhanced Decomposed Transformer for Long-term Series Forecastingcited by 1,210
- 2021Random Feature Attentioncited by 354
- 2022Diagonal State Spaces are as Effective as Structured State Spacescited by 350
- 2021FNet: Mixing Tokens with Fourier Transformscited by 347
- 2021XCiT: Cross-Covariance Image Transformerscited by 338
- 2022Long Range Language Modeling via Gated State Spacescited by 265
- 2023LongLoRA: Efficient Fine-tuning of Long-Context Large Language Modelscited by 244
- 2021Multi-Scale Vision Longformer: A New Vision Transformer for High-Resolution Image Encodingcited by 207
- 2025MoBA: Mixture of Block Attention for Long-Context LLMscited by 198
- 2023Efficient Large Language Models: A Surveycited by 186
- 2022Separable Self-attention for Mobile Vision Transformerscited by 163
- 2024Eagle and Finch: RWKV with Matrix-Valued States and Dynamic Recurrencecited by 159
38 more
- 2021Self-attention Does Not Need O(n2) Memorycited by 154
- 2022FiLM: Frequency improved Legendre Memory Model for Long-term Time Series Forecastingcited by 147
- 2021Lawformer: A Pre-trained Language Model for Chinese Legal Long Documentscited by 116
- 2021Long-Short Transformer: Efficient Transformers for Language and Visioncited by 102
- 2021Finetuning Pretrained Transformers into RNNscited by 96
- 2021Luna: Linear Unified Nested Attentioncited by 96
- 2023Modeling Dense Multimodal Interactions Between Biological Pathways and Histology for Survival Predictioncited by 92
- 2024Beyond the Limits: A Survey of Techniques to Extend the Context Length in Large Language Modelscited by 76
- 2021A Practical Survey on Faster and Lighter Transformerscited by 73
- 2023Advancing Transformer Architecture in Long-Context Large Language Models: A Comprehensive Surveycited by 72
- 2021Token Pooling in Vision Transformerscited by 69
- 2025Training-free and Adaptive Sparse Attention for Efficient Long Video Generationcited by 69
- 2022Single Stage Virtual Try-on via Deformable Attention Flowscited by 65
- 2024Model Compression and Efficient Inference for Large Language Models: A Surveycited by 64
- 2021Skyformer: Remodel Self-Attention with Gaussian Kernel and Nystr"om Methodcited by 63
- 2022ClusterGNN: Cluster-based Coarse-to-Fine Graph Neural Network for Efficient Feature Matchingcited by 62
- 2021Poolingformer: Long Document Modeling with Pooling Attentioncited by 60
- 2021Memory-efficient Transformers via Top-k Attentioncited by 60
- 2024Sparser is Faster and Less is More: Efficient Sparse Attention for Long-Range Transformerscited by 52
- 2021Stable, Fast and Accurate: Kernelized Attention with Relative Positional Encodingcited by 51
- 2022Revisiting Transformer-based Models for Long Document Classificationcited by 49
- 2021Efficient Visual Tracking with Exemplar Transformerscited by 48
- 2021H-Transformer-1D: Fast One-Dimensional Hierarchical Attention for Sequencescited by 47
- 2023A Survey on Long Text Modeling with Transformerscited by 47
- 2021SparseBERT: Rethinking the Importance Analysis in Self-attentioncited by 46
- 2024Is Mamba Capable of In-Context Learning?cited by 45
- 2024Post-Training Sparse Attention with Double Sparsitycited by 45
- 2023On Sparse Modern Hopfield Modelcited by 42
- 2022ViTCoD: Vision Transformer Acceleration via Dedicated Algorithm and Accelerator Co-Designcited by 41
- 2021FLAT: An Optimized Dataflow for Mitigating Attention Bottleneckscited by 39
- 2022Efficient Long Sequence Modeling via State Space Augmented Transformercited by 37
- 2024Foundation Models for Music: A Surveycited by 37
- 2023How Much Temporal Long-Term Context is Needed for Action Segmentation?cited by 36
- 2021Hi-Transformer: Hierarchical Interactive Transformer for Efficient and Effective Long Document Modelingcited by 35
- 2024The What, Why, and How of Context Length Extension Techniques in Large Language Models -- A Detailed Surveycited by 33
- 2024AI Safety in Generative AI Large Language Models: A Surveycited by 33
- 2022Enable Deep Learning on Mobile Devices: Methods, Systems, and Applicationscited by 32
- 2023OcTr: Octree-based Transformer for 3D Object Detectioncited by 32
50 most cited of 159 All 159 (JSON)
- 2026Titan-Llama: Retrofitting Pre-trained LLMs with External Neural Memory Modules13 Oct 2026
- 2026BCMT: Blockwise Causal Memory Transformer6 Jul 2026
- 2026Stream-CQSA: Exact Out-of-Memory Recovery for Attention22 Apr 2026
- 2026Beyond Pairwise Attention: Higher-Order Modular Attention for Efficient Sequence Learning11 Mar 2026
- 2026LLM-CoOpt: A Co-Design and Optimization Framework for Efficient LLM Inference on Heterogeneous Platforms10 Feb 2026
- 2026A Provable Expressiveness Hierarchy in Hybrid Linear-Full Attention2 Feb 2026
- 2025Trellis: Learning to Compress Key-Value Memory in Attention Models29 Dec 2025
- 2025Optimizing Mixture of Block Attention14 Nov 2025
- 2025Compact Attention: Exploiting Structured Spatio-Temporal Sparsity for Fast Video Generation18 Aug 2025
- 2025When Does Divide and Conquer Work for Long Context LLM? A Noise Decomposition Framework19 Jun 2025
- 2025Generalized Neighborhood Attention: Multi-dimensional Sparse Attention at the Speed of Light23 Apr 2025
- 2025ARLED: Leveraging LED-based ARMAN Model for Abstractive Summarization of Persian Long Documents13 Mar 2025
38 more
- 2025Training-free and Adaptive Sparse Attention for Efficient Long Video Generation28 Feb 2025
- 2025MoBA: Mixture of Block Attention for Long-Context LLMs18 Feb 2025
- 2025Surface Vision Mamba: Leveraging Bidirectional State Space Model for Efficient Spherical Manifold Representation24 Jan 2025
- 2024V2PE: Improving Multimodal Long-Context Capability of Vision-Language Models with Variable Visual Position Encoding12 Dec 2024
- 2024kNN Attention Demystified: A Theoretical Exploration for Scalable Transformers6 Nov 2024
- 2024Rethinking Transformer for Long Contextual Histopathology Whole Slide Image Analysis18 Oct 2024
- 2024A Little Goes a Long Way: Efficient Long Context Training and Inference with Partial Contexts2 Oct 2024
- 2024GLMHA A Guided Low-rank Multi-Head Self-Attention for Efficient Image Restoration and Spectral Reconstruction1 Oct 2024
- 2024Generative AI-driven forecasting of oil production24 Sep 2024
- 2024Foundation Models for Music: A Survey26 Aug 2024
- 2024Macformer: Transformer with Random Maclaurin Feature Attention21 Aug 2024
- 2024ELASTIC: Efficient Linear Attention for Sequential Interest Compression18 Aug 2024
- 2024S3Attention: Improving Long Sequence Attention with Smoothed Skeleton Sketching16 Aug 2024
- 2024Ex3: Automatic Novel Writing by Extracting, Excelsior and Expanding16 Aug 2024
- 2024Post-Training Sparse Attention with Double Sparsity11 Aug 2024
- 2024S2-Attention: Hardware-Aware Context Sharding Among Attention Heads25 Jul 2024
- 2024SPLAT: A framework for optimised GPU code-generation for SParse reguLar ATtention23 Jul 2024
- 2024TorchGT: A Holistic System for Large-scale Graph Transformer Training19 Jul 2024
- 2024Hybrid Dynamic Pruning: A Pathway to Efficient Transformer Inference17 Jul 2024
- 2024DeepGate3: Towards Scalable Circuit Representation Learning15 Jul 2024
- 2024AI Safety in Generative AI Large Language Models: A Survey6 Jul 2024
- 2024Let the Code LLM Edit Itself When You Edit the Code3 Jul 2024
- 2024Sparser is Faster and Less is More: Efficient Sparse Attention for Long-Range Transformers24 Jun 2024
- 2024A Primal-Dual Framework for Transformers and Neural Networks19 Jun 2024
- 2024Short-Long Convolutions Help Hardware-Efficient Linear Attention to Focus on Long Sequences12 Jun 2024
- 2024Activator: GLU Activation Function as the Core Component of a Vision Transformer24 May 2024
- 2024Equipping Transformer with Random-Access Reading for Long-Context Understanding21 May 2024
- 2024The CAP Principle for LLM Serving: A Survey of Long-Context Large Language Model Serving18 May 2024
- 2024Efficient Heterogeneous Large Language Model Decoding with Model-Attention Disaggregation3 May 2024
- 2024Hierarchical Context Merging: Better Long Context Understanding for Pre-trained LLMs16 Apr 2024
- 2024Eagle and Finch: RWKV with Matrix-Valued States and Dynamic Recurrence8 Apr 2024
- 2024Nonparametric Modern Hopfield Models5 Apr 2024
- 2024BiSHop: Bi-Directional Cellular Learning for Tabular Data with Generalized Sparse Modern Hopfield Model4 Apr 2024
- 2024NiNformer: A Network in Network Transformer with Token Mixing Generated Gating Function4 Mar 2024
- 2024Orchid: Flexible and Data-Dependent Convolution for Sequence Modeling28 Feb 2024
- 2024Multimodal Transformer With a Low-Computational-Cost Guarantee23 Feb 2024
- 2024Do Efficient Transformers Really Save Computation?21 Feb 2024
- 2024Perceiving Longer Sequences With Bi-Directional Cross-Attention Transformers19 Feb 2024
50 newest of 159 All 159 (JSON)
- 2024Let the Code LLM Edit Itself When You Edit the Codecited by 10
- 2024A Little Goes a Long Way: Efficient Long Context Training and Inference with Partial Contextscited by 9
- 2024The CAP Principle for LLM Serving: A Survey of Long-Context Large Language Model Servingcited by 3
- 2026LLM-CoOpt: A Co-Design and Optimization Framework for Efficient LLM Inference on Heterogeneous Platformscited by 0
- 2026Titan-Llama: Retrofitting Pre-trained LLMs with External Neural Memory Modulescited by 0
- 2026Titan-Llama: Retrofitting Pre-trained LLMs with External Neural Memory Modules13 Oct 2026
- 2026LLM-CoOpt: A Co-Design and Optimization Framework for Efficient LLM Inference on Heterogeneous Platforms10 Feb 2026
- 2024A Little Goes a Long Way: Efficient Long Context Training and Inference with Partial Contexts2 Oct 2024
- 2024Let the Code LLM Edit Itself When You Edit the Code3 Jul 2024
- 2024The CAP Principle for LLM Serving: A Survey of Long-Context Large Language Model Serving18 May 2024
- 2025Training-free and Adaptive Sparse Attention for Efficient Long Video Generationcited by 69
- 2024Foundation Models for Music: A Surveycited by 37
- 2025Training-free and Adaptive Sparse Attention for Efficient Long Video Generation28 Feb 2025
- 2024Foundation Models for Music: A Survey26 Aug 2024
- 2025MoBA: Mixture of Block Attention for Long-Context LLMscited by 198
- 2025MoBA: Mixture of Block Attention for Long-Context LLMs18 Feb 2025
- 2022FEDformer: Frequency Enhanced Decomposed Transformer for Long-term Series Forecastingcited by 1,210
- 2021Random Feature Attentioncited by 354
- 2021FNet: Mixing Tokens with Fourier Transformscited by 347
- 2021XCiT: Cross-Covariance Image Transformerscited by 338
- 2022Long Range Language Modeling via Gated State Spacescited by 265
- 2023LongLoRA: Efficient Fine-tuning of Long-Context Large Language Modelscited by 244
- 2021Multi-Scale Vision Longformer: A New Vision Transformer for High-Resolution Image Encodingcited by 207
- 2023Efficient Large Language Models: A Surveycited by 186
- 2022Separable Self-attention for Mobile Vision Transformerscited by 163
- 2024Eagle and Finch: RWKV with Matrix-Valued States and Dynamic Recurrencecited by 159
- 2021Self-attention Does Not Need O(n2) Memorycited by 154
- 2022FiLM: Frequency improved Legendre Memory Model for Long-term Time Series Forecastingcited by 147
12 most cited of 146 All 146 (JSON)
- 2026BCMT: Blockwise Causal Memory Transformer6 Jul 2026
- 2026Stream-CQSA: Exact Out-of-Memory Recovery for Attention22 Apr 2026
- 2026Beyond Pairwise Attention: Higher-Order Modular Attention for Efficient Sequence Learning11 Mar 2026
- 2026A Provable Expressiveness Hierarchy in Hybrid Linear-Full Attention2 Feb 2026
- 2025Trellis: Learning to Compress Key-Value Memory in Attention Models29 Dec 2025
- 2025Optimizing Mixture of Block Attention14 Nov 2025
- 2025Compact Attention: Exploiting Structured Spatio-Temporal Sparsity for Fast Video Generation18 Aug 2025
- 2025When Does Divide and Conquer Work for Long Context LLM? A Noise Decomposition Framework19 Jun 2025
- 2025Generalized Neighborhood Attention: Multi-dimensional Sparse Attention at the Speed of Light23 Apr 2025
- 2025ARLED: Leveraging LED-based ARMAN Model for Abstractive Summarization of Persian Long Documents13 Mar 2025
- 2024kNN Attention Demystified: A Theoretical Exploration for Scalable Transformers6 Nov 2024
- 2024Rethinking Transformer for Long Contextual Histopathology Whole Slide Image Analysis18 Oct 2024
12 newest of 146 All 146 (JSON)