← नवीनतम पेपर
🤖 machine learning

Sticky Routing: Training MoE Models for Memory-Efficient Inference

यह शोधपत्र StickyMoE का प्रस्ताव करता है, जो एक डिफरेंशियल योग्य (differentiable) रूटिंग कंसिस्टेंसी लॉस है जो Mixture-of-Experts मॉडल्स को अर्थपूर्ण रूप से सुसंगत टोकन स्पैन (token spans) में विशेषज्ञ असाइनमेंट बनाए रखने के लिए प्रशिक्षित करता है, जिससे एज डिवाइसेस पर मेमोरी-गहन वेट स्वैपिंग (weight swapping) को न्यूनतम परप्लेक्सिटी डिग्रेडेशन के साथ काफी कम किया जा सकता है।

मूल लेखक: Ali Kayyam

प्रकाशित 2026-07-13
📖 7 मिनट में पढ़ें🧠 गहराई से पढ़ें

मूल लेखक: Ali Kayyam

मूल पेपर CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/) के तहत लाइसेंस किया गया है। नीचे दिए गए पेपर की यह व्याख्या AI से तैयार की गई है। इसे लेखकों ने न तो लिखा है, न इसका समर्थन किया है। तकनीकी सटीकता के लिए मूल पेपर देखें। पूरा डिस्क्लेमर पढ़ें

कल्पना कीजिए कि आपके पास 64 अलग-अलग विशेषज्ञ पुस्तकालयाध्यक्षों (librarians) का एक विशाल पुस्तकालय है, लेकिन आपके छोटे से एज डिवाइस (जैसे कि फोन या स्मार्ट स्पीकर) के पास केवल इतनी शेल्फ स्पेस है कि वह एक समय में डेस्क पर केवल 2 विशेषज्ञों को रख सके। बाकी सभी बेसमेंट में रखे गए हैं।

एक मानक "मिक्सचर-ऑफ-एक्सपर्ट्स" (MoE) मॉडल में, हर बार जब AI एक नया शब्द पढ़ता है, तो वह यह तय करने के लिए सिक्का उछालता है कि किस पुस्तकालयाध्यक्ष से पूछना है। समस्या यह है कि वह हर एक शब्द के लिए सिक्का उछालता है। इसलिए, वह "एप्पल" के लिए पुस्तकालयाध्यक्ष A से पूछता है, फिर "पाई" के लिए पुस्तकालध्यक्ष B से, फिर "क्रस्ट" के लिए पुस्तकालध्यक्ष C से। हर बार जब वह स्विच करता है, तो उसे बेसमेंट में दौड़ना पड़ता है, पुस्तकालध्यक्ष B को बाहर निकालना पड़ता है, और पुस्तकालध्यक्ष C को घसीटकर ऊपर लाना पड़ता है। यह "दौड़-भाग" इतनी धीमी है कि AI कहानी पढ़ने के बजाय पुस्तकालयाध्यक्षों को लाने-ले जाने में अधिक समय बिता देता है।

यह शोध पत्र StickyMoE नामक एक नया तरीका पेश करता है, जिससे इन मॉडल्स को इस तरह प्रशिक्षित किया जाता है कि वे सिक्के उछालने की इस पागलपन भरी प्रक्रिया को रोक सकें। इसके बजाय, मॉडल "चिपचिपा" (sticky) होना सीख जाता है। यदि वह "एप्पल" के लिए पुस्तकालध्यक्ष A से पूछता है, तो उसे "पाई" और "क्रस्ट" के लिए भी पुस्तकालध्यक्ष A से ही पूछते रहने के लिए प्रोत्साहित किया जाता है, जब तक कि विषय तर्कसंगत हो।

बड़ी खोज: इसे सही तरीके से प्रशिक्षित करें, बाद में ठीक न करें

लेखकों ने इस "दौड़-भाग" की समस्या को ठीक करने के लिए कुछ चीजें आजमाईं, और उन्हें यह पता चला:

  1. "पोस्ट-हॉक" (बाद में किया गया) सुधार विफल रहा: उन्होंने एक मॉडल लिया जो पहले से ही प्रशिक्षित था और फिर उसके सिक्का उछालने वाले तंत्र (राउटर) को कम यादृच्छिक (random) बनाने के लिए उसे "फाइन-ट्यूनिंग" करने की कोशिश की। उन्होंने इसे ReMoE कहा। परिणाम? यह काम नहीं आया। स्विच दर (switch rate) में बहुत कम बदलाव आया (0.5% से भी कम पूर्ण परिवर्तन)। लेखक तर्क देते हैं कि ऐसा इसलिए हुआ क्योंकि पुस्तकालयाध्यक्ष (विशेषज्ञ) पहले से ही विशिष्ट, अलग-थलग शब्दों के लिए विशेषज्ञ के रूप में प्रशिक्षित थे। आप सिक्के उछालने वाले को "चिपचिपा" होने के लिए केवल तभी नहीं कह सकते यदि पुस्तकालयाध्यक्ष स्वयं लंबे, निरंतर विषयों को संभालने के लिए तैयार न हों। यह एक स्प्रिंटर (तेज धावक) को केवल उसकी शुरुआती मुद्रा बदलकर मैराथन धावक सिखाने जैसा है; मांसपेशियां (विशेषज्ञ भार/weights) छोटी दौड़ के लिए बनी हैं, लंबी दौड़ के लिए नहीं।

  2. "हार्ड" (कठोर) सुधार बहुत अनम्य था: उन्होंने एक और तरीका भी आजमाया जिसमें एक सख्त नियम (एक "हार्ड-विंडो") का उपयोग करके मॉडल को कुछ शब्दों के लिए एक ही पुस्तकालध्यक्ष पर टिके रहने के लिए मजबूर किया गया। हालांकि इससे स्विचिंग कम हुई, लेकिन इसने AI के उत्तरों को बहुत खराब कर दिया (मध्यम मॉडल पर "परप्लेक्सिटी" को 6.8% तक बढ़ा दिया)। यह एक पुस्तकालध्यक्ष को एक पूरा उपन्यास पढ़ने के लिए मजबूर करने जैसा था, भले ही विषय पूरी तरह से अलग शैली में बदल गया हो; कहानी की गुणवत्ता प्रभावित हुई।

  3. "स्टिकी" (चिपचिपा) समाधान काम कर गया: जीतने वाला तरीका, StickyMoE, प्रशिक्षण के सबसे पहले चरण में एक सौम्य "निरंतरता दंड" (consistency penalty) जोड़ता है। यह मॉडल को बताता है: "हे, यदि आप दो लगातार शब्दों के बीच विशेषज्ञों को बदलते हैं, तो यह थोड़ा दंडनीय है।" यह मॉडल को भाषा सीखते समय ही यह सीखने के लिए प्रोत्साहित करता है।

परिणाम: चिपचिपा होना बेहतर है

जब उन्होंने WikiText-2 डेटासेट (विकिपीडिया लेखों का एक संग्रह) का उपयोग करके छोटे और मध्यम आकार के मॉडल्स पर इसका परीक्षण किया, तो परिणाम स्पष्ट थे:

  • कम स्विचिंग: एक एकल नॉब (जिसे λ कहा जाता है) को ट्यून करके, वे विशेषज्ञों को बदलने की संख्या को 59% तक कम कर सके। उदाहरण के लिए, मध्यम मॉडल पर, स्विचिंग 0.71 (लग हर बार स्विच करना) से गिरकर 0.29 रह गई।
  • बेहतर गति (सिमुलेटेड): चूंकि मॉडल कम स्विच करता था, इसलिए उन्होंने एक कैश (डेस्क शेल्फ) का सिमुलेशन किया जो 2 विशेषज्ञों को रख सकता है। "कैश हिट रेट" (कितनी बार सही पुस्तकालध्यक्ष डेस्क पर पहले से मौजूद था) 0.54 से बढ़कर 0.88 हो गया। इसका मतलब है कि मॉडल को पहले की तुलना में 3.92 गुना कम बार बेसमेंट में दौड़ना पड़ेगा।
  • बेहतर गुणवत्ता: आश्चर्यजनक रूप से, मध्यम मॉडल पर, मध्यम मात्रा में "चिपचिपाहट" (λ = 0.05) का उपयोग करने से वास्तव में AI अधिक स्मार्ट हो गया, जिससे इसकी "परप्लेक्सिटी" (भ्रम का एक माप) 4.1% कम हो गई। मॉडल ने सीखा कि विषय पर टिके रहने से संदर्भ को समझने में मदद मिलती है।
  • कोई पतन (Collapse) नहीं: एक बड़ा डर था कि मॉडल "आलसी" हो जाएगा और हर चीज़ के लिए एक ही पुस्तकालध्यक्ष का उपयोग करेगा। लेकिन "यूटिलाइजेशन एंट्रॉपी" (Utilisation Entropy) उच्च रही (अधिकतम 2.0 में से 1.92 बिट्स से ऊपर), जिसका अर्थ है कि सभी 4 विशेषज्ञों का अभी भी निष्पक्ष रूप से उपयोग किया जा रहा था। मॉडल आलसी नहीं हुआ; वह बस व्यवस्थित हो गया।

क्या काम नहीं आया (और क्यों)

शोध पत्र स्पष्ट रूप से इस विचार को खारिज करता है कि आप मॉडल बनने के बाद इस समस्या को ठीक कर सकते हैं। ReMoE दृष्टिकोण (एक पुराने मॉडल को फाइन-ट्यून करना) अप्रभावी साबित हुआ क्योंकि "विशेषज्ञ प्रतिनिधित्व" (पुस्तकालयाध्यक्षों के मस्तिष्क) पहले से ही एक ऐसी प्रशिक्षण प्रक्रिया द्वारा आकार ले चुके थे जिसे विषयों पर टिके रहने की परवाह नहीं थी। आप बाद में स्थानीयता (locality) को जोड़ नहीं सकते; आपको इसे शुरुआत से ही शामिल करना होगा।

उन्होंने यह भी नोट किया कि मॉडल की पहली परत (लेयर 0) को "चिपचिपा" बनाना सबसे कठिन है। सर्वोत्तम सेटिंग्स के बावजूद, वहां स्विच दर 0.49 से ऊपर रही। ऐसा इसलिए है क्योंकि पहली परत कच्चे शब्दों को देखती है इससे पहले कि उन्हें संदर्भ के साथ मिलाया जाए, इसलिए "एप्पल" और "पाई" इस स्तर पर वास्तव में बहुत अलग दिख सकते हैं।

मुख्य निष्कर्ष

शोध पत्र सुझाव देता है कि यदि आप चाहते हैं कि आपका AI सीमित मेमोरी वाले डिवाइस पर तेजी से चले, तो आपको केवल कैशिंग नियमों के साथ चालाक होने या मॉडल को बाद में ठीक करने की कोशिश नहीं करनी चाहिए। इसके बजाय, आपको मॉडल को पहले दिन से ही "चिपचिपा" बनाने के लिए प्रशिक्षित करना चाहिए। दो आसन्न शब्दों के बीच विशेषज्ञों को बहुत अधिक बार बदलने के लिए एक साधारण दंड जोड़कर, आप एक ऐसा मॉडल प्राप्त करते हैं जो स्वाभाविक रूप से विषय पर टिका रहता है, उसे धीमे बेसमेंट में कम चक्कर लगाने पड़ते हैं, और वास्तव में कहानी को बेहतर ढंग से समझता है।

लेखक स्वीकार करते हैं कि यह सिमुलेशन और विशिष्ट डेटासेट्स (WikiText-2) के साथ छोटे और मध्यम मॉडल्स (अधिकतम 22M पैरामीटर्स) पर आधारित है। वे सुझाव देते हैं कि जैसे-जैसे मॉडल बड़े होंगे, यह "चिपचिपाहट" और भी बेहतर काम कर सकती है, लेकिन उन्होंने अभी तक बड़े बिलियन-पैरामीटर वाले मॉडल्स का परीक्षण नहीं किया है। वे यह भी नोट करते हैं कि वर्तमान में उनका तरीका यह नहीं जानता कि वाक्य कब समाप्त होता है, इसलिए यह पैराग्राफ ब्रेक के दौरान "बहुत अधिक चिपचिपा" हो सकता है, जो भविष्य के कार्य के लिए एक समस्या है।

अपने क्षेत्र के पेपरों की भीड़ में उलझे हुए हैं?

आपके रिसर्च कीवर्ड से मेल खाने वाले सबसे नए और अलग सोच वाले पेपरों का रोज़ाना Digest पाएँ—तकनीकी सारांश के साथ, आपकी भाषा में।

Digest आज़माएँ →