arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2304.01196cs.CLcs.AI
收藏

Baize:基于自聊数据的参数高效微调开源聊天模型

Baize: An Open-Source Chat Model with Parameter-Efficient Tuning on Self-Chat Data

University of California, San Diego(加州大学圣地亚哥分校);Microsoft Research Asia(微软亚洲研究院);Sun Yat-sen University(中山大学)

Canwen Xu, Daya Guo, Nan Duan, Julian McAuley

AI总结:

本文提出利用 ChatGPT 自聊生成高质量多轮对话数据,并以参数高效微调增强 LLaMA,得到带护栏的开源聊天模型 Baize,还引入 Self-Distill with Feedback 技术进一步提升性能。

摘要:

像 ChatGPT 这样的聊天模型已经展现出令人印象深刻的能力,并迅速在众多领域得到应用。然而,这些模型只能通过受限的 API 访问,给该领域的新研究和进展造成了障碍。我们提出一个流程,通过利用 ChatGPT 与自身进行对话,自动生成高质量的多轮聊天语料库。随后,我们采用参数高效微调来增强开源大语言模型 LLaMA。由此得到的模型名为 Baize,在多轮对话中表现出良好性能,并带有护栏以最小化潜在风险。此外,我们提出一种名为 Self-Distill with Feedback 的新技术,利用 ChatGPT 的反馈进一步改善 Baize 模型的性能。Baize 模型和数据仅用于研究目的,发布于 https://github.com/project-baize/baize-chatbot。在线演示也可在 https://huggingface.co/spaces/project-baize/chat-with-baize 获取。

英文摘要:

Chat models, such as ChatGPT, have shown impressive capabilities and have been rapidly adopted across numerous domains. However, these models are only accessible through a restricted API, creating barriers for new research and progress in the field. We propose a pipeline that can automatically generate a high-quality multi-turn chat corpus by leveraging ChatGPT to engage in a conversation with itself. Subsequently, we employ parameter-efficient tuning to enhance LLaMA, an open-source large language model. The resulting model, named Baize, demonstrates good performance in multi-turn dialogues with guardrails that minimize potential risks. Furthermore, we propose a new technique called Self-Distill with Feedback, to further improve the performance of the Baize models with feedback from ChatGPT. The Baize models and data are released for research purposes only at https://github.com/project-baize/baize-chatbot. An online demo is also available at https://huggingface.co/spaces/project-baize/chat-with-baize.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑