Search papers, labs, and topics across Lattice.
This paper addresses the high computational costs and debugging challenges associated with reinforcement learning (RL) post-training of large language models (LLMs) by introducing a proxy-model construction method. The authors systematically analyze failures during RL training on the Huawei Ascend platform, identifying key model-side factors that contribute to these failures. Their proposed proxy models significantly reduce computational requirements by 50%-87.5% and cut per-step NPU-hour costs by up to 33.3x, while effectively reproducing fault responses of the original models.
Proxy models can slash RL post-training costs by up to 87.5% while maintaining critical fault reproduction capabilities.
Reinforcement learning (RL) post-training of large language models (LLMs) is computationally intensive and involves complex system pipelines with substantial debugging overhead. In practice, factors such as framework adaptation, numerical precision, and operator implementation can cause failures, including gradient overflow and loss divergence. Reproducing such failures directly on large models requires considerable time and computational resources. This paper systematically analyzes failures encountered during large-scale RL training on the Huawei Ascend platform, summarizes representative failure types, and identifies three model-side factors relevant to fault reproduction. Based on these factors, we propose a proxy-model construction method for low-cost fault investigation and auxiliary diagnosis. It employs structure-preserving, clustering-based expert pruning to select representative experts while retaining the model's backbone architecture, routing mechanism, and basic task capabilities. Our experimental results show that the proxy models reduce accelerator requirements by 50%-87.5% and achieve up to a 33.3x reduction in per-step NPU-hour cost, while preserving major training dynamics and reproducing fault responses consistent with the original models. Overall, the proxy models can serve as low-cost surrogates for fault reproduction, targeted validation, and auxiliary diagnosis in RL post-training.
