Search papers, labs, and topics across Lattice.
This paper introduces Proxy-guided Update Signal Transfer (PUST), a novel framework that decouples the exploration of update signals from distribution alignment in post-training large language models (LLMs). By utilizing a lightweight proxy model for exploration, PUST reduces computational costs and allows for the asynchronous generation and reuse of optimization signals, which can enhance the performance of stronger primary models. Evaluations on Qwen3-family models reveal that signals from weaker proxies can effectively improve the capabilities of more advanced models in both math and code tasks.
A lightweight proxy model can efficiently discover high-reward behaviors, enabling significant performance boosts in stronger LLMs without the computational burden of traditional methods.
Post-training is essential for refining the domain-specific capabilities of large language models (LLMs), yet existing reward optimization and distribution matching methods tightly couple policy exploration with distribution alignment. This coupling forces expensive exploration directly on the policy model and severely hinders the asynchronous generation, reuse, and cross-model transfer of optimization signals. In this paper, we propose Proxy-guided Update Signal Transfer (PUST), a novel post-training framework that fundamentally decouples update-signal exploration from distribution alignment. Instead of utilizing the primary model for costly exploration, PUST employs a lightweight proxy model as an efficient testbed to discover high-reward behaviors. We extract the relative improvement signal between the proxy's initial and optimized states, transferring this directional update to the primary model to guide its policy alignment. This decoupled pipeline, comprising proxy exploration, update-signal extraction, and signal transfer, significantly reduces computational overhead and enables optimization signals to be asynchronously generated, cached, and reused. Crucially, by transferring relative improvements rather than absolute policy distributions, PUST naturally supports weak-to-strong improvement and seamless cross-model transfer. Systematic evaluations on Qwen3-family models across math and code domains demonstrate that update signals extracted from substantially weaker proxies can robustly and adjustably enhance stronger primary models. Ultimately, PUST transforms post-training from a monolithic online optimization process into a highly modular, reusable, and cost-efficient paradigm.
