2月24日,DeepSeek啟動(dòng)“開源周”,開源了首個(gè)代碼庫(kù)FlashMLA。
據(jù)介紹,這是DeepSeek針對(duì)Hopper GPU優(yōu)化的高效MLA解碼內(nèi)核,專為處理可變長(zhǎng)度序列而設(shè)計(jì),現(xiàn)在已經(jīng)投入生產(chǎn)使用!霸贖800上能實(shí)現(xiàn)3000 GB/s 的內(nèi)存帶寬 & 580 TFLOPS 的計(jì)算性能!盌eepSeek說(shuō)。
簡(jiǎn)單來(lái)說(shuō),F(xiàn)lashMLA 是一個(gè)能讓大語(yǔ)言模型在 H800這樣的GPU上跑得更快、更高效的優(yōu)化方案,尤其適用于高性能AI任務(wù)。這一代碼能夠加速大語(yǔ)言模型的解碼過(guò)程,從而提高模型的響應(yīng)速度和吞吐量,這對(duì)于實(shí)時(shí)生成任務(wù)(如聊天機(jī)器人、文本生成等)尤為重要。
MLA (Multi-Layer Attention,多層注意力機(jī)制)是一種改進(jìn)的注意力機(jī)制,旨在提高Transformer模型在處理長(zhǎng)序列時(shí)的效率和性能。MLA通過(guò)多個(gè)頭(head)的并行計(jì)算,讓模型能夠同時(shí)關(guān)注文本中不同位置和不同語(yǔ)義層面的信息,從而更全面、更深入地捕捉長(zhǎng)距離依賴關(guān)系和復(fù)雜語(yǔ)義結(jié)構(gòu)。
此前,有從業(yè)者解析DeepSeek架構(gòu)時(shí)提到,MLA的本質(zhì)是對(duì)KV(Key-Value,一種緩存機(jī)制)的有損壓縮,提高存儲(chǔ)信息,“該技術(shù)首次在DeepSeek-V2中引入,MLA是目前開源模型里顯著減小KV 緩存大小的最佳方法!
DeepSeek開源這一代碼有何影響?記者用這一問(wèn)題詢問(wèn)了DeepSeek,它回答稱,這一代碼就像給AI推理引擎裝上了一臺(tái)“渦輪增壓器“,讓大模型在處理復(fù)雜任務(wù)時(shí)更快、更省資源,同時(shí)降低了技術(shù)門檻。FlashMLA的意義不單是技術(shù)優(yōu)化,更是打破算力壟斷、加速AI普惠的關(guān)鍵一步。
具體來(lái)說(shuō),F(xiàn)lashMLA可以突破GPU算力瓶頸,降低成本。傳統(tǒng)解碼方法在處理不同長(zhǎng)度的序列(如翻譯不同長(zhǎng)度的句子)時(shí),GPU的并行計(jì)算能力會(huì)被浪費(fèi),就像用卡車運(yùn)小包裹,大部分空間閑置。而FlashMLA的改進(jìn)是:通過(guò)動(dòng)態(tài)調(diào)度和內(nèi)存優(yōu)化,讓Hopper GPU(如H100)的算力被“榨干”,相同硬件下吞吐量顯著提升。這意味著企業(yè)可以用更少的GPU服務(wù)器完成同樣的任務(wù),直接降低推理成本。
另一方面,F(xiàn)lashMLA可以推動(dòng)大模型落地應(yīng)用?勺冮L(zhǎng)度序列是現(xiàn)實(shí)場(chǎng)景中的常態(tài)(如聊天對(duì)話、文檔生成),但傳統(tǒng)方法需要填充(Padding)到固定長(zhǎng)度,導(dǎo)致計(jì)算冗余。FlashMLA支持動(dòng)態(tài)處理變長(zhǎng)輸入,讓AI應(yīng)用(如客服機(jī)器人、代碼生成)響應(yīng)更快、更流暢,用戶體驗(yàn)提升,加速商業(yè)化落地。
此前高效解碼內(nèi)核多由科技巨頭閉源壟斷(如CUDA優(yōu)化庫(kù)),中小企業(yè)和研究者難以復(fù)現(xiàn)。FlashMLA開源后,開發(fā)者可免費(fèi)獲得“工業(yè)級(jí)優(yōu)化方案”,降低技術(shù)門檻,促進(jìn)更多創(chuàng)新應(yīng)用(如垂直領(lǐng)域小模型)的誕生。
"the whale is making waves!(鯨魚正在掀起波浪。庇芯W(wǎng)友在DeepSeek的帖子下留言稱。(注:DeepSeek的企業(yè)LOGO是鯨魚)。
也有網(wǎng)友希望DeepSeek開源網(wǎng)頁(yè)搜索(Web Search)相關(guān)的代碼,并提到,“DeepSeek 是真正的OpenAI(開放人工智能)”。
這僅僅是開始,上周2月21日DeepSeek宣布,下周開始,會(huì)陸續(xù)開源5個(gè)代碼庫(kù),“以完全透明的方式分享我們微小但真誠(chéng)的進(jìn)展”。DeepSeek表示,這些在線服務(wù)中的基礎(chǔ)構(gòu)建模塊已經(jīng)經(jīng)過(guò)文檔化、部署,并在生產(chǎn)環(huán)境中經(jīng)過(guò)實(shí)戰(zhàn)檢驗(yàn)。
DeepSeek在公告中稱自己是探索 AGI 的小公司,作為開源社區(qū)的一部分,每分享一行代碼,都會(huì)成為加速AI行業(yè)發(fā)展的集體動(dòng)力。同時(shí),DeepSeek 稱,沒(méi)有高不可攀的象牙塔,只有純粹的車庫(kù)文化(不少美國(guó)著名企業(yè)從車庫(kù)里誕生)和社區(qū)驅(qū)動(dòng)的創(chuàng)新。
(本文來(lái)自第一財(cái)經(jīng))