EndPrompt: Efficient Long-Context Extension via Terminal Anchoring
cs.CL
Submitted: 2026-05-14
Updated: 2026-09-05
Code: https://github.com/clx1415926/EndPrompt
License: http://creativecommons.org/licenses/by/4.0/
The gist: Extending the context window of large language models typically requires training on sequences at the target length, incurring quadratic memory and computational costs that make long-context
Terminology
Abstract
Extending the context window of large language models typically requires training on sequences at the target length, incurring quadratic memory and computational costs that make long-context adaptation expensive and difficult to reproduce. We propose EndPrompt, a method that achieves effective context extension using only short training sequences. The core insight is that exposing a model to long-range relative positional distances does not require constructing full-length inputs: we preserve the original short context as an intact first segment and append a brief terminal prompt as a second segment, assigning it positional indices near the target context length. This two-segment construction introduces both local and long-range relative distances within a short physical sequence while maintaining the semantic continuity of the training text--a property absent in chunk-based simulation approaches that split contiguous context. We provide a theoretical analysis grounded in Rotary Position Embedding and the Bernstein inequality, showing that position interpolation induces a rigorous smoothness constraint over the attention function, with shared Transformer parameters further suppressing unstable extrapolation to unobserved intermediate distances. Applied to LLaMA-family models extending the context window from 8K to 64K, EndPrompt achieves an average RULER score of 76.03 and the highest average on LongBench, surpassing LCEG (72.24), LongLoRA (72.95), and full-length fine-tuning (69.23) while requiring substantially less computation. These results demonstrate that long-context generalization can be induced from sparse positional supervision, challenging the prevailing assumption that dense long-sequence training is necessary for reliable context-window extension. The code is available at https://github.com/clx1415926/EndPrompt.
Sources
- LongAlign: A Recipe for Long Context Alignment of Large Language Models
- LongBench: A Bilingual, Multitask Benchmark for Long Context Understanding
- Longformer: The Long-Document Transformer
- L-Eval: Instituting Standardized Evaluation for Long Context Language Models
- Evaluating Large Language Models Trained on Code
- Extending Context Window of Large Language Models via Positional Interpolation
- Generating Long Sequences with Sparse Transformers
- Training Verifiers to Solve Math Word Problems
- LongRoPE: Extending LLM Context Window Beyond 2 Million Tokens
- RULER: What's the Real Context Size of Your Long-Context Language Models?
- Mistral 7B
- E^2-LLM: Efficient and Extreme Length Extension of Large Language Models
- Llama 2: Open Foundation and Fine-Tuned Chat Models
- Long Context Compression with Activation Beacon
Related papers
- Exploring Solution Divergence and Its Effect on Large Language Model Problem Solving
- Ishigaki-IDS-Bench: A Benchmark for Generating Information Delivery Specification from BIM Information Requirements
- Subliminal Steering: Stronger Encoding of Hidden Signals
- MedStruct-S: A Benchmark for Key Discovery, Key-Conditioned QA and Semi-Structured Extraction from OCR Clinical Reports
- The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
- Untangling the Mechanisms of Misleading Context in Medical Question Answering