6月15日消息,王小川新創(chuàng)業(yè)公司百川智能,推出70億參數(shù)量的中英文預(yù)訓(xùn)練大模型——baichuan-7B。目前這款大模型已在Hugging Face、Github以及Model Scope平臺(tái)發(fā)布。
公司CEO王小川表示:“本次開源模型的發(fā)布是百川智能成立2個(gè)月后的第一個(gè)里程碑,對(duì)百川智能而言是一個(gè)良好的開局。baichuan-7B模型,不僅能為中國(guó)的AGI事業(yè)添磚加瓦,也為世界大模型開源社區(qū)貢獻(xiàn)了新的力量?!?/p>
據(jù)了解,baichuan-7B在C-Eval、AGIEval和Gaokao三個(gè)中文評(píng)估基準(zhǔn)進(jìn)行了綜合評(píng)估。
在中文C-EVAL的評(píng)測(cè)中,baichuan-7B的綜合評(píng)分達(dá)到了42.8分,超過(guò)了ChatGLM-6B的38.9分。
C-Eval評(píng)測(cè)基準(zhǔn)由上海交通大學(xué)、清華大學(xué)以及愛丁堡大學(xué)聯(lián)合創(chuàng)建,是面向中文語(yǔ)言模型的綜合考試評(píng)測(cè)集,覆蓋了52個(gè)來(lái)自不同行業(yè)領(lǐng)域的學(xué)科。
在AGIEval的評(píng)測(cè)里,baichuan-7B綜合評(píng)分達(dá)到34.4分,超過(guò)LLaMa-7B、Falcon-7B、Bloom-7B以及ChatGLM-6B等其他開源模型。
AGIEval評(píng)測(cè)基準(zhǔn)由微軟研究院發(fā)起,旨在全面評(píng)估基礎(chǔ)模型在人類認(rèn)知和問(wèn)題解決相關(guān)任務(wù)上的能力,包含了中國(guó)的高考、司法考試,以及美國(guó)的SAT、LSAT、GRE和GMAT等20個(gè)公開且嚴(yán)謹(jǐn)?shù)墓俜饺雽W(xué)和職業(yè)資格考試。
在GAOKAO評(píng)測(cè)中,baichuan-7B的綜合評(píng)分達(dá)到了36.2分,領(lǐng)先于同參數(shù)規(guī)模的各種其他預(yù)訓(xùn)練模型。
GAOKAO評(píng)測(cè)基準(zhǔn)是復(fù)旦大學(xué)研究團(tuán)隊(duì)創(chuàng)建的評(píng)測(cè)框架,以中國(guó)高考題目作為數(shù)據(jù)集,用于測(cè)試大模型在中文語(yǔ)言理解和邏輯推理能力方面的表現(xiàn)。
此外baichuan-7B在英文上表現(xiàn)同樣亮眼。在MMLU的評(píng)測(cè)中baichuan-7B綜合評(píng)分高達(dá)42.5分,領(lǐng)先英文開源預(yù)訓(xùn)練模型LLaMA-7B的34.2分以及中文開源模型ChatGLM-6B的36.9分。
MMLU由加州大學(xué)伯克利分校等知名高校共同打造,集合了科學(xué)、工程、數(shù)學(xué)、人文、社會(huì)科學(xué)等領(lǐng)域的57個(gè)科目,主要目標(biāo)是對(duì)模型的英文跨學(xué)科專業(yè)能力進(jìn)行深入測(cè)試。其內(nèi)容廣泛,從初級(jí)水平一直涵蓋到高級(jí)專業(yè)水平。
百川智能介紹稱,訓(xùn)練語(yǔ)料對(duì)大模型的訓(xùn)練結(jié)果至關(guān)重要。在構(gòu)建預(yù)訓(xùn)練語(yǔ)料庫(kù)方面,公司以高質(zhì)量中文語(yǔ)料為基礎(chǔ),同時(shí)融合了優(yōu)質(zhì)的英文數(shù)據(jù)。在數(shù)據(jù)質(zhì)量方面,通過(guò)質(zhì)量模型對(duì)數(shù)據(jù)進(jìn)行打分,對(duì)原始數(shù)據(jù)集進(jìn)行篇章級(jí)和句子級(jí)的精確篩選;在內(nèi)容多樣性方面,利用自研超大規(guī)模局部敏感哈希聚類系統(tǒng)和語(yǔ)義聚類系統(tǒng),對(duì)數(shù)據(jù)進(jìn)行了多層次多粒度的聚類,最終構(gòu)建了包含1.2萬(wàn)億token的兼顧質(zhì)量和多樣性的預(yù)訓(xùn)練數(shù)據(jù)。相較于其他同參數(shù)規(guī)模的開源中文預(yù)訓(xùn)練模型,數(shù)據(jù)量提高了超過(guò)50%。
在萬(wàn)億優(yōu)質(zhì)中英文數(shù)據(jù)的基礎(chǔ)上,為了更好地提升訓(xùn)練效率,baichuan-7B深度整合了模型算子來(lái)加快計(jì)算流程,并針對(duì)任務(wù)負(fù)載和集群配置,自適應(yīng)優(yōu)化了模型并行策略以及重計(jì)算策略。通過(guò)高效的訓(xùn)練過(guò)程調(diào)度通信,baichuan-7B成功地實(shí)現(xiàn)了計(jì)算與通信的高效重疊,進(jìn)而達(dá)到了超線性的訓(xùn)練加速,在千卡集群上訓(xùn)練吞吐達(dá)到180+Tflops的水平。
同時(shí),已有的開源模型窗口長(zhǎng)度在2K以內(nèi),對(duì)于一些長(zhǎng)文本建模任務(wù),如需要引入外部知識(shí)做搜索增強(qiáng)的場(chǎng)景,更長(zhǎng)的處理長(zhǎng)度有助于模型在訓(xùn)練與推理階段捕獲越多的上下文信息,2K的處理長(zhǎng)度存在比較大的制約。baichuan-7B基于高效的attention算子優(yōu)化實(shí)現(xiàn)了萬(wàn)級(jí)別超長(zhǎng)動(dòng)態(tài)窗口的擴(kuò)張能力,本次開源的預(yù)訓(xùn)練模型開放了4K上下文窗口,使模型應(yīng)用場(chǎng)景更加廣泛。
此外,baichuan-7B還對(duì)模型訓(xùn)練流程進(jìn)行了深度優(yōu)化,采用了更科學(xué)且穩(wěn)定的訓(xùn)練流程和超參數(shù)選擇,使得baichuan-7B模型的收斂速度大大提升。與同等參數(shù)規(guī)模的模型相比,baichuan-7B在困惑度(PPL)和訓(xùn)練損失(training loss)等關(guān)鍵性能指標(biāo)上表現(xiàn)更加優(yōu)秀。
百川智能稱,baichuan-7B代碼采用Apache-2.0協(xié)議,模型權(quán)重采用了免費(fèi)商用協(xié)議,只需進(jìn)行簡(jiǎn)單登記即可免費(fèi)商用。
baichuan-7B此次開源的內(nèi)容包含了推理代碼、INT4量化實(shí)現(xiàn)、微調(diào)代碼,以及預(yù)訓(xùn)練模型的權(quán)重。其中,微調(diào)代碼方便用戶對(duì)模型進(jìn)行調(diào)整和優(yōu)化;推理代碼與INT4量化實(shí)現(xiàn)則有助于開發(fā)者低成本地進(jìn)行模型的部署和應(yīng)用;預(yù)訓(xùn)練模型權(quán)重開源后,用戶則可以直接使用預(yù)訓(xùn)練模型進(jìn)行各種實(shí)驗(yàn)研究。
據(jù)了解,北京大學(xué)和清華大學(xué)兩所頂尖大學(xué)已率先使用baichuan-7B模型推進(jìn)相關(guān)研究工作,并計(jì)劃在未來(lái)與百川智能深入合作,共同推動(dòng)baichuan-7B模型的應(yīng)用和發(fā)展。(易科)
(免責(zé)聲明:本網(wǎng)站內(nèi)容主要來(lái)自原創(chuàng)、合作伙伴供稿和第三方自媒體作者投稿,凡在本網(wǎng)站出現(xiàn)的信息,均僅供參考。本網(wǎng)站將盡力確保所提供信息的準(zhǔn)確性及可靠性,但不保證有關(guān)資料的準(zhǔn)確性及可靠性,讀者在使用前請(qǐng)進(jìn)一步核實(shí),并對(duì)任何自主決定的行為負(fù)責(zé)。本網(wǎng)站對(duì)有關(guān)資料所引致的錯(cuò)誤、不確或遺漏,概不負(fù)任何法律責(zé)任。
任何單位或個(gè)人認(rèn)為本網(wǎng)站中的網(wǎng)頁(yè)或鏈接內(nèi)容可能涉嫌侵犯其知識(shí)產(chǎn)權(quán)或存在不實(shí)內(nèi)容時(shí),應(yīng)及時(shí)向本網(wǎng)站提出書面權(quán)利通知或不實(shí)情況說(shuō)明,并提供身份證明、權(quán)屬證明及詳細(xì)侵權(quán)或不實(shí)情況證明。本網(wǎng)站在收到上述法律文件后,將會(huì)依法盡快聯(lián)系相關(guān)文章源頭核實(shí),溝通刪除相關(guān)內(nèi)容或斷開相關(guān)鏈接。 )