A Recipe for Long-Context Reasoning in Large Language Models via On-Policy Optimization and Distillation
Existing approaches to post-train models for long-context tasks face complementary limitations: (i) supervised fine-tuning (SFT) provides stable supervision but suffers from exposure bias; (ii) reinforcement learning methods such as Group Relative Policy Optimization (GRPO) train on model-generated trajectories but struggle with long-horizon credit assignment and sparse rewards; and (iii) on-policy distillation (OPD) provides dense token-level guidance but does not directly optimize task rewards. We study these complementary strategies for long-context alignment and derive a recipe that combines GRPO with OPD-style teacher guidance: the student learns from its own rollouts using outcome-l...
| Authors | Miguel Moura Ramos, Duarte M. Alves, André F. T. Martins |
|---|---|
| Primary category | cs.CL |
| Published | 12 May 2026 |
| Last updated at source | 10 September 2026 |
Connected in the knowledge graph
- utter-project/LongBlocksDataset · cited by
- Theafricatechguy/LongBlocksDataset · cited by
Models and datasets referencing this paper
- utter-project/LongBlocksDataset
- Theafricatechguy/LongBlocksDataset
Silytics shows public metadata from arXiv and links to the original. Information may be incomplete or out of date. Last refreshed 17 October 2026. Licenses and terms of the original apply to the work itself. Not financial, legal or professional advice.