DeepSpeed-Chat:面向全规模ChatGPT类模型的简单、快速且经济的RLHF训练
DeepSpeed-Chat: Easy, Fast and Affordable RLHF Training of ChatGPT-like Models at All Scales
AI总结:
本文提出DeepSpeed-Chat系统,通过提供易用的训练推理体验、复刻InstructGPT的DeepSpeed-RLHF管线及统一优化系统,实现了低成本、高效率的百亿至千亿参数级ChatGPT类模型RLHF训练,使RLHF大众化。
摘要:
ChatGPT类模型在从摘要、编程到翻译等人工智能应用中引发了革命,其表现已匹敌甚至超越人类水平。然而,当前领域缺乏一条易于访问、高效且高性价比的端到端RLHF(基于人类反馈的强化学习)训练管线来训练这些强大模型,尤其是在数十亿参数规模下。本文介绍了DeepSpeed-Chat,这一新系统使RLHF训练大众化,让AI社区均可访问。DeepSpeed-Chat提供三大核心能力:面向ChatGPT类模型的易用训练与推理体验;复刻InstructGPT训练管线的DeepSpeed-RLHF管线;以及将训练与推理的各类优化统一结合的稳健DeepSpeed-RLHF系统。该系统提供了无与伦比的效率与可扩展性,能在创纪录的时间内以极低成本训练含数千亿参数的模型。借助此项进展,DeepSpeed-Chat为更广泛地获取高级RLHF训练铺平了道路,即使是资源有限的数据科学家也能参与,从而促进了AI领域的创新与进一步发展。
英文摘要:
ChatGPT-like models have revolutionized various applications in artificial intelligence, from summarization and coding to translation, matching or even surpassing human performance. However, the current landscape lacks an accessible, efficient, and cost-effective end-to-end RLHF (Reinforcement Learning with Human Feedback) training pipeline for these powerful models, particularly when training at the scale of billions of parameters. This paper introduces DeepSpeed-Chat, a novel system that democratizes RLHF training, making it accessible to the AI community. DeepSpeed-Chat offers three key capabilities: an easy-to-use training and inference experience for ChatGPT-like models, a DeepSpeed-RLHF pipeline that replicates the training pipeline from InstructGPT, and a robust DeepSpeed-RLHF system that combines various optimizations for training and inference in a unified way. The system delivers unparalleled efficiency and scalability, enabling training of models with hundreds of billions of parameters in record time and at a fraction of the cost. With this development, DeepSpeed-Chat paves the way for broader access to advanced RLHF training, even for data scientists with limited resources, thereby fostering innovation and further development in the field of AI.
机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。