LADM: Long-context Training Data Selection with Attention-based Dependency Measurement for LLMs
Not on the map. Why?
Abstract (arXiv)
Long-context modeling has drawn more and more attention in the area of Large Language Models (LLMs). Continual training with long-context data becomes the de-facto method to equip LLMs with the ability to process long inputs. However, it still remains an open challenge to measure the quality of long-context training data. To address this issue, we propose a Long-context data selection framework with Attention-based Dependency Measurement (LADM), which can efficiently identify high-quality long-context data from a large-scale, multi-domain pre-training corpus. LADM leverages the retrieval capabilities of the attention mechanism to capture contextual dependencies, ensuring a comprehensive quality measurement of long-context data. Experimental results show that our LADM framework significantly boosts the performance of LLMs on multiple long-context tasks with only 1B tokens for continual training.
Key facts. Cited by 11 papers in the map (data through 7 Oct 2026). Builds on 35 of its 54 references that are in the map. First posted on arXiv 4 Mar 2025.
Builds on
35 of 54 references.
References · 35
- 2023GPT-4 Technical Reportcited by 18,132
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2020Measuring Massive Multitask Language Understandingcited by 7,768
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2023Lost in the Middle: How Language Models Use Long Contextscited by 3,383
- 2024Gemini 1.5: Unlocking multimodal understanding across millions of tokens of contextcited by 3,175
- 2023Mistral 7Bcited by 3,065
- 2021TruthfulQA: Measuring How Models Mimic Human Falsehoodscited by 2,919
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
23 more
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
- 2023Efficient Streaming Language Models with Attention Sinkscited by 2,115
- 2023LongBench: A Bilingual, Multitask Benchmark for Long Context Understandingcited by 1,378
- 2019ZeRO: Memory Optimizations Toward Training Trillion Parameter Modelscited by 1,278
- 2023Extending Context Window of Large Language Models via Positional Interpolationcited by 721
- 2017The NarrativeQA Reading Comprehension Challengecited by 590
- 2021A Dataset of Information-Seeking Questions and Answers Anchored in Research Paperscited by 475
- 2019Multi-News: a Large-Scale Multi-Document Summarization Dataset and Abstractive Hierarchical Modelcited by 431
- 2021QMSum: A New Benchmark for Query-based Multi-domain Meeting Summarizationcited by 363
- 2021Efficient Attentions for Long Document Summarizationcited by 342
- 2025Demystifying Long Chain-of-Thought Reasoning in LLMscited by 323
- 2024Retrieval Head Mechanistically Explains Long-Context Factualitycited by 219
- 2024Data Engineering for Scaling Language Models to 128K Contextcited by 183
- 2023A Simple LLM Framework for Long-Range Video Question-Answeringcited by 176
- 2024LongWriter: Unleashing 10,000+ Word Generation from Long Context LLMscited by 142
- 2024Make Your LLM Fully Utilize the Contextcited by 118
- 2024Found in the Middle: Calibrating Positional Attention Bias Improves Long Context Utilizationcited by 97
- 2024Can Perplexity Reflect Large Language Model's Ability in Long Text Understanding?cited by 45
- 2023Never Lost in the Middle: Mastering Long-Context Question Answering with Position-Agnostic Decompositional Trainingcited by 31
- 2025LongPO: Long Context Self-Evolution of Large Language Models through Short-to-Long Preference Optimizationcited by 24
- 2024From Artificial Needles to Real Haystacks: Improving Retrieval Capabilities in LLMs by Finetuning on Synthetic Datacited by 18
- 2023Structured Packing in LLM Training Improves Long Context Utilizationcited by 17
- 2023CoCA: Fusing Position Embedding with Collinear Constrained Attention in Transformers for Long Context Window Extendingcited by 5
Built on by
Citing papers per year: 2025: 6; 2026: 5.
11 papers.
- 2025Look Again, Think Slowly: Enhancing Visual Reflection in Vision-Language Modelscited by 16
- 2025CROP: Contextual Region-Oriented Visual Token Pruningcited by 14
- 2025Structured Memory Mechanisms for Stable Context Representation in Large Language Modelscited by 9
- 2025ACE-RL: Adaptive Constraint-Enhanced Reward for Long-form Generation Reinforcement Learningcited by 9
- 2025TRIM: Token-wise Attention-Derived Saliency for Data-Efficient Instruction Tuningcited by 3
- 2026DySCO: Dynamic Attention-Scaling Decoding for Long-Context Language Modelscited by 2
- 2025Short-Context Dominance: How Much Local Context Natural Language Actually Needs?cited by 1
- 2026Information Abundance Paradox: Long-Context Training Undermines Parametric Knowledgecited by 1
- 2026UT-ACA: Uncertainty-Triggered Adaptive Context Allocation for Long-Context Inferencecited by 0
- 2026The Announcement Carries the Cue: Markup, Boundaries, and the Notation of Pre-Training Corporacited by 0
- 2026What Should Data Teach? Moving Bottlenecks Across Circuit, Store, and Usecited by 0
- 2026What Should Data Teach? Moving Bottlenecks Across Circuit, Store, and Use26 Sep 2026
- 2026Information Abundance Paradox: Long-Context Training Undermines Parametric Knowledge12 Aug 2026
- 2026The Announcement Carries the Cue: Markup, Boundaries, and the Notation of Pre-Training Corpora10 Aug 2026
- 2026UT-ACA: Uncertainty-Triggered Adaptive Context Allocation for Long-Context Inference19 Mar 2026
- 2026DySCO: Dynamic Attention-Scaling Decoding for Long-Context Language Models25 Feb 2026
- 2025Short-Context Dominance: How Much Local Context Natural Language Actually Needs?8 Dec 2025
- 2025TRIM: Token-wise Attention-Derived Saliency for Data-Efficient Instruction Tuning8 Oct 2025
- 2025Look Again, Think Slowly: Enhancing Visual Reflection in Vision-Language Models15 Sep 2025
- 2025ACE-RL: Adaptive Constraint-Enhanced Reward for Long-form Generation Reinforcement Learning5 Sep 2025
- 2025Structured Memory Mechanisms for Stable Context Representation in Large Language Models28 May 2025
- 2025CROP: Contextual Region-Oriented Visual Token Pruning27 May 2025