English

PolicyLong: Towards On-Policy Context Extension

Machine Learning 2026-04-10 v1 Artificial Intelligence

Abstract

Extending LLM context windows is hindered by scarce high-quality long-context data. Recent methods synthesize data with genuine long-range dependencies via information-theoretic verification, selecting contexts that reduce a base model's predictive entropy. However, their single-pass offline construction with a fixed model creates a fundamental off-policy gap: the static screening landscape misaligns with the model's evolving capabilities, causing the training distribution to drift. We propose PolicyLong, shifting data construction towards a dynamic on-policy paradigm. By iteratively re-executing data screening (entropy computation, retrieval, and verification) using the current model, PolicyLong ensures the training distribution tracks evolving capabilities, yielding an emergent self-curriculum. Crucially, both positive and hard negative contexts derive from the current model's entropy landscape, co-evolving what the model learns to exploit and resist. Experiments on RULER, HELMET, and LongBench-v2 (Qwen2.5-3B) show PolicyLong consistently outperforms EntropyLong and NExtLong, with gains growing at longer contexts (e.g., +2.54 at 128K on RULER), confirming the value of on-policy data evolution.

Keywords

Cite

@article{arxiv.2604.07809,
  title  = {PolicyLong: Towards On-Policy Context Extension},
  author = {Junlong Jia and Ziyang Chen and Xing Wu and Chaochen Gao and TingHao Yu and Feng Zhang and Songlin Hu},
  journal= {arXiv preprint arXiv:2604.07809},
  year   = {2026}
}

Comments

Work in progress. Correspondence to ucaswu@tencent.com or wuxing@iie.ac.cn

R2 v1 2026-07-01T12:00:33.565Z