Beyond Model Collapse: Scaling Up with Synthesized Data Requires Reinforcement

LLM RLHF ML RL
合成模型生成的数据越来越被视为大型语言模型微调的人工注释数据的替代选择。这引起了有关模型崩溃的担忧:即在生成数据上微调的模型性能下降。考虑到对于人类和机器来说,区分好坏示例比生成高质量样本更容易,我们研究了在合成数据上使用反馈来防止模型崩溃。我们推导了一些理论条件,使得高斯混合分类模型在训练反馈增强的合成数据时可以达到渐进最优性能,并为有限范围提供了支持模拟。我们在两个实际问题上说明了我们的理论预测:使用Transformer计算矩阵特征值和使用大型语言模型进行新闻摘要,这两个问题在使用模型生成数据进行训练时都会遭遇模型崩溃。我们展示了通过修剪不正确的预测或从多个猜测中选择最佳猜测来从反馈增强的合成数据中进行训练可以防止模型崩溃,从而验证了像RLHF这样的流行方法。
Synthesized data from generative models is increasingly considered as an alternative to human-annotated data for fine-tuning Large Language Models. This raises concerns about model collapse: a drop in performance of models fine-tuned on generated data. Considering that it is easier for both humans and machines to tell between good and bad examples than to generate high-quality samples, we investigate the use of feedback on synthesized data to prevent model collapse. We derive theoretical conditions under which a Gaussian mixture classification model can achieve asymptotically optimal performance when trained on feedback-augmented synthesized data, and provide supporting simulations for finite regimes. We illustrate our theoretical predictions on two practical problems: computing matrix eigenvalues with transformers and news summarization with large language models, which both undergo model collapse when trained on model-generated data. We show that training from feedback-augmented synthesized data, either by pruning incorrect predictions or by selecting the best of several guesses, can prevent model collapse, validating popular approaches like RLHF.
许愿