智能開源大模型baichuan-7B技術(shù)改進(jìn)

baichuan-7B 主要是參考LLaMA進(jìn)行的改進(jìn)，且模型架構(gòu)與LLaMA一致。而在開源大模型中，LLaMA無疑是其中最閃亮的星，但LLaMA存在如下問題：

LLaMA 原生僅支持 Latin 或 Cyrillic 語系，只使用了少量中文數(shù)據(jù)集進(jìn)行訓(xùn)練，因此，對于中文支持不是特別理想。

原版LLaMA模型的詞表大小是32K，僅有少量中文詞，對于中文的解碼效率較低。

baichuan-7B的改進(jìn)如下：

效果改進(jìn)：用于提升模型的效果以及解碼效率。

分詞改進(jìn)：詞表大小為64K（使用2000萬條以中英為主的多語言語料訓(xùn)練分詞模型，顯著提升對于中文的壓縮率），而LLaMA詞表大小為32K。

數(shù)據(jù)集改進(jìn)：使用了大約 1.2T 中英 tokens 進(jìn)行訓(xùn)練（基于開源的中英文數(shù)據(jù)和自行抓取的中文互聯(lián)網(wǎng)數(shù)據(jù)以及部分高質(zhì)量知識性數(shù)據(jù)進(jìn)行的數(shù)據(jù)清洗），而 LLaMA 7B 使用 1T 英文 tokens 進(jìn)行訓(xùn)練。

技術(shù)改進(jìn)：用于提升訓(xùn)練穩(wěn)定性和吞吐量。

算子優(yōu)化技術(shù)：采用更高效算子，如 Flash-attention，NVIDIA apex 的 RMSNorm 等。

算子切分技術(shù)：將部分計算算子進(jìn)行切分，減小內(nèi)存峰值。

混合精度技術(shù)：降低在不損失模型精度的情況下加速計算過程。

訓(xùn)練容災(zāi)技術(shù)：訓(xùn)練平臺和訓(xùn)練框架聯(lián)合優(yōu)化，IaaS + PaaS 實現(xiàn)分鐘級的故障定位和任務(wù)恢復(fù)。

通信優(yōu)化技術(shù)，具體包括：

采用拓?fù)涓兄募贤ㄐ潘惴ǎ苊?a target="_blank">網(wǎng)絡(luò)擁塞問題，提高通信效率。

根據(jù)卡數(shù)自適應(yīng)設(shè)置 bucket size，提高帶寬利用率。

根據(jù)模型和集群環(huán)境，調(diào)優(yōu)通信原語的觸發(fā)時機(jī)，從而將計算和通信重疊。

此外，該模型開源可商用，也算是一個優(yōu)勢吧。

可以看到，現(xiàn)在的大模型，從算法層面改進(jìn)的空間似乎很小了，更多的是從工程和數(shù)據(jù)層面上進(jìn)行改進(jìn)從而來提升其性能。

最后，希望國產(chǎn)大模型越來越好~~
責(zé)任編輯：彭菁