Baidu की ERNIE टीम ने ERNIE-5.5-Flash-Base के जारी होने की घोषणा की है, जो ERNIE 5.5 Flash मॉडल का एक अनप्रोसेस्ड प्री-ट्रेनिंग चेकपॉइंट है। यह रिलीज़ 5 नवंबर को Hugging Face प्लेटफॉर्म पर पोस्ट किया गया था।
विरल आर्किटेक्चर मिक्सचर-ऑफ-एक्सपर्ट्स (MoE) का उपयोग करने वाला यह मॉडल कुल 280 बिलियन मापदंडों का है, जिसमें प्रति टोकन 14 बिलियन पैरामीटर सक्रिय होते हैं। इसे 33.5 ट्रिलियन टोकन पर स्क्रैच से प्रशिक्षित किया गया था, जिसका अधिकतम प्रशिक्षण संदर्भ आकार 128 हजार टोकन है, हालांकि फाइन-ट्यूनिंग और इन्फेरेंस के लिए संदर्भ समर्थन 1 मिलियन टोकन तक पहुंचता है।
Baidu ने जानबूझकर मॉडल को अपूर्ण रूप में जारी किया है। वेट्स प्री-ट्रेनिंग के समापन चरण से प्राप्त किए गए हैं और अभी तक इंटरमीडिएट या बाद के प्रशिक्षण चरणों से नहीं गुजरे हैं। कंपनी का दावा है कि यह डोमेन अनुकूलन, इंस्ट्रक्शन ट्यूनिंग और रीइन्फोर्समेंट लर्निंग में अनुसंधान के लिए एक सामान्य शुरुआती बिंदु बनाता है। टीम ने बताया कि बाद में ERNIE 5.5 Flash का पोस्ट-ट्रेन्ड संस्करण जारी किया जाएगा।
उपलब्ध वेट्स Baidu ERNIE 5.5 Flash Base Model v1.0 लाइसेंसिंग समझौते के अनुसार FP8 और BF16 प्रारूपों में प्रदान किए जाते हैं। इस मॉडल का उपयोग SGLang और vLLM के माध्यम से किया जा सकता है, और NVIDIA Megatron-Bridge रिपॉजिटरी में फाइन-ट्यूनिंग का एक उदाहरण भी शामिल है।
यह 43-परत वाला मॉडल दक्षता पर ध्यान केंद्रित करते हुए डिज़ाइन किया गया है। अधिकांश परतें 'Baidu अत्यधिक संपीड़ित ध्यान' का उपयोग करती हैं, जो प्रत्येक 128 टोकन को एक कैश प्रविष्टि में संयोजित करता है। केवल सात परतों में से एक टोकन स्तर पर पूर्ण संदर्भ तक पहुंच बनाए रखती है, जो टॉप-k पर विरल चयन के साथ मल्टी-हेड लेटेंट अटेंशन के माध्यम से होती है। कुल मिलाकर, 43 में से 37 परतें संपीड़ित अनुक्रम के साथ काम करती हैं, जिससे एक छोटा KV-कैश बनाए रखना संभव होता है। इसके अलावा, डेवलपर्स ने बताया कि लो-रैंक गेटेड अटेंशन डिज़ाइन गेट प्रोजेक्शन मापदंडों को 75% तक कम करता है।
MoE के संबंध में, Baidu ने मानक एकल राउटर को कई स्वतंत्र रूप से आरंभ किए गए राउटर से बदल दिया है जो विशेषज्ञों का सहयोगात्मक मूल्यांकन करते हैं। कंपनी का दावा है कि यह रूटिंग मूल्यांकन और विशेषज्ञ की वास्तविक उपयोगिता के बीच मिलान में सुधार करता है, जबकि सक्रिय विशेषज्ञों की संख्या, कम्प्यूटेशनल शक्ति या संचार को नहीं बदलता है। प्रत्येक MoE परत 512 रूट किए गए विशेषज्ञों में से 10 और एक सामान्य विशेषज्ञ का चयन करती है, जो संपीड़ित लेटेंट स्पेस में काम करती है, जिससे विशेषज्ञ संचार और गणना लगभग आधा हो जाता है। ECHO नामक एक घटक भविष्य के टोकन की भविष्यवाणी करने के लिए मॉडल की अंतिम परत का पुन: उपयोग करता है, बजाय इसके कि एक अलग मल्टी-टोकन भविष्यवाणी मॉड्यूल जोड़ा जाए, और अनुमानित डिकोडिंग के लिए एक ड्राफ्ट मॉडल के रूप में कार्य करता है।
प्री-ट्रेनिंग के लिए हजारों एक्सेलेरेटर पर मिश्रित परिशुद्धता FP8 के साथ Muon ऑप्टिमाइज़र और नए FlashMask-4 अटेंशन कोर का उपयोग किया गया था। Baidu का दावा है कि यह कोर बड़े हेड आकारों पर SM100 क्लास के GPU पर FlashAttention-4 की तुलना में फॉरवर्ड पास में 175% तक तेज चलता है। डेटा मिश्रण में रिपॉजिटरी स्तर पर कोड पर जोर दिया गया था, जिसमें कमिट, मर्ज अनुरोध, कोड समीक्षा और CI द्वारा सत्यापित फिक्स शामिल थे, साथ ही लंबे दस्तावेज़ भी शामिल थे।
Baidu ने एक एकीकृत मूल्यांकन प्रणाली का उपयोग करके 30 बेंचमार्क पर मॉडल का परीक्षण किया, इसकी तुलना DeepSeek-V4-Flash-Base, Hy3-preview-Base और MiMo-V2.5-Base से की। घोषित परिणामों में MMLU-Pro पर DeepSeek-V4-Flash-Base के 69.2 की तुलना में 78.4 अंक और HumanEval+ पर 69.5 की तुलना में 82.3 अंक शामिल हैं। हालांकि, तुलना किए गए मॉडल कुछ परीक्षणों जैसे ChineseSimpleQA, चेन ऑफ थॉट प्रॉम्प्टिंग के साथ BBH और InfiniteBench में आगे थे।



