人間誠實(shí)周鴻祎：360 All in 大模型的六個(gè)解讀
主筆 / 村口有牛文章架構(gòu)師 / 毛自聰出品 / 巨頭財(cái)經(jīng)5月至今，人間躁動(dòng)，各路大模型你方唱罷我登場，VC圈互聯(lián)網(wǎng)圈媒體圈已近癲狂。誰也沒想到，今日，360再度刷屏，老牌互聯(lián)網(wǎng)巨頭展現(xiàn)
11 月 10 日消息，大語言模型（LLM）快速崛起，在語言生成和理解方面表現(xiàn)出光明的前景，影響超越了語言領(lǐng)域，延伸到邏輯、數(shù)學(xué)、物理學(xué)等領(lǐng)域。
不過想要解鎖這些“非凡能量”，需要付出高額的代價(jià)，例如訓(xùn)練 540B 模型，需要 Project PaLM 的 6144 個(gè) TPUv4 芯片；而訓(xùn)練 175B 的 GPT-3，需要數(shù)千 Petaflop/s-day。
目前一個(gè)不錯(cuò)的解決方案就是低精度訓(xùn)練，可以提高處理速度，降低內(nèi)存使用量和通信成本。包括 Megatron-LM、MetaSeq 和 Colossal-AI 等主流訓(xùn)練系統(tǒng)，默認(rèn)使用 FP16 / BF16 混合精度或 FP32 全精度來訓(xùn)練大型語言模型。
雖然這些精度水平對于大語言模型來說是必不可少的，但它們的計(jì)算成本很高。
如果采用 FP8 低精度，可以將速度提高 2 倍、內(nèi)存成本降低 50% 至 75%，并且可節(jié)省通信成本。
目前只有 Nvidia Transformer Engine 兼容 FP8 框架，主要利用這種精度進(jìn)行 GEMM（通用矩陣乘法）計(jì)算，同時(shí)以 FP16 或 FP32 高精度保持主權(quán)重和梯度。
為了應(yīng)對這一挑戰(zhàn)，來自 Microsoft Azure 和 Microsoft Research 的一組研究人員推出了一個(gè)高效的 FP8 混合精度框架，專為大型語言模型訓(xùn)練量身定制。
微軟引入了三個(gè)優(yōu)化階段，利用 FP8 進(jìn)行分布式和混合精度訓(xùn)練。隨著這些層級的進(jìn)展，F(xiàn)P8 集成程度的提高變得明顯，這表明對 LLM 訓(xùn)練過程的影響更大。
此外為了克服數(shù)據(jù)溢出或下溢等問題，微軟研究人員提出自動(dòng)采樣和精確解耦兩種關(guān)鍵方法，前者涉及對精度不敏感的組件降低精度，動(dòng)態(tài)調(diào)整 Tensor 采樣因子，以確保梯度值保持在 FP8 表示范圍內(nèi)。這可以防止全減少通信期間的下溢和溢流事件，確保培訓(xùn)過程更加順暢。
微軟經(jīng)過測試，與廣泛采用的 BF16 混合精度方法相比，內(nèi)存占用減少 27% 至 42%，權(quán)重梯度通信開銷顯著降低 63% 至 65%。運(yùn)行速度比廣泛采用的 BF16 框架（例如 Megatron-LM）快了 64%，比 Nvidia Transformer Engine 的速度快了 17%。
在訓(xùn)練 GPT-175B 模型時(shí)，混合 FP8 精度框架在 H100 GPU 平臺(tái)上節(jié)省 21% 的內(nèi)存，而且相比較 TE（Transformer Engine），訓(xùn)練時(shí)間減少 17%。
在此附上 GitHub 地址和論文地址：https://doi.org/10.48550/arXiv.2310.18313
本文鏈接：http://www.tebozhan.com/showinfo-45-2519-0.html微軟推出 FP8 混合精度訓(xùn)練框架：比 BF16 快 64%，內(nèi)存占用少 42%
聲明：本網(wǎng)頁內(nèi)容旨在傳播知識(shí)，若有侵權(quán)等問題請及時(shí)與本網(wǎng)聯(lián)系，我們將在第一時(shí)間刪除處理。郵件：2376512515@qq.com
上一篇： GitHub Copilot Enterprise 明年 2 月推出，允許企業(yè)整合自家代碼模型創(chuàng)建專用 AI 助手
下一篇： Humane 可穿戴設(shè)備 Ai Pin 正式發(fā)布：手掌內(nèi)投影、語音手勢交互，699 美元加月訂閱費(fèi)
Copyright ? 2016-2023 天津谷騏科技有限公司 版權(quán)所有 sitemap.xml
違法及侵權(quán)請聯(lián)系：2376512515@qq.com 津ICP備18001702號(hào)
津公網(wǎng)安備 12010102000574號(hào)