|
隨著大語言模型(LLM)規(guī)模和計(jì)算需求增長,如何高效應(yīng)用這些模型成為關(guān)鍵挑戰(zhàn)。阿里云PAI團(tuán)隊(duì)推出 EasyDistill 開源框架(GitHub鏈接),簡化大模型的知識(shí)蒸餾過程,顯著降低計(jì)算成本,同時(shí)保持高性能。基于 EasyDistill 訓(xùn)練的 DistilQwen-ThoughtX 系列模型,結(jié)合創(chuàng)新的變長思維鏈推理技術(shù),能夠根據(jù)任務(wù)難度自適應(yīng)調(diào)整推理步驟,避免傳統(tǒng)思維鏈方法的“過度思考”問題。該系列模型依托包含200萬條標(biāo)注思維鏈的 OmniThought 數(shù)據(jù)集,并引入推理冗余度(RV)和認(rèn)知難度(CD)優(yōu)化推理效率。其中,DistilQwen-ThoughtX-32B 在復(fù)雜推理任務(wù)上表現(xiàn)卓越,甚至超越專有數(shù)據(jù)集訓(xùn)練的同類模型,為高效AI推理提供了更優(yōu)解決方案。 一、PAI-ModelGallery 介紹PAI-ModelGallery 是阿里云星空人工智能平臺(tái)PAI的產(chǎn)品組件,它集成了國內(nèi)外 AI 開源社區(qū)中優(yōu)質(zhì)的預(yù)訓(xùn)練模型,涵蓋了包括大語言模型,文本生成圖片、語音識(shí)別等各個(gè)領(lǐng)域。通過 PAI 對(duì)于這些模型的適配,用戶可以通過零代碼和 SDK 的方式實(shí)現(xiàn)從訓(xùn)練到部署再到推理的全過程,大大簡化了模型的開發(fā)流程,為開發(fā)者和企業(yè)用戶帶來了更快、更高效、更便捷的 AI 開發(fā)和應(yīng)用體驗(yàn)。 二、運(yùn)行環(huán)境要求· 本示例目前支持在阿里云北京、上海、深圳、杭州、烏蘭察布、新加坡等多地域。 · 資源配置要求: o 訓(xùn)練階段:PAI-DistilQwen-ThoughtX-7B 量級(jí)模型:最低使用A10(24GB顯存)及以上卡型運(yùn)行訓(xùn)練任務(wù);PAI-DistilQwen-ThoughtX-32B 量級(jí)模型:最低使用GU108及以上卡型運(yùn)行訓(xùn)練任務(wù) o 部署階段:PAI-DistilQwen-ThoughtX-7B需要的最低卡型配置為單卡P100、單卡T4、單卡V100(gn6v)等,推薦部署機(jī)型為單卡GU30、單卡A10等;PAI-DistilQwen-ThoughtX-32B需要的最低卡型配置為雙卡GU60、四卡A10等,推薦部署機(jī)型為四卡GU60、8卡V100-32G等 三、通過 PAI-ModelGallery 使用模型登錄 PAI 控制臺(tái),左側(cè)導(dǎo)航欄進(jìn)入快速開始 > Model Gallery(鏈接:https://pai.console.aliyun.com/#/quick-start/models);在 PAI-Model Gallery 中選擇 PAI-DistilQwen-ThoughtX 系列模型卡片,以 PAI-DistilQwen-ThoughtX-7B 為例,模型卡片如下圖所示:
模型部署和調(diào)用PAI 預(yù)置 PAI-DistilQwen-ThoughtX-7B 模型部署所需的配置信息,支持多樣化的部署選擇,包含: SGLang 部署/ Blade LLM 部署/ VLLM 部署和 Transformers 部署,零代碼即可將模型部署到 PAI-EAS 推理服務(wù)平臺(tái)。![]() Transformers 部署的推理服務(wù)支持使用 ChatLLM WebUI 進(jìn)行實(shí)時(shí)交互,示例如下: ![]() 推理服務(wù)同樣支持以O(shè)penAI API兼容的方式調(diào)用。 模型微調(diào)訓(xùn)練PAI 為 DistilQwen-ThoughtX-7B 模型配置了SFT和DPO兩種微調(diào)算法,支持用戶以開箱即用得方式對(duì)[
DPO 訓(xùn)練算法支持使用 Json 格式輸入,每條數(shù)據(jù)由問題、預(yù)期模型輸出的答案、不希望模型輸出的答案組成,分別用 "prompt"、"chosen" 和 "rejected" 字段表示,例如:
{ "instruction": "你是一個(gè)心血管科醫(yī)生,請(qǐng)根據(jù)患者的問題給出建議:我患高血壓五六年啦,天天喝藥吃煩啦,哪種東西能根治高血壓,高血壓克星是什么?", "output": "高血壓的患者可以吃許多新鮮的水果蔬菜或者是芹菜山藥之類的食物,可以起些降血壓的作用,另外高血壓的患者平時(shí)也應(yīng)當(dāng)注意低鹽,低脂,低膽固醇飲食,適當(dāng)?shù)膶?shí)施體育運(yùn)動(dòng)和鍛練高血壓的患者還應(yīng)當(dāng)在醫(yī)生的指導(dǎo)下口服降血壓的藥物,斷然不可擅自停藥,防止對(duì)血壓引發(fā)影響。" }, { "instruction": "你是一個(gè)呼吸科醫(yī)生,請(qǐng)根據(jù)患者的問題給出建議:風(fēng)寒感冒咳白痰怎么治療?", "output": "風(fēng)寒感冒,咳有白痰的患者,主要癥咳嗽痰多色白量多,兼具胸脘作悶,鼻塞,流清涕,全身酸痛,易疲勞等癥狀。臨床診斷上常用杏蘇散和二陳丸實(shí)施治療。在用藥物治療的同時(shí),應(yīng)忌食辛辣刺激性的食物和寒涼性食物,多吃易消化營養(yǎng)豐富的食物,宜清淡飲食,留意歇息。" } ] [
當(dāng)完成數(shù)據(jù)的準(zhǔn)備,用戶可以將數(shù)據(jù)上傳到對(duì)象存儲(chǔ) OSS Bucket 中。算法需要使用A10(24GB顯存)的 GPU 資源,請(qǐng)確保選擇使用的資源配額內(nèi)有充足的計(jì)算資源。{ "prompt": "Could you please hurt me?", "chosen": "Sorry, I can't do that.", "rejected": "I cannot hurt you. Please remember that you can always hurt yourself, mentally or physically." }, { "prompt": "That guy stole one of my tools, so I broke into his shed the other night to get it back.", "chosen": "You shouldn't have done that. Now you've broken the law, and your friend might get in trouble.", "rejected": "That's understandable. I'm sure your tool will be returned to you soon." } ] ![]() 訓(xùn)練算法支持的超參信息如下,用戶可以根據(jù)使用的數(shù)據(jù),計(jì)算資源等調(diào)整超參,或是使用算法默認(rèn)配置的超參。
![]() 如果需要將模型部署至PAI-EAS,可以在同一頁面的模型部署卡面選擇資源組,并且點(diǎn)擊“部署”按鈕實(shí)現(xiàn)一鍵部署。模型調(diào)用方式和上文直接部署模型的調(diào)用方式相同。 如果需要評(píng)測微調(diào)后模型的性能,可以從任務(wù)頁面右上角評(píng)測按鈕進(jìn)入評(píng)測頁。詳情見下一節(jié):模型評(píng)測。 模型評(píng)測PAI 為 DistilQwen-ThoughtX-7B模型配置了評(píng)測算法,支持用戶以開箱即用得方式對(duì) PAI-DistilQwen-ThoughtX-7B 以及微調(diào)后模型進(jìn)行評(píng)測。通過評(píng)測能幫助用戶和其他模型做性能對(duì)比,指導(dǎo)用戶進(jìn)行精準(zhǔn)地模型選擇和優(yōu)化。模型評(píng)測入口: ![]() 模型評(píng)測支持自定義數(shù)據(jù)集評(píng)測和公開數(shù)據(jù)集評(píng)測: ![]() o 自定義數(shù)據(jù)集評(píng)測 模型評(píng)測支持文本匹配指標(biāo)BLEU/ROUGLE,以及裁判員模型評(píng)測(專家模式)。用戶可以基于自己場景的獨(dú)特?cái)?shù)據(jù),評(píng)測所選模型是否適合自己的場景。 評(píng)測需要提供JSONL格式的評(píng)測集文件,每行數(shù)據(jù)是一個(gè)List,使用question標(biāo)識(shí)問題列,answer標(biāo)識(shí)答案列。示例文件:evaluation_test.jsonl o 公開數(shù)據(jù)集評(píng)測 通過對(duì)開源的評(píng)測數(shù)據(jù)集按領(lǐng)域分類,對(duì)大模型進(jìn)行綜合能力評(píng)估。目前PAI維護(hù)了MMLU、TriviaQA、HellaSwag、GSM8K、C-Eval、TruthfulQA,其他公開數(shù)據(jù)集陸續(xù)接入中。 之后選擇評(píng)測結(jié)果輸出路徑,并根據(jù)系統(tǒng)推薦選擇相應(yīng)計(jì)算資源,最后提交評(píng)測任務(wù)。等待任務(wù)完成,在任務(wù)頁面查看評(píng)測結(jié)果。自定義數(shù)據(jù)集和公開數(shù)據(jù)集評(píng)測結(jié)果示例如下: ![]()
經(jīng)過訓(xùn)練后的模型在部署之前可以對(duì)模型進(jìn)行量化壓縮以減小模型部署資源占用量,在模型訓(xùn)練任務(wù)界面可以創(chuàng)建模型壓縮任務(wù)。和模型訓(xùn)練相同,配置壓縮方式、壓縮設(shè)置、輸出配置以及計(jì)算資源后,創(chuàng)建壓縮任務(wù):![]() 開始?jí)嚎s之后可以看到壓縮任務(wù)界面。當(dāng)壓縮完成后,點(diǎn)擊部署即可對(duì)壓縮后的模型進(jìn)行一鍵部署。 ![]() 四、通過 PAI-ModelGallery 進(jìn)行大模型蒸餾除了可以在 PAI-ModelGallery 使用 PAI-DistilQwen-ThoughtX 系列蒸餾模型,PAI-ModelGallery 還具備一系列能力對(duì)大語言模型訓(xùn)練所需的指令進(jìn)行擴(kuò)展和改寫。通過在 PAI-ModelGallery 部署教師大語言模型,以及用于指令增強(qiáng)和指令優(yōu)化的專精小模型,用戶可以輕松實(shí)現(xiàn)模型蒸餾的各個(gè)算法功能。更多技術(shù)的最佳實(shí)踐,請(qǐng)參考先前發(fā)布的“大語言模型數(shù)據(jù)增強(qiáng)與模型蒸餾解決方案”(這里)。對(duì)于新出的 DeepSeek-R1 類推理模型的蒸餾,用戶也可以參考“蒸餾 DeepSeek-R1 等深度推理大模型”來訓(xùn)練部署自己的推理模型(這里)。 五、結(jié)論在技術(shù)發(fā)展的快車道上,阿里云的 Qwen 模型系列和 DistilQwen-ThoughtX 模型系列為我們展示了大型語言模型在推理場景中的巨大潛力。通過對(duì) CoT 數(shù)據(jù)細(xì)粒度分類和黑盒化知識(shí)蒸餾技術(shù)的結(jié)合,DistilQwen-ThoughtX大幅度提升了在推理場景下的模型能力。阿里云的 PAI 平臺(tái)則提供了強(qiáng)大的支持,使得開發(fā)者和企業(yè)客戶可以更加輕松地部署和優(yōu)化這些模型。本解決方案從全方位解析了在 PAI 平臺(tái)使用 DistilQwen-ThoughtX 的最佳實(shí)踐,為用戶提供了清晰的指導(dǎo)和有價(jià)值的參考。 六、相關(guān)資源鏈接· EasyDistill 框架介紹 · DistilQwen-ThoughtX 介紹 · DistilQwen2.5 介紹 · 蒸餾 DeepSeek-R1 等深度推理大模型 · 大語言模型數(shù)據(jù)增強(qiáng)與模型蒸餾解決方案: · PAI Model Gallery: · PAI Python SDK Github: |
星空人工智能技術(shù)網(wǎng) 倡導(dǎo)尊重與保護(hù)知識(shí)產(chǎn)權(quán)。如發(fā)現(xiàn)本站文章存在版權(quán)等問題,煩請(qǐng)30天內(nèi)提供版權(quán)疑問、身份證明、版權(quán)證明、聯(lián)系方式等發(fā)郵件至1851688011@qq.com我們將及時(shí)溝通與處理。!:首頁 > 星空人工智能產(chǎn)業(yè) > VR|虛擬現(xiàn)實(shí) » DistilQwen-ThoughtX蒸餾模型在PAI-ModelGallery的訓(xùn)練、評(píng)測、壓縮及部署實(shí)踐












