English

Enhancing Reinforcement Learning Fine-Tuning with an Online Refiner

Machine Learning 2026-03-20 v1 Artificial Intelligence

Abstract

Constraints are essential for stabilizing reinforcement learning fine-tuning (RFT) and preventing degenerate outputs, yet they inherently conflict with the optimization objective because stronger constraints limit the ability of a fine-tuned model to discover better solutions. We propose \textit{dynamic constraints} that resolve this tension by adapting to the evolving capabilities of the fine-tuned model based on the insight that constraints should only intervene when degenerate outputs occur. We implement this by using a reference model as an \textit{online refiner} that takes the response from the fine-tuned model and generates a minimally corrected version which preserves correct content verbatim while fixing errors. A supervised fine-tuning loss then trains the fine-tuned model to produce the refined output. This mechanism yields a constraint that automatically strengthens or relaxes based on output quality. Experiments on dialogue and code generation show that dynamic constraints outperform both KL regularization and unconstrained baselines, achieving substantially higher task rewards while maintaining training stability.

Keywords

Cite

@article{arxiv.2603.18088,
  title  = {Enhancing Reinforcement Learning Fine-Tuning with an Online Refiner},
  author = {Hao Ma and Zhiqiang Pu and Yang Liu and Xiaolin Ai},
  journal= {arXiv preprint arXiv:2603.18088},
  year   = {2026}
}
R2 v1 2026-07-01T11:26:50.818Z