隨著大語(yǔ)言模型在文本分析、智能問答等場(chǎng)景的廣泛應(yīng)用,處理1M超長(zhǎng)文本推理時(shí),常常面臨顯存不足、運(yùn)算速度卡頓的行業(yè)痛點(diǎn),嚴(yán)重限制了超長(zhǎng)文本場(chǎng)景的應(yīng)用。近日,上海交通大學(xué)李健教授團(tuán)隊(duì)依托上海交通大學(xué) 鯤鵬昇騰科教創(chuàng)新卓越中心的算力支持,基于vLLM-Ascend 推理框架研發(fā)出一套針對(duì)超長(zhǎng)上下文推理的稀疏注意力 KV Cache 分層緩存管理系統(tǒng)。在昇騰 AI 軟硬件平臺(tái)的全方位賦能下,該項(xiàng)目成功破解單卡支持超長(zhǎng)上下文推理的顯存與性能雙重難題,同時(shí)大幅提升吞吐量。

項(xiàng)目核心創(chuàng)新在于設(shè)計(jì)了 KV Cache 分級(jí)緩存集成機(jī)制。該機(jī)制先對(duì)推理任務(wù)進(jìn)行實(shí)時(shí)分析,智能識(shí)別Top-K 重要塊并集中算力處理,從源頭提升計(jì)算效率;同時(shí)采用數(shù)據(jù)冷熱分層存儲(chǔ)策略,根據(jù)數(shù)據(jù)訪問頻率,將生成數(shù)據(jù)動(dòng)態(tài)劃分為高頻熱數(shù)據(jù)與低頻冷數(shù)據(jù),再針對(duì)性優(yōu)化存儲(chǔ)位置,減少資源浪費(fèi)。這一機(jī)制的落地依托昇騰CANN異構(gòu)計(jì)算架構(gòu)靈活的動(dòng)態(tài)調(diào)度能力,能精準(zhǔn)控制冷熱數(shù)據(jù)在顯存與主存間的流轉(zhuǎn),大幅降低數(shù)據(jù)遷移開銷。最終,該方案實(shí)現(xiàn)單卡流暢處理超過1M的超長(zhǎng)文本推理任務(wù),系統(tǒng)推理吞吐量超過39%,徹底突破傳統(tǒng)系統(tǒng)在長(zhǎng)序列處理上的顯存與性能瓶頸。
同時(shí)項(xiàng)目進(jìn)行了元數(shù)據(jù)結(jié)構(gòu)優(yōu)化與緩存機(jī)制設(shè)計(jì),其中數(shù)據(jù)索引與掩碼是關(guān)鍵支撐 —— 通過精簡(jiǎn)索引結(jié)構(gòu)、合并掩碼維護(hù)步驟,有效減少重復(fù)運(yùn)算,使昇騰NPU算力更集中于注意力計(jì)算與文本生成等核心任務(wù),提升硬件利用效率。相關(guān)優(yōu)化已通過vLLM-Ascend推理框架靈活集成,保障了技術(shù)方案的順利落地。
目前,該項(xiàng)目源代碼已在 Gitee 社區(qū)中開源,后續(xù)將進(jìn)一步推送到昇騰開源生態(tài),合入GitHub社區(qū) vLLM-Ascend 項(xiàng)目專區(qū)。此次技術(shù)突破,不僅為超長(zhǎng)文本推理提供了高效解決方案,更印證了昇騰生態(tài)在AI創(chuàng)新中的賦能價(jià)值。未來(lái),隨著該系統(tǒng)在更多行業(yè)場(chǎng)景的落地,昇騰將持續(xù)為AI技術(shù)研發(fā)提供算力與技術(shù)保障,推動(dòng)大語(yǔ)言模型在長(zhǎng)文本分析、智能辦公、數(shù)字孿生等千行百業(yè)的深度應(yīng)用,加速人工智能產(chǎn)業(yè)化進(jìn)程。
請(qǐng)輸入留言內(nèi)容
我要留言
請(qǐng)輸入留言內(nèi)容
我要留言
確定留言
|
|
騰訊QQ未保體系升級(jí)“新芽守護(hù)行動(dòng)”,為青少年筑起網(wǎng)絡(luò)安全護(hù)欄當(dāng)數(shù)字技術(shù)深度融入青少年的學(xué)習(xí)、社交與成長(zhǎng),如何從源頭上構(gòu)建安全、健康的網(wǎng)絡(luò)環(huán)境,已成為全社會(huì)共同..
|
|
|
|
聲動(dòng)京津冀”公益行動(dòng)啟動(dòng)—學(xué)大教育以閱讀賦能區(qū)域文化協(xié)同發(fā)展5月31日,“聲動(dòng)京津冀?書香家庭誦讀者”暨首屆京津冀兒童少年閱讀推廣公益行動(dòng)正式啟幕。本次活動(dòng)由中..
|
|
|
|
算力筑基,智教未來(lái):浪潮計(jì)算機(jī)產(chǎn)教協(xié)同高質(zhì)量發(fā)展會(huì)議圓滿召開5月29日,浪潮計(jì)算機(jī)“算力筑基,智教未來(lái)——產(chǎn)教協(xié)同高質(zhì)量發(fā)展會(huì)議”在濟(jì)南浪潮科技園成功舉辦。本次..
|
|
|
|
科儀創(chuàng)未來(lái)丨2026年全國(guó)大學(xué)生科學(xué)儀器創(chuàng)新大賽初賽成功舉辦近日,2026年全國(guó)大學(xué)生科學(xué)儀器創(chuàng)新大賽初賽在中關(guān)村發(fā)展集團(tuán)旗下專業(yè)特色園區(qū)中關(guān)村集成電路設(shè)計(jì)園(以..
|
|
|
|
學(xué)大教育首屆526公益行動(dòng),聚焦“開明閱讀成長(zhǎng)計(jì)劃”2026年5月26日,學(xué)大教育集團(tuán)攜手開明慈善基金會(huì)在北京啟動(dòng)首屆“526公益行動(dòng)”。作為深交所主板的上市公..
|
|
|
|
全國(guó)首個(gè)心青年就業(yè)支持“共同行動(dòng)單位”成立一座城市的文明高度,往往取決于它如何對(duì)待最需要幫助的群體。2026年5月21日,在上海舉行的“改變的力量..
|
|
|
|
從穩(wěn)定增長(zhǎng)到價(jià)值重估:學(xué)大教育25周年的新起點(diǎn)5月15日,廈門轄區(qū)上市公司2025年年報(bào)業(yè)績(jī)說明會(huì)如期舉行。不同于許多企業(yè)在業(yè)績(jī)會(huì)上小心翼翼地解釋經(jīng)營(yíng)..
|
|
|
|
湖北首家華為產(chǎn)教融合全國(guó)樣板點(diǎn)落地武漢華夏理工學(xué)院AI越“智能”,人才越“稀缺”——這并非悖論,而是當(dāng)下高質(zhì)量ICT人才市場(chǎng)的真實(shí)寫照。畢馬威《2026年全..
|
|
|
|
行動(dòng)亞洲微塑料主題課堂覆蓋全國(guó)3萬(wàn)青少年2026年4月20日至26日,深圳市青少年發(fā)展基金會(huì)—生命關(guān)懷教育專項(xiàng)基金(下稱“行動(dòng)亞洲”)作為世界地球..
|
|
|
|
南京商業(yè)學(xué)校中澳研學(xué)圓滿收官近日,南京商業(yè)學(xué)校師生代表團(tuán)遠(yuǎn)赴澳大利亞米爾迪拉市,在維多利亞州公辦TAFE名校桑瑞亞理工學(xué)院(SuniTAFE)..
|
|
|
首頁(yè)
|
紅人堂
|
發(fā)布
|
我發(fā)的
|
更多
|