跳至主要內容

工具與應用

百度 ERNIE 5.5 Flash Base 完整解析:280B 混合注意力 MoE 基座模型、六大架構創新、30 項基準評測與本地部署指南

百度開源新一代 MoE 基座模型 ERNIE 5.5 Flash Base,總參數 280B、激活 14B,支援 1M 超長上下文。本文深度解析六大架構創新、33.5T 預訓練與 30 項基準評測。

百度 ERNIE 5.5 Flash Base 完整解析:280B 混合注意力 MoE 基座模型、六大架構創新、30 項基準評測與本地部署指南

文章摘要:百度文心(Baidu ERNIE)團隊於 2026 年 11 月 5 日正式發布並開源新一代稀疏混合專家(MoE)基礎語言模型 ERNIE-5.5-Flash-Base(官方中文發布頁 / 英文技術部落格)。該模型總參數量為 2800 億(280B),每處理一個 Token 僅激活 140 億(14B)參數,使用 33.5 兆(33.5T)Tokens 從頭預訓練,訓練階段最大上下文長度為 128K,微調與推論階段則原生支援 100 萬(1M / 1,048,576)Tokens 超長上下文。與一般經過指令微調的對話模型不同,本次釋出的權重是預訓練結束當下的純淨基座檢查點(Pre-training Checkpoint),未經過任何中期訓練(Mid-training)或後訓練(Post-training),旨在為開發者與研究團隊提供領域適配(Domain Adaptation)、指令微調(Instruction Tuning)、強化學習(RL)與代理(Agent)研發的高效率起點。本文完整整理官方技術報告、Hugging Face 模型卡與關聯論文中的六大底層架構設計、預訓練資料與系統優化、30 項同尺寸基座模型評測對比,以及 SGLang 與 vLLM 生產環境部署方法。

ERNIE 5.5 Flash Base 四大核心設計定位

根據 百度 ERNIE 5.5 Flash Base 官方技術發布頁 與 Hugging Face 官方模型卡,研發團隊認為不同垂直領域、推理任務與代理(Agentic)應用需要各自專屬的資料分佈、訓練目標與行為策略;而在後續強化學習反覆採樣(Rollout)與多輪長上下文互動中,運算與記憶體成本會持續累加。因此,ERNIE 5.5 Flash Base 從設計初期便圍繞四個核心目標展開:

  1. 開放純淨的預訓練起點(Open Starting Point for Further Training):直接公開預訓練結束時的原始基座權重(提供 FP8 權重庫 與 BF16 權重庫),不預先綁定特定的後訓練策略,讓研究人員與企業團隊能在統一且乾淨的基準點上,自由驗證各種微調與強化學習演算法的效果,省去從頭進行大規模預訓練的高昂算力成本。

  2. 為後續訓練與長上下文推論量身打造的極致效率:模型將激活算力(Activated Compute)、KV 快取記憶體(KV Cache)與長上下文運算成本視為單一整體預算進行協同設計,透過 37 層重度壓縮注意力(HCA)搭配 6 層多頭潛在稀疏注意力(MLA + DSA),以及運行於半寬度潛在空間(2048 維)的 多路由器潛在混合專家(Multi-Router Latent MoE),大幅降低訓練與推論的資源消耗。

  3. 預埋難以靠後訓練補足的底層能力:預訓練語料庫除了大規模多語系網頁知識外,特別強化了儲存庫級別(Repository-scale)程式碼演進軌跡(涵蓋 Commit、Issue、Pull Request、Code Review 與 CI 驗證修復)與合成長上下文多跳問答資料,為後續的軟體工程代理與長文本檢索打下基礎。

  4. 演算法與底層訓練系統協同設計(Co-designed Infrastructure):為了在數千張 GPU 規模下高效完成 33.5T Tokens 的預訓練,百度團隊同步開發了通訊高效的分散式 Muon 優化器(Distributed Muon)、記憶體均衡分片機制,以及支援超大注意力頭維度的 FlashMask-4 核心(在 NVIDIA SM100 架構上較 FlashAttention-4 MaskMod 前向加速最高達 175%、反向加速最高達 74%)。

底層架構與關鍵技術規格總覽

結合 Hugging Face 官方模型卡、百度官方技術報告 與 vLLM 官方部署配方,ERNIE-5.5-Flash-Base 的核心模型參數配置整理如下表:

規格維度ERNIE-5.5-Flash-Base 詳細配置資料來源
模態與訓練階段純文字基座模型(Text-only Base Model);預訓練結束檢查點(未經 Mid/Post-training)Hugging Face 模型卡
總參數量 / 激活參數總參數 280B(含輸入嵌入與輸出頭);每 Token 激活 14B(計入輸出頭、不含嵌入查表;兩者皆不含 0.36B 的 ECHO 參數)Hugging Face 模型卡 / MarkTechPost
網路層數與隱藏層維度共 43 層 Transformer(第 1 層為稠密 FFN + 後 42 層為 MoE);隱藏層維度(Hidden Size)為 4096百度技術報告
混合稀疏注意力機制共 64 個注意力頭:37 層重度壓縮注意力(HCA,頭維度 512) + 6 層多頭潛在稀疏注意力(MLA-DSA,頭維度 256),全層導入低秩門控注意力與 虛擬頭注意力(VHA)Hugging Face 模型卡 / arXiv:2610.17959
MoE 專家與路由配置每層 MoE 設 512 個路由專家(Top-10 激活)+ 1 個共享專家;採用獨立 Sigmoid 多路由器(Multi-Router),潛在維度(Latent Dim)與專家維度皆為 2048百度技術報告 / arXiv:2610.40577
殘差連接與輸出頭4 通道流形約束超連接(mHC,Expansion Rate = 4)與可學習分段溫度之 MultiMax 輸出頭百度技術報告
多 Token 預測與投機解碼採用 ECHO 自遞迴機制取代傳統外掛 MTP 模組,直接重用骨幹最後一層預測未來 Token 並兼任內建投機解碼草稿模型百度技術報告 / arXiv:2610.24766
分詞器(Tokenizer)全新 BPE 分詞器,詞表大小 201,216;數字採由右至左(R2L)每 3 位數分組切分Hugging Face 模型卡
預訓練規模與優化器使用 Muon 優化器與 FP8 混合精度,從頭預訓練 33.5T Tokens百度技術報告
上下文視窗長度預訓練最大 131,072(128K)Tokens;微調與推論支援 1,048,576(1M)Tokens(釋出配置已預設開啟 1M)Hugging Face 模型卡
開源權重格式與授權提供 FP8(約 280 GB) 與 BF16(約 560 GB) Safetensors;採 ERNIE 5.5 Flash Base Model License v1.0Hugging Face LICENSE / Shattered.io

六大核心架構創新深度解析

根據 百度官方技術部落格 與三篇同步發表於 arXiv 的技術論文,ERNIE 5.5 Flash 從注意力層、專家路由、殘差流到輸出頭進行了全面重構:

1. 潛在門控混合稀疏注意力(Hybrid MLA-DSA + HCA)與 VHA

為了在百萬級上下文下同時兼顧檢索精度與 KV 快取記憶體預算,ERNIE 5.5 Flash 採用了 1:6 的混合注意力配比:

  • 37 層重度壓縮注意力(Heavily Compressed Attention, HCA):多數網路層只需要對遠距離歷史保持粗粒度感知。HCA 將每 128 個連續 Tokens 壓縮為單一快取條目(128 倍壓縮),並同時關注所有壓縮歷史條目與近期未壓縮的短視窗 Tokens。由於序列長度壓縮了 128 倍、注意力計算成本極低,HCA 層得以配置更寬的 512 維注意力頭(Head Dimension = 512) 以強化表達能力。

  • 6 層多頭潛在稀疏注意力(MLA + DSA):每 7 層中保留 1 層具備完整 Token 級別視野的 MLA 層,並將頭維度設為較窄的 256 維。為了避免長序列下 Token 級別注意力運算量暴增,這 6 層結合了動態稀疏注意力(DSA),透過輕量級索引器(Indexer)為每個 Query 對歷史位置打分,僅針對得分最高的 Top-$k$ 位置計算核心注意力,使單次 Query 的注意力成本隨上下文增長維持常數。官方消融實驗證實,在相同 KV 快取預算下,「HCA + 少量全視野 MLA」的組合明顯優於「HCA + 4 倍壓縮稀疏注意力(CSA)」。

  • 低秩門控注意力(Low-Rank Gated Attention)與虛擬頭注意力(VHA):所有 43 層皆在注意力輸出端乘上學習門控(Gate),且門控直接由窄 4 倍的「低秩 Query 潛在向量」計算而非完整隱藏狀態,減少了 75% 的門控投影參數並提升訓練穩定度。此外,模型全面導入入選 NeurIPS 2026 的 虛擬頭注意力(Virtual Head Attention, VHA)(GitHub 開源實作),在注意力計算後加入跨頭低秩混合步驟與恆等跳躍連接(Identity Skip Connection),推論時可直接融合進注意力算子中,以趨近於零的額外開銷促進不同注意力頭之間的資訊共享。

2. 多路由器潛在混合專家(Multi-Router Latent MoE)

傳統 MoE 架構依賴單一路由器(Single Router)搭配硬性 Top-$k$ 選取專家,但在中深層網路中,單一路由器的評分與專家的實際效用(Utility)相關性偏弱;且未被選中的專家無法獲得梯度更新,導致錯誤的路由決策容易自我強化。

根據百度發表的論文《Multi-View MoE: Letting Every Router Shape Expert Computation》與 官方技術報告:

  • 多路由器協同評分(Multi-Router):模型以多個獨立初始化的 Sigmoid 路由器取代單一路由器。訓練過程中,不同路由器會自然分化出相異的專家偏好,一個專家可因單一路由器強烈推薦或多個路由器共同支持而被選中,有效降低對雜訊特徵與分佈偏移的敏感度。在 10B-A1B 對照實驗中,Multi-Router 將全局路由分數與專家效用的秩相關性提升了 8.1%(在單路由器對齊最差的網路層更相對提升達 58%),且激活專家數、FLOPs 與跨節點通訊量完全維持不變。

  • 半寬度潛在空間專家(Latent Experts)與穩定化設計:除第 1 層為稠密層外,其餘 42 層皆配置 512 個路由專家(每 Token 激活 Top-10)與 1 個共享專家,並採用無輔助損失(Auxiliary-loss-free)的偏差修正負載均衡。Token 進入 MoE 層時會先從 4096 維投影至 2048 維潛在空間,專家網路在 2048 維空間內完成運算後,再透過共享的上投影矩陣映射回 4096 維,使專家分發/合併(Dispatch/Combine)通訊量與專家運算量雙雙減半。同時結合 Stable LatentMoE 設計(上投影前加入 RMSNorm 並採用 SiTU-GLU 激活函數)與每個路由專家專屬的可學習純量權重,確保 43 層網路的專家輸出尺度穩定。

3. 流形約束超連接(Manifold-Constrained Hyper-Connections, mHC)

ERNIE 5.5 Flash Base 以擴展率為 4 的 流形約束超連接(mHC) 取代標準殘差連接。每一層網路皆透過受約束的混合矩陣(Constrained Mixing Matrix)同時讀寫 4 條平行殘差流,在幾乎不增加額外運算量的前提下,大幅拓寬了跨層訊號傳遞的通道寬度。

4. MultiMax 輸出頭:可學習分段溫度消除 Softmax 兩難

傳統輸出層使用單一 Softmax 溫度時面臨兩難:降低溫度雖能使分佈稀疏,卻會壓抑其他合理的候選 Token;調高溫度雖能保留合理候選,卻會讓不合理的長尾噪音機率膨脹。根據 官方技術報告,MultiMax 輸出頭改用「可學習的分段溫度(Learnable Piecewise Temperature)」,在壓抑長尾無效 Token 的同時保留多個高機率合理候選。該元素級別的 Logits 縮放直接融合在分塊(Chunked)Triton 交叉熵核心的 Softmax 之前執行,不佔用額外顯示記憶體,並顯著降低了預訓練損失。

5. ECHO 自遞迴機制:取代外掛 MTP 模組並內建投機解碼

傳統多 Token 預測(MTP,如 DeepSeek-V3 等架構採用的方案)會在主幹網路外附加額外的預測模組,導致未來 Token 的監督訊號主要用於訓練外掛模組,主幹本身仍僅受下一個 Token 的監督。

根據百度論文《ECHO: Efficient Causal Self-Recursion Beyond Auxiliary Multi-Token Prediction》與 官方技術報告,ERNIE 5.5 Flash 採用單步 ECHO(Efficient Causal Self-Recursion) 取代 MTP 模組,直接重用骨幹網路自己的最後一層(Final Decoder Layer)與輸出頭來預測未來 Token(僅透過約 0.36B 參數的輕量融合層連接)。由於沒有獨立的 MTP 層,未來視角的損失會直接回傳並優化產生下一個 Token 的核心參數,促使主幹模型預判序列走向,在數學推理與程式碼任務上帶來顯著增益。在推論階段,同一層可直接作為內建投機解碼(Speculative Decoding)的草稿模型(Drafter),且具備跨深度泛化能力:即使預訓練僅針對預測額外 1 個 Token 訓練,推論時仍能連續起草多個候選 Token,接受率(Acceptance Rate)亦高於同等訓練方式的外掛 MTP 模組。

6. 右至左(R2L)三位數分組 BPE 分詞器

ERNIE 5.5 Flash Base 採用詞表容量達 201,216 的全新 BPE 分詞器。針對數值處理,分詞器會將數字由右至左(Right-to-Left, R2L)以最多 3 位數為一組進行切分(例如 1,234,567 會被切分為 1、234、567),確保不同長度的數字在相同位值(個十百、千、百萬)上嚴格對齊,有效提升算術運算與密集數值文本的處理準確度。

33.5T 預訓練資料工程與分散式訓練基礎設施

根據 百度官方技術報告第 3 節,ERNIE-5.5-Flash-Base 的預訓練語料規模達 33.5 兆(33.5T)Tokens,訓練上下文逐步擴展至 128K,並在資料組成與底層系統上進行了三項針對性強化:

  1. 三維度高密度語料(網頁、動態軟體工程、長上下文):

    • 多語系高保真網頁資料:經大規模爬取、高保真擷取與多維度品質評分模型過濾,確保高知識密度。
    • 儲存庫級程式碼與演化軌跡(Repository-level Code & Evolution):除了經過嚴格去重與多粒度分級的靜態程式碼外,語料庫特別納入軟體開發的動態演化與執行驗證資料,涵蓋 Commits、Issues、Pull Requests、Code Reviews、「需求至 PR(Requirement-to-PR)」與多 Commit 軌跡、自動生成測試以及 CI 驗證通過的修復紀錄(CI-verified fixes),強化儲存庫級別推理與代理軟體工程能力。
    • 三源互補長上下文資料:結合(1)經嵌入去重與 LLM 品質篩選的自然長文件、(2)在自然長文中穿插語義相似檢索干擾項(Distractors)拼接而成的長序列,以及(3)針對 MRCR 風格多針檢索、引文歸因問答(Citation-grounded QA)與多跳問答(Multi-hop QA)生成的合成資料。
  2. 通訊高效且記憶體均衡的分散式 Muon 優化器:不同於 Adam 等逐元素優化器,Muon 會對整個參數矩陣的更新進行正交化(Orthogonalization)。百度團隊針對千卡級叢集進行了兩項改造:

    • 稠密與 MoE 參數同步融合:將原本透過全叢集單一環狀(Ring)傳遞的稠密參數同步,移至與 MoE 專家平行相同的「跨節點同 Rank(Same-rank)」通訊路徑,並將稠密與 MoE 傳輸合併為單一集合通訊操作(Collective Operation),大幅增大單次訊息體積、提升頻寬利用率並消除跨節點延遲抖動。
    • 跨節點記憶體均衡分片(Memory-balanced Sharding):解決傳統 Muon 為維持大訊息而將優化器狀態集中在少數前段 GPU 節點導致的 OOM 與檢查點存取瓶頸,將優化器狀態均勻分散至各節點,同時維持高效通訊所需的大訊息尺寸。
  3. 支援超大頭維度的 FlashMask-4 注意力核心:由於 HCA 層採用高達 512 維的注意力頭,傳統 FlashAttention 核心容易受限於晶片共享記憶體(Shared Memory)與張量記憶體(Tensor Memory)容量。百度基於 CuTe-DSL 為 NVIDIA SM90(Hopper) 與 SM100(Blackwell) 架構打造了 FlashMask-4,透過 2-CTA 鋪磚平行計算與 SM100 張量記憶體雙緩衝輪轉(Double-buffer Rotation),支援高達 qk_head_dim = 576 與 v_head_dim = 512(超越 FlashAttention-4 上限),並保留 $O(N)$ 欄位遮罩表示法以跳過全遮罩區塊,在 SM100 GPU 上實現前向最高加速 175%、反向最高加速 74%。

30 項基準測試完整評測:對決 DeepSeek-V4-Flash、Hy3-preview 與 MiMo-V2.5

根據 Hugging Face 官方模型卡 與 百度官方發布頁,官方採用完全統一的評測管線、提示詞、Few-shot 設定與相同的自由問答 LLM 裁判,將 ERNIE-5.5-Flash-Base(280B-A14B)與三款同量級開源基座模型——DeepSeek-V4-Flash-Base(284B-A13B)、騰訊混元 Hy3-preview-Base(295B-A21B)以及小米 MiMo-V2.5-Base(310B-A15B)——在涵蓋綜合考試、知識問答、STEM 數理、邏輯推理、多語系、程式碼與長上下文的 7 大領域共 30 項基準測試中進行了完整橫向對比:

評測領域基準測試項目(評估指標)Few-Shot 設定ERNIE-5.5-Flash-Base (280B-A14B)DeepSeek-V4-Flash-Base (284B-A13B)Hy3-preview-Base (295B-A21B)MiMo-V2.5-Base (310B-A15B)
綜合學術考試 (Exam)MMLU (EM)5-shot93.389.487.587.6
綜合學術考試 (Exam)MMLU-Redux (EM)5-shot94.190.988.789.4
綜合學術考試 (Exam)MMLU-Pro (EM)5-shot78.469.266.965.7
綜合學術考試 (Exam)CMMLU (EM)5-shot90.089.789.387.7
事實知識問答 (Knowledge)SimpleQA (LLM-Judge)10-shot34.129.125.920.8
事實知識問答 (Knowledge)ChineseSimpleQA (LLM-Judge)10-shot73.075.373.268.8
事實知識問答 (Knowledge)PreciseWikiQA (LLM-Judge)10-shot59.352.050.346.0
事實知識問答 (Knowledge)HotpotQA (LLM-Judge)10-shot57.351.850.950.1
事實知識問答 (Knowledge)PopQA (LLM-Judge)10-shot52.248.749.345.7
科學與數學 (STEM)GPQA (EM)5-shot60.255.950.251.4
科學與數學 (STEM)MGSM (EM)8-shot CoT87.084.676.983.8
科學與數學 (STEM)MathBench-CN (LLM-Judge)5-shot80.878.075.365.6
邏輯與常識推理 (Reasoning)BBH (direct) (EM)3-shot77.169.067.568.0
邏輯與常識推理 (Reasoning)BBH (CoT) (EM)3-shot CoT89.287.590.089.8
邏輯與常識推理 (Reasoning)WinoGrande (EM)5-shot87.884.982.485.2
邏輯與常識推理 (Reasoning)PIQA (EM)5-shot96.095.095.293.6
多語系理解 (Multilingual)MMMLU-lite (EM)5-shot81.078.576.172.2
多語系理解 (Multilingual)INCLUDE-lite (EM)5-shot79.579.877.774.5
多語系理解 (Multilingual)MultiLoKo (LLM-Judge)5-shot43.342.839.536.3
程式設計與軟體工程 (Code)HumanEval+ (Pass@1)1-shot82.369.578.074.4
程式設計與軟體工程 (Code)MBPP+ (Pass@1)0-shot81.567.278.374.3
程式設計與軟體工程 (Code)LiveCodeBench (Pass@1)1-shot40.131.933.532.4
程式設計與軟體工程 (Code)FullStackBench (Pass@1)1-shot54.643.356.754.1
程式設計與軟體工程 (Code)RepoBench (EM)0-shot50.349.149.148.9
長上下文處理 (Long Context)HELMET-RAG (SubEM)2-shot71.568.967.269.0
長上下文處理 (Long Context)HELMET-Rerank (NDCG@10)0-shot56.260.055.756.6
長上下文處理 (Long Context)HELMET-ICL (EM)many-shot85.983.883.080.2
長上下文處理 (Long Context)RepoQA (Pass@1)0-shot91.781.091.356.5
長上下文處理 (Long Context)LongBench-v2 (EM)0-shot48.744.539.844.6
長上下文處理 (Long Context)InfiniteBench (Avg)0-shot46.548.033.046.7

評測結果的三大關鍵觀察

  1. 在 30 項基座測試中拿下 24 項第一(對比 DeepSeek-V4-Flash-Base 則勝出 26 項):根據 官方公布數據 與 MarkTechPost 分析,ERNIE-5.5-Flash-Base 在 30 項基準中有 24 項位居四款同級模型之冠。尤其在高難度學術考試 MMLU-Pro 上拿到 78.4 分,領先 DeepSeek-V4-Flash-Base(69.2 分)達 9.2 個百分點;在博士級科學推理 GPQA 亦取得 60.2 分(領先次高 DeepSeek 的 55.9 分達 4.3 個百分點)。

  2. 程式碼生成與儲存庫級檢索優勢明顯:受惠於預訓練階段導入的軟體演化軌跡與 ECHO 自遞迴監督,模型在 HumanEval+(82.3% vs. DeepSeek 69.5%)、MBPP+(81.5% vs. DeepSeek 67.2%)與競賽級編程 LiveCodeBench(40.1% vs. Hy3-preview 33.5%)皆取得明顯領先;在長上下文程式碼檢索基準 RepoQA 更達到 91.7%,超越 DeepSeek-V4-Flash-Base(81.0%)逾 10 個百分點。

  3. 128 倍壓縮注意力下仍維持高水準長上下文表現:儘管 43 層網路中有 37 層採用了 128 倍序列壓縮的 HCA 注意力,但憑藉 6 層全視野 MLA-DSA 層與針對性長文本預訓練語料,模型在 HELMET-RAG(71.5)、HELMET-ICL(85.9)、RepoQA(91.7)與 LongBench-v2(48.7)四項長上下文基準中均排名第一,僅在 HELMET-Rerank(56.2 vs. DeepSeek 60.0)與 InfiniteBench(46.5 vs. DeepSeek 48.0)略低於對照組。

本地部署與推論實戰指南(Transformers、SGLang、vLLM)

根據 Hugging Face 官方模型卡 與 百度官方快速開始指南,ERNIE-5.5-Flash-Base 已在 Hugging Face 與 ModelScope 同步釋出 FP8 權重(約 280 GB) 與 BF16 權重(約 560 GB)。由於此模型為未經對話微調的基座模型(Base Model),部署後應透過 /v1/completions 端點(而非 /v1/chat/completions)進行文本續寫呼叫,且模型預期每筆輸入開頭皆帶有 BOS Token,因此分詞時請務必維持 add_special_tokens=True。

1. 使用 Hugging Face Transformers 直接載入推論

適合用於微調腳本開發或單次前向驗證(以下以載入 baidu/ERNIE-5.5-Flash-Base-BF16 為例):

from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

model_name = "baidu/ERNIE-5.5-Flash-Base-BF16"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(
    model_name,
    torch_dtype=torch.bfloat16,
    device_map="auto",
)

# 注意:模型需要輸入開頭包含 BOS token,請保持 add_special_tokens=True
model_inputs = tokenizer(
    ["The hybrid attention architecture of ERNIE 5.5 Flash"],
    add_special_tokens=True,
    return_tensors="pt",
).to(model.device)

outputs = model.generate(**model_inputs, max_new_tokens=256)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))

2. 使用 SGLang 進行生產環境部署(支援 4 卡平行與 1M 上下文)

根據 Hugging Face 模型卡說明 與 SGLang Cookbook 指南,官方提供了預先建置好的專用 Docker 映像檔 lmsysorg/sglang:dev-ernie-55-flash,可透過 4 路張量平行(--tp 4)與注意力資料平行(--attn-dp-size 4)啟動服務:

# 1. 啟動官方預建置的 SGLang Docker 容器
docker run -d -it --name ernie55 \
  --gpus all \
  --network host \
  --ipc host \
  -v ~/.cache/huggingface:/root/.cache/huggingface \
  lmsysorg/sglang:dev-ernie-55-flash

# 2. 在容器內啟動 SGLang 推論伺服器(支援最高 1,048,576 context-length)
sglang serve \
  --model-path baidu/ERNIE-5.5-Flash-Base \
  --tp 4 \
  --attn-dp-size 4 \
  --context-length 131072 \
  --max-running-requests 1024 \
  --host 0.0.0.0 \
  --port 30000

3. 使用 vLLM 搭配 vLLM-ERNIE 外掛部署

若採用 vLLM 推論引擎,根據 Hugging Face 模型卡 與 vLLM Recipes 配方頁,需安裝 vllm==0.32.0 並掛載百度官方開源的 vLLM-ERNIE 擴充套件,即可啟用 fp8 KV 快取、專家平行(Expert Parallel)與 humming MoE 後端:

# 1. 建立 Python 3.12 虛擬環境並安裝 vLLM 0.32.0 與 vLLM-ERNIE 外掛
uv venv --python 3.12 ernie55-venv
source ernie55-venv/bin/activate
uv pip install vllm==0.32.0
uv pip install git+https://github.com/baidu/vLLM-ERNIE.git

# 2. 啟動 vLLM 伺服器(長上下文推論可將 --max-model-len 調升至 1048576)
vllm serve baidu/ERNIE-5.5-Flash-Base \
  --kv-cache-dtype fp8 \
  --enable-expert-parallel \
  --data-parallel-size 4 \
  --max-model-len 131072 \
  --port 30000 \
  --moe-backend humming

4. 發送 Completions API 測試請求

無論使用 SGLang 或 vLLM 啟動服務,皆可透過標準 OpenAI 相容的 /v1/completions 端點驗證推論結果:

curl http://localhost:30000/v1/completions \
  -H "Content-Type: application/json" \
  -d '{
    "model": "baidu/ERNIE-5.5-Flash-Base",
    "prompt": "Large language models are",
    "max_tokens": 256,
    "temperature": 0
  }'

授權條款注意事項與後續版本展望

在採用 ERNIE-5.5-Flash-Base 進行商業開發或學術研究前,建議留意以下兩點關鍵資訊:

  • 授權條款(ERNIE 5.5 Flash Base Model License Agreement v1.0):根據模型倉庫內的 官方 LICENSE 文件 與 外媒授權解析,該授權允許學術研究與自由修改分發;針對相關產品或服務**年營收低於或等於 5,000 萬美元(US$50 million)的企業可免費進行商業使用(超過該營收門檻需另行向百度申請商業授權),但授權地理範圍明確排除了歐盟(EU)、英國(UK)與南韓(South Korea)**境內的使用與分發。

  • 後訓練版本(Post-trained ERNIE 5.5 Flash)即將推出:百度官方技術報告結語 明確預告,本次釋出的 Base 權重僅是系列的第一步,經過完整後訓練(支援對話、深度推理與工具代理能力)的正式版 ERNIE 5.5 Flash 將於後續發布。

常見問題解答(FAQ)

Q1:ERNIE-5.5-Flash-Base 可以直接用來做聊天機器人(Chat)嗎?

不行。根據 Hugging Face 官方模型卡,ERNIE-5.5-Flash-Base 是預訓練結束時的純基座檢查點,尚未經過指令微調(SFT)或強化學習(RLHF),因此不具備遵循對話格式或工具呼叫的行為策略。一般工程團隊應將其作為垂直領域繼續預訓練(Continual Pre-training)、指令微調或 RL 演算法研究的基底;若需要開箱即用的對話與代理模型,建議等待百度後續釋出的 Post-trained 版本,或自行基於此檢查點執行後訓練。

Q2:應該選擇下載 FP8 版本還是 BF16 版本?硬體記憶體需求為何?

根據 Shattered.io 硬體規格分析 與 官方模型卡:

  • ERNIE-5.5-Flash-Base(FP8 Safetensors):每個參數佔用 1 Byte,純權重體積約 280 GB(未計入 KV 快取與激活值),適合直接用於長上下文推論服務、投機解碼驗證與顯示記憶體受限的實驗環境(官方 SGLang 與 vLLM 範例預設採用 4 路平行配置)。

  • ERNIE-5.5-Flash-Base-BF16(BF16 Safetensors):每個參數佔用 2 Bytes,純權重體積約 560 GB,適合需要以更高數值精度作為起點的全參數微調、LoRA 微調或強化學習訓練工作流。

相關參考連結

謝謝你,讀到這裡。

保持好奇,明天繼續探索。

© 2026 Communeify. All rights reserved.