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

BACH: A Bayesian Admixture of Contrastive Heads for Multi-Interest Two-Tower Retrieval

BACH, जो मल्टी-इंटरेस्ट टू-टावर रिट्रीवल के लिए एक 'बेयसियन एडमिक्चर ऑफ कॉन्ट्रास्टिव हेड्स' फ्रेमवर्क पेश करता है, वेरिएशनल इन्फरेंस का उपयोग करके उपयोगकर्ता हितों को एक सॉफ्ट मिक्सचर के रूप में मॉडल करता है, जो प्रभावी रूप से रूटिंग कोलैप्स को कम करता है और प्रति-उपयोगकर्ता हित भार (per-user interest weights) प्रदान करते हुए बड़े पैमाने के बेंचमार्क पर रिट्रीवल प्रदर्शन में सुधार करता है।

मूल लेखक: Quoc Phong Nguyen, Paul Albert, Long Vuong, Vuong Le, Julien Monteil

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

मूल लेखक: Quoc Phong Nguyen, Paul Albert, Long Vuong, Vuong Le, Julien Monteil

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

कल्पना कीजिए कि आप लाखों किताबों वाली एक विशाल, अनंत लाइब्रेरी में घूम रहे हैं। आपकी एक बहुत ही विशिष्ट पसंद है: आपको 1980 के दशक की साइंस-फिक्शन (sci-fi) पसंद है, लेकिन आप गुप्त रूप से बेकिंग ट्यूटोरियल और प्राचीन इतिहास को भी बहुत पसंद करते हैं।

पुराने ज़माने के रिकमेंडेशन सिस्टम एक ऐसे लाइब्रेरियन की तरह हैं जो आपकी सभी रुचियों को एक ही व्यक्तित्व (single personality) में दबाने की कोशिश करते हैं। वे अपने दिमाग में आपके लिए एक ही "आप" खोजने की कोशिश करते हैं। समस्या यह है कि अगर आपको साइंस-फिक्शन सबसे ज्यादा पसंद है, तो लाइब्रेरियन मान लेता है कि आप केवल साइंस-फिक्शन ही चाहते हैं। वे बेकिंग और इतिहास की किताबों को अलमारी के पीछे धकेल देते हैं क्योंकि वे उस एकल "साइंस-फिक्शन" लेबल में फिट नहीं बैठतीं। यह वह "सिंगल-वेक्टर" दृष्टिकोण है जिसे पेपर में बताया गया है, और यह आपकी कई रुचियों को शेल्फ पर ही छोड़ देता है।

फिर, कुछ स्मार्ट लाइब्रेरियनों ने एक नया तरीका आजमाया: मल्टी-इंटरेस्ट (Multi-Interest)। एक व्यक्तित्व के बजाय, उन्होंने आपके दिमाग में kk अलग-अलग "हेड्स" (या अवतार) दिए। एक हेड साइंस-फिक्शन प्रेमी है, एक बेकर है, और एक इतिहासकार है। जब आप एक किताब मांगते हैं, तो लाइब्रेरियन सभी kk हेड्स की जांच करता है और उस हेड को चुनता है जो सबसे बेहतर मेल खाता है।

लेकिन यहाँ वह गड़बड़ी है जो पेपर ने खोजी:
इन स्मार्ट लाइब्रेरियनों को प्रशिक्षित करने का पुराना तरीका एक "विजेता-ही-सब-कुछ" (winner-take-all) खेल की तरह था। जब आपने साइंस-फिक्शन बुक चुनी, तो लाइब्रेरियन ने केवल साइंस-फिक्शन हेड को ही 'हाई-फाइव' दिया। बेकर और इतिहासकार हेड्स को अनदेखा कर दिया गया। समय के साथ, साइंस-फिक्शन हेड बहुत मजबूत हो गया, जबकि अन्य हेड आलसी हो गए, उन्होंने सीखना बंद कर दिया, और अंततः वे साइंस-फिक्शन हेड की नकल करने लगे। इसे "रूटिंग कोलैप्स" (routing collapse) कहा जाता है। लाइब्रेरियन के पास kk हेड्स तो होते हैं, लेकिन वे सभी एक ही व्यक्ति की तरह व्यवहार करते हैं। इसके अलावा, लाइब्रेरियन को यह पता ही नहीं था कि आप बेकिंग या इतिहास के प्रति वास्तव में कितना ध्यान देते हैं; उन्होंने बस अंदाजा लगाया।

समाधान: BACH (Bayesian Admixture of Contrastive Heads)

लेखकों ने, जो अमेज़न की एक टीम है, BACH नामक एक नया सिस्टम बनाया है। BACH को एक ऐसे लाइब्रेरियन के रूप में सोचें जो न केवल आपसे बात करने के लिए एक हेड चुनता है, बल्कि आपकी रुचियों का एक कस्टम कॉकटेल (custom cocktail) बनाता है।

BACH कैसे काम करता है, यहाँ इसके वास्तविक मैकेनिक्स दिए गए हैं:

1. सॉफ्ट मिक्सचर (अब कोई "विजेता-ही-सब-कुछ" नहीं)
सिर्फ एक हेड को काम करने के लिए चुनने के बजाय, BACH पूछता है: "साइंस-फिक्शन हेड ने कितनी भूमिका निभाई? बेकर हेड ने कितनी भूमिका निभाई?"

  • पुराना तरीका: "साइंस-फिक्शन हेड जीत गया! बाकी सबको जीरो पॉइंट्स मिलते हैं।"
  • BACH का तरीका: "साइंस-फिक्शन हेड को 70% क्रेडिट मिलता है, बेकर को 20%, और इतिहासकार को 10%।"
    इसे सॉफ्ट मिक्सचर (soft mixture) कहा जाता है। क्योंकि हर हेड को आपके किसी भी इंटरैक्शन के साथ थोड़ा सा क्रेडिट (एक "ग्रेडिएंट") मिलता है, इसलिए कोई भी हेड कभी आलसी नहीं होता या ढलता नहीं है। वे सभी सीखते रहते हैं और तेज बने रहते हैं।

2. पर्सनलाइज्ड वेट (The "Admixture")
BACH आपकी रुचियों का केवल अंदाजा नहीं लगाता; यह आपके लिए एक विशिष्ट वेट (weight) (πu\pi_u) की गणना करता है।

  • आपके लिए, वेट हो सकता है: 70% साइंस-फिक्शन, 20% बेकिंग, 10% इतिहास।
  • आपके दोस्त के लिए, वेट हो सकता है: 10% साइंस-फिक्शन, 80% बेकिंग, 10% इतिहास।
    पेपर में दिखाया गया है कि ये वेट्स वैरिएशनल इन्फरेंस (variational inference) नामक विधि का उपयोग करके सीखे जाते हैं। यह ऐसा है जैसे लाइब्रेरियन प्रत्येक उपयोगकर्ता के लिए एक विशेष डायल को एडजस्ट कर रहा है ताकि यह पता चल सके कि प्रत्येक रुचि को कितना द्रव्यमान (mass) देना है। इस वेट का उपयोग प्रशिक्षण के दौरान और वास्तव में सिफारिशें देते समय, दोनों के लिए किया जाता है, ताकि सिस्टम सुसंगत बना रहे।

3. "ग्लोबल कोडबुक" विकल्प
पेपर में एक और कूल ट्रिक भी मिली। आप "साइंस-फिक्शन" और "बेकिंग" हेड्स को सभी के लिए साझा (shared) रख सकते हैं (विषयों की एक वैश्विक सूची), जबकि केवल वेट्स प्रत्येक व्यक्ति के लिए बदलते हैं।

  • यह क्यों शानदार है: लाइब्रेरियन एक बार "साइंस-फिक्शन बुक लिस्ट" और "बेकिंग बुक लिस्ट" की पूर्व-गणना कर सकता है और उन्हें स्टोर कर सकता है। जब आप आते हैं, तो वे बस आपके व्यक्तिगत वेट्स के आधार पर उन पूर्व-निर्मित सूचियों को मिला देते हैं। यह बहुत तेज़ है और नए उपयोगकर्ताओं (कोल्ड स्टार्ट समस्या) के लिए भी बहुत अच्छा काम करता है।

इस पेपर ने वास्तव में क्या साबित किया (और क्या नहीं)

लेखकों ने केवल अनुमान नहीं लगाया; उन्होंने तीन बड़े वास्तविक दुनिया के डेटासेट्स पर परीक्षण किया: MovieLens-20M (2 करोड़ मूवी रेटिंग), Taobao (एक विशाल चीनी शॉपिंग साइट), और Netflix

  • परिणाम: तीनों डेटासेट्स पर, BACH ने पुराने "विजेता-ही-सब-कुछ" वाले मल्टी-इंटरेस्ट मॉडल्स और सिंगल-पर्सनैलिटी मॉडल्स को पछाड़ दिया।
    • MovieLens-20M पर, BACH का सबसे अच्छा संस्करण (32 हेड्स के साथ) 0.069 का AUPRC तक पहुँचा, जिसने अगले सबसे अच्छे मॉडल (0.067) को पीछे छोड़ दिया।
    • Taobao पर, BACH ने पिछले सर्वश्रेष्ठ तरीके की तुलना में शीर्ष-सूची के परिणामों में लगभग 3% से 5% का सुधार किया।
    • Netflix पर, सुधार और भी अधिक स्पष्ट था, जहाँ BACH ने 32 हेड्स के साथ AUPRC में प्रतिस्पर्धा को 12.7% से पछाड़ दिया।
  • "रूटिंग" की खोज: पेपर स्पष्ट रूप से पुराने प्रशिक्षण तरीके को खारिज करता है जहाँ सिस्टम केवल "जीतने वाले" हेड को अपडेट करता है (जिसे "pos-multihead" कहा जाता है)। उन्होंने पाया कि "केवल विजेता" नियम के साथ प्रशिक्षण करना, जहाँ हर कैंडिडेट को उसके सबसे अच्छे हेड द्वारा स्कोर किया जाता है, उससे भी बदतर है। वास्तव में, पेपर ने दिखाया कि पुराने "विजेता-ही-सब-कुछ" प्रशिक्षण ने शीर्ष रैंकिंग में (41% तक खराब) भारी गिरावट पैदा की।
  • "कंसंट्रेशन" का आश्चर्य: पेपर ने तर्क दिया कि सिस्टम को स्थिर रखने के लिए जटिल "प्रायर्स" (पहले से निर्धारित नियम) की आवश्यकता नहीं है। उन्होंने पाया कि सिस्टम स्वयं-नियमित (self-regulates) होता है। "कंसंट्रेशन" (एक रुचि कितनी तीखी या धुंधली है) स्वाभाविक रूप से एक स्वस्थ सीमा (उनके विशिष्ट गणित के लिए लगभग 17 से 30) के भीतर रहती है, बिना किसी अतिरिक्त नियमों के।

निचोड़ (The Bottom Line)

पेपर सुझाव देता है कि उपयोगकर्ता की रुचियों को एक एकल विकल्प या कठोर विजेताओं के सेट के बजाय एक लचीले, भारित मिश्रण (flexible, weighted mixture) के रूप में मानकर, हम अधिक सटीक, "कोलैप्स" के प्रति कम संवेदनशील और तेजी से चलने वाले रिकमेंडेशन सिस्टम बना सकते हैं।

उन्होंने 32 हेड्स (रुचियों) के साथ परीक्षण किया और पाया कि यह सबसे अच्छा काम करता है, लेकिन उन्होंने यह भी दिखाया कि यह 8, 16, 128, और 256 हेड्स के साथ भी काम करता है। सिस्टम मजबूत है, और उनके द्वारा आजमाए गए दो अलग-अलग गणित संस्करणों (जिन्हें p-BACH और v-BACH कहा जाता है) ने लगभग समान प्रदर्शन किया, जिससे पता चलता है कि मिश्रण का विचार महत्वपूर्ण है, न कि विशिष्ट गणितीय स्वाद।

इसलिए, यदि आप लाखों अलग-अलग शौक रखने वाले एक जिज्ञासु किशोर हैं, तो BACH वह लाइब्रेरियन है जो अंततः आपको एक बॉक्स में फिट करने की कोशिश करना बंद कर देता है और इसके बजाय आपके लिए एक कस्टम, परफेक्ट शेल्फ बनाता है।

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

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

Digest आज़माएँ →