2025-11-19T15:49:13.925681

Myopic Bayesian Decision Theory for Batch Active Learning with Partial Batch Label Sampling

Hu, Mussmann

Over the past couple of decades, many active learning acquisition functions have been proposed, leaving practitioners with an unclear choice of which to use. Bayesian Decision Theory (BDT) offers a universal principle to guide decision-making. In this work, we derive BDT for (Bayesian) active learning in the myopic framework, where we imagine we only have one more point to label. This derivation leads to effective algorithms such as Expected Error Reduction (EER), Expected Predictive Information Gain (EPIG), and other algorithms that appear in the literature. Furthermore, we show that BAIT (active learning based on V-optimal experimental design) can be derived from BDT and asymptotic approximations. A key challenge of such methods is the difficult scaling to large batch sizes, leading to either computational challenges (BatchBALD) or dramatic performance drops (top-$B$ selection). Here, using a particular formulation of the decision process, we derive Partial Batch Label Sampling (ParBaLS) for the EPIG algorithm. We show experimentally for several datasets that ParBaLS EPIG gives superior performance for a fixed budget and Bayesian Logistic Regression on Neural Embeddings. Our code is available at https://github.com/ADDAPT-ML/ParBaLS.

academic

बैच सक्रिय शिक्षण के लिए आंशिक बैच लेबल नमूनाकरण के साथ मायोपिक बायेसियन निर्णय सिद्धांत

मूल जानकारी

पेपर ID: 2510.09877
शीर्षक: बैच सक्रिय शिक्षण के लिए आंशिक बैच लेबल नमूनाकरण के साथ मायोपिक बायेसियन निर्णय सिद्धांत
लेखक: कांगपिंग हू, स्टीफन मुसमैन (जॉर्जिया इंस्टीट्यूट ऑफ टेक्नोलॉजी)
वर्गीकरण: cs.LG cs.AI stat.ML
प्रकाशन समय: 25 अक्टूबर 10 (प्रीप्रिंट)
पेपर लिंक: https://arxiv.org/abs/2510.09877v1

सारांश

पिछले कई दशकों में, कई सक्रिय शिक्षण अधिग्रहण कार्य प्रस्तावित किए गए हैं, लेकिन व्यावहारिककर्ताओं को अक्सर उपयुक्त विधि चुनने में कठिनाई होती है। बायेसियन निर्णय सिद्धांत (BDT) निर्णय लेने के लिए एक सामान्य सिद्धांत प्रदान करता है। यह पेपर मायोपिक ढांचे के तहत (बायेसियन) सक्रिय शिक्षण के लिए BDT प्राप्त करता है, यह मानते हुए कि केवल एक अतिरिक्त डेटा बिंदु को लेबल करने की आवश्यकता है। यह व्युत्पत्ति प्रभावी एल्गोरिदम जैसे अपेक्षित त्रुटि में कमी (EER), अपेक्षित भविष्यसूचक सूचना लाभ (EPIG) आदि उत्पन्न करती है। इसके अलावा, लेखक साबित करते हैं कि BAIT को BDT और स्पर्शोन्मुख सन्निकटन के माध्यम से प्राप्त किया जा सकता है। इस प्रकार की विधियों की मुख्य चुनौती बड़े बैच आकार तक स्केल करना है, जिससे कम्प्यूटेशनल चुनौतियाँ (BatchBALD) या प्रदर्शन में तीव्र गिरावट (शीर्ष-B चयन) होती है। यह पेपर एक विशिष्ट निर्णय प्रक्रिया सूत्रीकरण के माध्यम से EPIG एल्गोरिदम के लिए आंशिक बैच लेबल नमूनाकरण (ParBaLS) विधि प्राप्त करता है। प्रयोग दर्शाते हैं कि निश्चित बजट और तंत्रिका एम्बेडिंग पर बायेसियन लॉजिस्टिक प्रतिगमन सेटिंग में, ParBaLS EPIG कई डेटासेट पर उत्कृष्ट प्रदर्शन करता है।

अनुसंधान पृष्ठभूमि और प्रेरणा

समस्या परिभाषा

सक्रिय शिक्षण का उद्देश्य बड़ी मात्रा में अनलेबल किए गए डेटा से सबसे सूचनात्मक डेटा का चयन करना है ताकि सीमित लेबलिंग बजट के तहत मॉडल प्रदर्शन को अधिकतम किया जा सके। मौजूदा विधियों में अनुमानी विधियाँ और संभाव्य विधियाँ शामिल हैं, लेकिन स्पष्ट चयन मार्गदर्शन सिद्धांतों की कमी है।

समस्या की महत्ता

व्यावहारिक आवश्यकता: आधुनिक मशीन लर्निंग में, डेटा आमतौर पर व्यक्तिगत रूप से नहीं बल्कि बैच के रूप में लेबल किया जाता है
विधि चयन कठिनाई: मौजूदा एल्गोरिदम में व्याख्यात्मकता की कमी है, व्यावहारिककर्ताओं को यह निर्धारित करना मुश्किल है कि कब कौन सी विधि प्रभावी है
स्केलेबिलिटी चुनौती: मौजूदा विधियाँ बड़े बैच आकार पर कम्प्यूटेशनल या प्रदर्शन समस्याओं का सामना करती हैं

मौजूदा विधियों की सीमाएँ

शीर्ष-B चयन: बैच लेबल के बीच निर्भरता को नजरअंदाज करता है, अनावश्यक नमूने चुन सकता है
अनुमानी विविधता: डेटासेट-विशिष्ट हाइपरपैरामीटर समायोजन की आवश्यकता है, जो सक्रिय शिक्षण में व्यावहारिक नहीं है
लालची बैच अधिग्रहण: BatchBALD जैसी विधियों की कम्प्यूटेशनल जटिलता बैच आकार के साथ घातीय रूप से बढ़ती है

अनुसंधान प्रेरणा

बायेसियन निर्णय सिद्धांत के माध्यम से एक एकीकृत सैद्धांतिक ढांचा प्रदान करना, मौजूदा एल्गोरिदम के कार्य सिद्धांत को समझाना, और बैच चयन को प्रभावी ढंग से संभालने में सक्षम नई विधि प्रस्तावित करना।

मुख्य योगदान

सैद्धांतिक एकीकरण: कई एल्गोरिदम (EER, EPIG, BAIT आदि) को मायोपिक बायेसियन निर्णय सिद्धांत (MBDT) की व्युत्पत्ति के परिणाम के रूप में एकीकृत करना
नई विधि प्रस्ताव: बैच सक्रिय शिक्षण की चुनौतियों को हल करने के लिए आंशिक बैच लेबल नमूनाकरण (ParBaLS) का परिचय
सैद्धांतिक विश्लेषण: ParBaLS के मोंटे कार्लो सन्निकटन त्रुटि को O(1/√m) साबित करना, जो बैच आकार पर निर्भर नहीं है
प्रायोगिक सत्यापन: 10 विभिन्न सेटिंग्स में ParBaLS EPIG की श्रेष्ठ प्रदर्शन को सत्यापित करना

विधि विवरण

कार्य परिभाषा

इनपुट डोमेन X, आउटपुट डोमेन Y और अनलेबल किए गए पूल डेटासेट D⊂X दिया गया है, लक्ष्य T बैच S⊂D को पुनरावृत्ति से चुनना है, प्रत्येक बैच आकार |S|=B लेबलिंग के लिए, ताकि लेबल किए गए सेट पर प्रशिक्षित परीक्षण हानि को कम किया जा सके।

मायोपिक बायेसियन निर्णय सिद्धांत (MBDT)

एकल-बिंदु चयन व्युत्पत्ति

मायोपिक ढांचे में, यह मानते हुए कि केवल एक अतिरिक्त डेटा बिंदु x̂ का चयन किया जाता है, अगला लेबल किया जाने वाला बिंदु है:

argmin_{x̂∈D} E_{ŷ~Y_{x̂}|L} [min_{P∈Δ^{|V|}_Y} E_{y⃗~Y_V|Y_{x̂}=ŷ,L} [∑_{j=1}^{|V|} ℓ(y_j, P_j)]]

नकारात्मक लॉग संभावना हानि के लिए, इष्टतम भविष्यसूचना पश्च वितरण है, अपेक्षित हानि एंट्रॉपी में सरल हो जाती है:

argmax_{x̂∈D} ∑_{x∈V} I(Y_x; Y_{x̂}|L)

यह EPIG और EER एल्गोरिदम के समतुल्य है।

बैच चयन चुनौती

मौजूदा बैच रणनीतियाँ तीन श्रेणियों में विभाजित हैं:

शीर्ष-B: सर्वोच्च स्कोर वाले B बिंदुओं का चयन, निर्भरता को नजरअंदाज करता है
अनुमानी विविधता: यादृच्छिकता या विविधता जोड़ता है, हाइपरपैरामीटर समायोजन की आवश्यकता है
लालची बैच अधिग्रहण: संपूर्ण बैच को अनुकूलित करता है, उच्च कम्प्यूटेशनल जटिलता

ParBaLS विधि

मूल विचार

पहले से प्रतिबद्ध लेकिन अनदेखे लेबल वाले आंशिक बैच S का परिचय दें, अगला इष्टतम बिंदु है:

argmax_{x̂∈D} E_{y_S~Y_S|L} [∑_{x∈V} I(Y_x; Y_{x̂}|Y_S = y_S, L)]

मोंटे कार्लो अनुमान

घातीय स्तर के योग को संभालने के लिए मोंटे कार्लो अनुमान का उपयोग करें:

argmax_{x̂∈D} (1/m) ∑_{i=1}^m ∑_{x∈V} I(Y_x; Y_{x̂}|Y_S = y_S^{(i)}, L)

एल्गोरिदम प्रवाह

ParBaLS एल्गोरिदम क्रमिक रूप से बैच का निर्माण करता है:

खाली बैच S=∅ को आरंभ करें
बायेसियन मॉडल M_L को प्रशिक्षित करें
m छद्म-लेबल संस्करण y^{(i)}~Y_D|L का नमूना लें
प्रत्येक बैच स्थिति के लिए:
- प्रत्येक उम्मीदवार बिंदु के लिए EPIG स्कोर की गणना करें
- सर्वोच्च स्कोर वाले बिंदु को बैच में जोड़ें
- छद्म-लेबल के साथ m समानांतर मॉडल को अपडेट करें
पूर्ण बैच लौटाएँ

BAIT की व्युत्पत्ति

अनौपचारिक स्पर्शोन्मुख सन्निकटन के माध्यम से, BAIT को भी MBDT सिद्धांत से प्राप्त किया जा सकता है:

Tr([∇²ℓ_{L∪S}(ŵ_L)]^{-1}∇²ℓ_D(ŵ_L))

प्रायोगिक सेटअप

डेटासेट

प्रयोग 6 प्रकार के डेटासेट को कवर करते हैं:

सारणीबद्ध डेटा: एयरलाइन यात्री संतुष्टि, क्रेडिट कार्ड धोखाधड़ी
मानक छवि डेटा: CIFAR-10, CIFAR-100
वास्तविक दुनिया की छवि डेटा: iWildCam, fMoW (WILDS बेंचमार्क से)
एक-से-कई छवि डेटा: बहु-वर्ग को असंतुलित द्विआधारी परिदृश्य में परिवर्तित करना
उप-समूह पारी छवि डेटा: तीन-वर्ग सेटिंग, केवल पहली दो कक्षाओं पर परीक्षण

मॉडल सेटअप

छवि डेटा: निश्चित एम्बेडिंग मॉडल का उपयोग करें (WILDS के लिए CLIP-ViT-B/32, CIFAR के लिए DINOv2-ViT-S/14)
सारणीबद्ध डेटा: सीधे बायेसियन लॉजिस्टिक प्रतिगमन लागू करें
बायेसियन सेटिंग: k=400 पश्च पैरामीटर नमूने, NUTS नमूनाकारी का उपयोग करें

मूल्यांकन मेट्रिक्स

परीक्षण सटीकता को मुख्य मूल्यांकन मेट्रिक के रूप में उपयोग करें

तुलनात्मक विधियाँ

बायेसियन विधियाँ: EPIG, BALD (शीर्ष-B या Gumbel शोर के साथ)
आधारभूत विधियाँ: Random, Confidence, BatchBALD
प्रस्तावित विधियाँ: ParBaLS-MAP EPIG, ParBaLS EPIG

प्रायोगिक पैरामीटर

T=10 पुनरावृत्तियाँ, प्रत्येक B=10 नमूने बजट
500 नमूनों का प्रारंभिक यादृच्छिक नमूनाकरण
भेदभाव बढ़ाने के लिए कुछ सेटिंग्स में B=20, प्रारंभिक 100 नमूने का उपयोग करें
प्रत्येक सेटिंग को 5 विभिन्न बीजों के साथ चलाएँ

प्रायोगिक परिणाम

मुख्य परिणाम

तालिका 1 के पूर्ण प्रायोगिक परिणामों के अनुसार, ParBaLS EPIG 10 सेटिंग्स में से 9 में सर्वश्रेष्ठ प्रदर्शन करता है:

एल्गोरिदम	उच्चतम माध्य	शीर्ष सूची में
ParBaLS EPIG	4	9
ParBaLS-MAP EPIG	2	7
SoftRankEPIG	0	4
EPIG	0	4
Confidence	3	5

विशिष्ट प्रदर्शन

सारणीबद्ध डेटासेट (सबसे उल्लेखनीय प्रदर्शन):

एयरलाइन यात्री संतुष्टि: ParBaLS EPIG 89.42±0.41% तक पहुँचता है
क्रेडिट कार्ड धोखाधड़ी: ParBaLS EPIG 93.55±0.23% तक पहुँचता है

उप-समूह पारी सेटिंग (सबसे चुनौतीपूर्ण):

fMoW: ParBaLS EPIG 31.37±6.60% तक पहुँचता है, अन्य विधियों से काफी बेहतर
iWildCam: ParBaLS EPIG 84.72±1.98% तक पहुँचता है

शिक्षण वक्र विश्लेषण

चित्र 2 सारणीबद्ध डेटासेट पर दिखाता है कि ParBaLS विधि पूरी शिक्षण प्रक्रिया में लगातार लाभ बनाए रखती है, विशेष रूप से कम बजट सेटिंग्स में अधिक उल्लेखनीय प्रदर्शन करती है।

विलोपन प्रयोग

ParBaLS बनाम ParBaLS-MAP: पूर्ण ParBaLS आमतौर पर केवल MAP लेबल का उपयोग करने वाले संस्करण से बेहतर है
बैच आकार प्रभाव: ParBaLS का लाभ बड़े बैच (B=20) पर अधिक स्पष्ट है
एकल-बिंदु बनाम बैच: परिशिष्ट प्रयोग दिखाते हैं कि हालांकि एकल-बिंदु चयन (B=1) बेहतर प्रदर्शन करता है, बैच चयन व्यावहारिक अनुप्रयोगों में अधिक कुशल है

निष्कर्ष और चर्चा

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

सैद्धांतिक एकीकरण: MBDT कई सक्रिय शिक्षण एल्गोरिदम के लिए एक एकीकृत सैद्धांतिक आधार प्रदान करता है
बैच समाधान: ParBaLS बैच सक्रिय शिक्षण की स्केलेबिलिटी समस्या को प्रभावी ढंग से हल करता है
प्रायोगिक सत्यापन: ParBaLS EPIG विभिन्न सेटिंग्स में उत्कृष्ट प्रदर्शन करता है, विशेष रूप से उच्च अनिश्चितता वाले परिदृश्यों के लिए उपयुक्त है

सीमाएँ

कम्प्यूटेशनल जटिलता: ParBaLS की समय जटिलता O(TBm) है, m समानांतर मॉडल कम्प्यूटेशनल बोझ बढ़ाते हैं
विधि प्रयोज्यता: मुख्य रूप से बायेसियन लॉजिस्टिक प्रतिगमन पर सत्यापित, गहन नेटवर्क के लिए विस्तार को आगे के अनुसंधान की आवश्यकता है
सैद्धांतिक विश्लेषण: BAIT की व्युत्पत्ति अनौपचारिक स्पर्शोन्मुख सन्निकटन पर निर्भर करती है, सैद्धांतिक कठोरता में सुधार की आवश्यकता है

भविष्य की दिशाएँ

कम्प्यूटेशनल दक्षता: कम्प्यूटेशनल रूप से कुशल सन्निकटन विधि खोजें, बड़े डेटासेट और मॉडल तक विस्तार करें
गहन शिक्षण एकीकरण: ParBaLS को पूर्ण गहन तंत्रिका नेटवर्क प्रशिक्षण तक विस्तारित करने का अनुसंधान करें
सैद्धांतिक सुधार: अधिक कठोर सैद्धांतिक विश्लेषण और अभिसरण गारंटी प्रदान करें

गहन मूल्यांकन

शक्तियाँ

सैद्धांतिक योगदान: सक्रिय शिक्षण एल्गोरिदम के लिए एक एकीकृत सैद्धांतिक ढांचा प्रदान करता है, व्याख्यात्मकता को बढ़ाता है
व्यावहारिक मूल्य: ParBaLS व्यावहारिक अनुप्रयोगों में बैच चयन समस्या को हल करता है
पर्याप्त प्रयोग: कई डेटा प्रकार और चुनौतीपूर्ण सेटिंग्स को कवर करता है, परिणाम विश्वसनीय हैं
विधि नवीनता: बैच निर्माण में छद्म-लेबल का अनुप्रयोग नवीन है

कमियाँ

कम्प्यूटेशनल ओवरहेड: m समानांतर मॉडल का रखरखाव कम्प्यूटेशनल लागत बढ़ाता है
सैद्धांतिक कठोरता: कुछ व्युत्पत्तियाँ (जैसे BAIT) अनौपचारिक सन्निकटन पर निर्भर करती हैं
प्रायोगिक सीमाएँ: मुख्य रूप से अपेक्षाकृत सरल मॉडल (लॉजिस्टिक प्रतिगमन) पर सत्यापित
हाइपरपैरामीटर संवेदनशीलता: m के चयन के लिए प्रदर्शन और कम्प्यूटेशन के बीच ट्रेड-ऑफ विश्लेषण पर्याप्त नहीं है

प्रभाव

सैद्धांतिक प्रभाव: सक्रिय शिक्षण के लिए एक नया सैद्धांतिक दृष्टिकोण प्रदान करता है, बाद के अनुसंधान को प्रेरित कर सकता है
व्यावहारिक मूल्य: ParBaLS विधि का प्रत्यक्ष अनुप्रयोग मूल्य है, विशेष रूप से बैच लेबलिंग परिदृश्यों में
पुनरुत्पादनीयता: खुला स्रोत कोड प्रदान करता है, पुनरुत्पादन और विस्तार को सुविधाजनक बनाता है

प्रयोज्य परिदृश्य

उच्च अनिश्चितता कार्य: सारणीबद्ध डेटा और उप-समूह पारी जैसे अपरिहार्य अनिश्चितता वाले परिदृश्य
बैच लेबलिंग आवश्यकता: ऐसे व्यावहारिक अनुप्रयोग जहाँ व्यक्तिगत लेबलिंग के बजाय बल्क लेबलिंग की आवश्यकता है
बायेसियन सेटिंग: ऐसे मॉडल और कार्य जहाँ बायेसियन अनुमान किया जा सकता है

संदर्भ

यह पेपर सक्रिय शिक्षण क्षेत्र के महत्वपूर्ण साहित्य का हवाला देता है, जिसमें शामिल हैं:

शास्त्रीय अनिश्चितता नमूनाकरण विधियाँ (Lewis, 1995)
बायेसियन सक्रिय शिक्षण विधियाँ (Houlsby et al., 2011; Gal et al., 2017)
बैच सक्रिय शिक्षण विधियाँ (Kirsch et al., 2019, 2023)
अपेक्षित त्रुटि में कमी विधियाँ (Roy and McCallum, 2001; Mussmann et al., 2022)

समग्र मूल्यांकन: यह सक्रिय शिक्षण क्षेत्र में महत्वपूर्ण सैद्धांतिक और व्यावहारिक मूल्य वाला एक पेपर है। MBDT के माध्यम से मौजूदा एल्गोरिदम को एकीकृत करके और बैच चयन समस्या को हल करने के लिए ParBaLS प्रस्तावित करके, यह क्षेत्र के लिए नई अनुसंधान दिशा प्रदान करता है। हालांकि कम्प्यूटेशनल दक्षता और सैद्धांतिक कठोरता के संदर्भ में सुधार की गुंजाइश है, लेकिन इसका योगदान महत्वपूर्ण है।