明敏 發(fā)自 凹非寺

量子位 | 公眾號 QbitAI

這邊ChatGPT、GPT-4等AI大模型和應(yīng)用打得火熱；

另一邊“平替”開源復(fù)現(xiàn)方案也加緊更新迭代。

這不，“首個開源ChatGPT低成本復(fù)現(xiàn)流程”就來了波大更新！

現(xiàn)在，僅需不到百億參數(shù)，利用RLHF簡單微調(diào)，模型即可掌握中、英雙語能力，達(dá)到與ChatGPT和GPT-3.5相當(dāng)?shù)男Ч?/p>

中文對話效果如下：

這就是ColossalChat。

由Colossal-AI推出。一個月前，Colossal-AI乘著ChatGPT熱潮火速開源了低成本復(fù)現(xiàn)流程。

而新升級的ColossalChat，以Meta最新開源的LLaMA為基礎(chǔ)預(yù)訓(xùn)練模型，能力更強(qiáng)、門檻還更低了：

• Demo：可直接在線體驗(yàn)?zāi)Ｐ托Ч瑹o需注冊或waitinglist
• 訓(xùn)練代碼：開源完整 RLHF 訓(xùn)練代碼，已開源至含7B、13B兩種模型
• 數(shù)據(jù)集：開源104K中、英雙語數(shù)據(jù)集
• 推理部署：4bit量化推理70億參數(shù)模型僅需4GB顯存
• 模型權(quán)重：僅需單臺服務(wù)器少量算力即可快速復(fù)現(xiàn)
• 更大規(guī)模模型、數(shù)據(jù)集、其他優(yōu)化等將保持高速迭代添加

要知道，模型開源、數(shù)據(jù)集、訓(xùn)練應(yīng)用成本、核心數(shù)據(jù)安全性等，是AI大模型浪潮下最被關(guān)注的一些問題。ColossalChat針對于此，一次性給出了應(yīng)對方法。

由此，想要快速跟進(jìn)ChatGPT這輪技術(shù)浪潮，門檻又低了一些。

開源地址：https://github.com/hpcaitech/ColossalAI

包含完整 RLHF 流程

具體來看，ColossalChat的能力包括知識問答、中英文對話、內(nèi)容創(chuàng)作、編程等。

和ChatGPT一樣，ColossalChat知道NBA和喬丹是誰：

會寫郵件，格式非常規(guī)范：

編程能力也不錯，能搞定算法：

據(jù)了解，為了能更逼近ChatGPT、GPT-4的驚艷效果，ColossalChat使用LLaMA作為預(yù)訓(xùn)練模型，并包含完整RLHF流程。

之所以這樣做，主要是因?yàn)楝F(xiàn)有開源方案都可被視為只得到了人類反饋強(qiáng)化學(xué)習(xí)（RLHF）中第一步的監(jiān)督微調(diào)模型，沒有進(jìn)行后續(xù)的對齊和微調(diào)工作。

△RLHF的三個階段

比如Meta 開源了LLaMA模型，其參數(shù)量從70億到650億不等，號稱130 億參數(shù)即可勝過1750億的GPT-3模型在大多數(shù)基準(zhǔn)測試的表現(xiàn)。

但是由于沒有被指令微調(diào)（instruct tuning），因此實(shí)際生成效果不夠理想。

斯坦福的Alpaca通過調(diào)用OpenAI API，以self-instruct方式生成訓(xùn)練數(shù)據(jù)，使得僅有70億參數(shù)的輕量級模型以極低成本微調(diào)后，即可獲得媲美GPT-3.5這樣千億參數(shù)的超大規(guī)模語言模型的對話效果。

而ChatGPT、GPT-4效果好的一大關(guān)鍵就是將RLHF引入訓(xùn)練流程，才能讓生成內(nèi)容更符合人類價值觀。

因此，基于LLaMA模型，包含完整RLHF流程的類Chat模型復(fù)現(xiàn)方案 ColossalChat，可以稱得上是目前最接近ChatGPT原始技術(shù)路線的實(shí)用開源項(xiàng)目。

總結(jié)來看，和Alpaca相比較ColossalChat具備4個方面的優(yōu)勢。

第一、ColossalChat開源了第一個完整的RLHF pipeline，斯坦福Alpaca沒有做RLHF，也就是沒有做Stage2和3。

第二、ColossalChat采用了更多的指令數(shù)據(jù)，質(zhì)量更好、范圍更大，并使用強(qiáng)化學(xué)習(xí)做了alignment使回答更接近人類。

第三、ColossalChat訓(xùn)練流程集成了Colossal-AI的諸多系統(tǒng)優(yōu)化，同等數(shù)據(jù)集和模型大小的訓(xùn)練速度可以比AIpaca快3倍左右，能讓科研人員和中小企業(yè)獨(dú)立訓(xùn)練部署自己的會話系統(tǒng)。

第四、ColossalChat采集了更多數(shù)據(jù)集：訓(xùn)練的英文一共 24M tokens，中文大約 30M tokens，總共約 54M tokens。其中ColossalChat自己收集的數(shù)據(jù)集英文 6M，中文 18M tokens。

訓(xùn)練數(shù)據(jù)集開源

數(shù)據(jù)集方面，ColossalChat開源了包含約10 萬條問答的中、英雙語數(shù)據(jù)集。

該數(shù)據(jù)集收集并清洗了社交平臺上人們的真實(shí)提問場景作為種子數(shù)據(jù)集，利用self-instruct技術(shù)擴(kuò)充數(shù)據(jù)，花費(fèi)約900美元進(jìn)行標(biāo)注。

對比其他self-instruct方法生成的數(shù)據(jù)集，該數(shù)據(jù)集的種子數(shù)據(jù)更加真實(shí)、豐富，生成的數(shù)據(jù)集涵蓋的話題更多。

該數(shù)據(jù)可以同時用于微調(diào)和RLHF訓(xùn)練。通過高質(zhì)量的數(shù)據(jù)，ColossalChat能進(jìn)行更好地對話交互，同時支持中文。

△ColossalChat數(shù)據(jù)集收集流程

RLHF算法復(fù)現(xiàn)

RLHF第一步（Stage1）是supervised-fintuning，即使用上文提到的數(shù)據(jù)集進(jìn)行模型微調(diào)。

RLHF第二步（Stage2）訓(xùn)練了獎勵模型，它通過對于同一個prompt的不同輸出進(jìn)行人工排序，得到對應(yīng)分?jǐn)?shù)，監(jiān)督訓(xùn)練獎勵模型。

RLHF第三步（Stage3）使用了強(qiáng)化學(xué)習(xí)算法，是訓(xùn)練流程中最復(fù)雜的一部分：

△RLHF-Stage3算法流程圖

在PPO部分，ColossalChat分為兩個階段進(jìn)行：

首先是Make Experience部分，利用SFT、Actor、RM、Critic模型計算生成Experience存入buffer中；之后是參數(shù)更新部分，利用Experience計算策略損失和價值損失。

在PTX部分，ColossalChat計算Actor輸出response和輸入語料的回答部分的交叉熵?fù)p失函數(shù)，用來在PPO梯度中加入預(yù)訓(xùn)練梯度，以保持語言模型原有性能防止遺忘。最后將策略損失、價值損失和 PTX 損失加和進(jìn)行反向傳播和參數(shù)更新。

快速上手

ColossalChat開源了基于 LLaMA 模型，復(fù)現(xiàn)訓(xùn)練 ChatGPT 三個階段的完整代碼。

第一階段，訓(xùn)練SFT模型：

# Training with a 4-GPU servers

colossalai run --nproc_per_node=4 train_sft.py

--pretrain "/path/to/LLaMa-7B/"

--model 'llama'

--strategy colossalai_zero2

--log_interval 10

--save_path /path/to/Coati-7B

--dataset /path/to/data.json

--batch_size 4

--accimulation_steps 8

--lr 2e-5

第二階段，訓(xùn)練獎勵模型：

# Training with a 4-GPU servers

colossalai run --nproc_per_node=4 train_reward_model.py

--pretrain "/path/to/LLaMa-7B/"

--model 'llama'

--strategy colossalai_zero2

--dataset /path/to/datasets

第三階段，使用RL訓(xùn)練：

# Training with a 8-GPU servers

colossalai run --nproc_per_node=8 train_prompts.py prompts.csv

--strategy colossalai_zero2

--pretrain "/path/to/Coati-7B"

--model 'llama'

--pretrain_dataset /path/to/dataset

在獲得最終模型權(quán)重后，還可通過量化降低推理硬件成本，并啟動在線推理服務(wù)，僅需單張約4GB顯存的GPU即可完成70億參數(shù)模型推理服務(wù)部署。

python server.py /path/to/pretrained --quant 4bit --gptq_checkpoint /path/to/coati-7b-4bit-128g.pt --gptq_group_size 128

系統(tǒng)性能優(yōu)化與開發(fā)加速

ColossalChat能夠快速跟進(jìn)ChatGPT完整RLHF流程復(fù)現(xiàn)，離不開AI大模型基礎(chǔ)設(shè)施Colossal-AI及相關(guān)優(yōu)化技術(shù)的底座支持，相同條件下訓(xùn)練速度相比Alpaca采用的FSDP(Fully Sharded Data Parallel) 可提升3倍以上。

系統(tǒng)基礎(chǔ)設(shè)施Colossal-AI

AI大模型開發(fā)系統(tǒng)Colossal-AI為該方案提供了基礎(chǔ)支持，它可基于PyTorch高效快速部署AI大模型訓(xùn)練和推理，從而降低AI大模型應(yīng)用的成本。

Colossal-AI由加州伯克利大學(xué)杰出教授James Demmel和新加坡國立大學(xué)校長青年教授尤洋領(lǐng)導(dǎo)開發(fā)。

自從它開源以來，Colossal-AI已經(jīng)多次在GitHub熱榜位列世界第一，獲得GitHub Star約兩萬顆，并成功入選SC、AAAI、PPoPP、CVPR、ISC等國際AI與HPC頂級會議的官方教程。

減少內(nèi)存冗余的ZeRO+Gemini

Colossal-AI支持使用無冗余優(yōu)化器 (ZeRO) 提高內(nèi)存使用效率，低成本容納更大模型，同時不影響計算粒度和通信效率。

自動Chunk機(jī)制可以進(jìn)一步提升ZeRO的性能，提高內(nèi)存使用效率，減少通信次數(shù)并避免內(nèi)存碎片。

異構(gòu)內(nèi)存空間管理器Gemini支持將優(yōu)化器狀態(tài)從GPU顯存卸載到CPU內(nèi)存或硬盤空間，以突破GPU顯存容量限制，擴(kuò)展可訓(xùn)練模型的規(guī)模，降低AI大模型應(yīng)用成本。

使用LoRA低成本微調(diào)

Colossal-AI支持使用低秩矩陣微調(diào)（LoRA）方法，對AI大模型進(jìn)行低成本微調(diào)。

LoRA方法認(rèn)為大語言模型是過參數(shù)化的，而在微調(diào)時，參數(shù)改變量是一個低秩矩陣。

因此，可以將這個矩陣分解為兩個更小的矩陣的乘積。

在微調(diào)過程中，大模型的參數(shù)被固定，只有低秩矩陣參數(shù)被調(diào)整，從而顯著減小了訓(xùn)練所需的參數(shù)量，并降低成本。

低成本量化推理

△GPTQ量化

為降低推理部署成本，Colossal-AI使用GPTQ 4bit量化推理。

在GPT/OPT/BLOOM類模型上，它比傳統(tǒng)的RTN(rount-to-nearest) 量化技術(shù)能夠獲得更好的Perplexity效果。相比常見的FP16推理，它可將顯存消耗降低75%，只損失極少量的吞吐速度與Perplexity性能。

以ColossalChat-7B為例，在使用4bit量化推理時，70億參數(shù)模型僅需大約4GB顯存即可完成短序列（生成長度為128）推理，在普通消費(fèi)級顯卡上即可完成（例如RTX 3060 Laptop），僅需一行代碼即可使用。

if args.quant == '4bit':

model = load_quant(args.pretrained, args.gptq_checkpoint, 4, args.gptq_group_size)

如果采用高效的異步卸載技術(shù)(offload)，還可以進(jìn)一步降低顯存要求，使用更低成本的硬件推理更大的模型。

開放協(xié)作

不過目前由于算力和數(shù)據(jù)集有限，在部分場景下的實(shí)際性能還有提升空間。

比如還是會被弱智吧問題難住：

而在這輪技術(shù)浪潮中，除了科技巨頭們，PyTorch、Hugging Face和OpenAI等開源社區(qū)與初創(chuàng)企業(yè)也起到了關(guān)鍵作用。

借鑒這些成功經(jīng)驗(yàn)，Colossal-AI也歡迎各方參與共建，并給出了多種參與方式：

• 在GitHub發(fā)布issue或提交pull request (PR)
• 加入Colossal-AI用戶微信或Slack群交流
• 發(fā)送正式合作提案到郵箱youy@comp.nus.edu.sg

如果你對這項(xiàng)工作感興趣，可以趕快和他們?nèi)〉寐?lián)系~

開源地址：

https://github.com/hpcaitech/ColossalAI

參考鏈接：

https://medium.com/@yangyou_berkeley/colossalchat-an-open-source-solution-for-cloning-chatgpt-with-a-complete-rlhf-pipeline-5edf08fb538b

— 完 —

量子位 QbitAI · 頭條號簽約