प्रीट्रेनिंग बड़े डेटासेट से मॉडल की बेसिक जानकारी बनाती है, जबकि पोस्ट-ट्रेनिंग ऑप्टिमाइज़ेशन खास कामों और इंसानी तालमेल के लिए उस बेस को बेहतर बनाता है। मॉडर्न AI डेवलपमेंट में दोनों स्टेज ज़रूरी हैं, जो एक-दूसरे की मदद करते हैं, न कि मुकाबला करते हैं।
मुख्य बातें
प्रीट्रेनिंग, रॉ डेटा के ट्रिलियन टोकन का इस्तेमाल करके बेसिक नॉलेज बनाता है।
पोस्ट-ट्रेनिंग ऑप्टिमाइज़ेशन, RLHF और DPO जैसी तकनीकों के ज़रिए मॉडल्स को उपयोगी, सुरक्षित और टास्क-स्पेसिफिक बनाता है।
प्रीट्रेनिंग में पोस्ट-ट्रेनिंग के मुकाबले कई गुना ज़्यादा कंप्यूट कॉस्ट आती है।
मॉडर्न AI डेवलपमेंट में पोस्ट-ट्रेनिंग वह जगह है जहाँ सबसे ज़्यादा प्रैक्टिकल कस्टमाइज़ेशन और अलाइनमेंट होता है।
पूर्व प्रशिक्षण क्या है?
शुरुआती ट्रेनिंग फ़ेज़, जिसमें मॉडल बहुत सारे रॉ टेक्स्ट या डेटा से आम पैटर्न सीखता है।
प्रीट्रेनिंग में आम तौर पर वेब, किताबों और कोड रिपॉजिटरी से निकाले गए बिना लेबल वाले डेटा के ट्रिलियन टोकन लगते हैं।
यह नेक्स्ट-टोकन प्रेडिक्शन जैसे सेल्फ-सुपरवाइज्ड लर्निंग ऑब्जेक्टिव का इस्तेमाल करता है, जहाँ मॉडल सीक्वेंस में अगले शब्द का अंदाज़ा लगाकर सीखता है।
यह फेज़ मॉडल डेवलपमेंट का सबसे कम्प्यूटेशनली महंगा हिस्सा है, जिसमें अक्सर GPU टाइम में लाखों डॉलर खर्च होते हैं।
GPT-3, LLaMA, और क्लाउड जैसे मॉडल सैकड़ों अरबों पैरामीटर पर बड़े पैमाने पर प्रीट्रेनिंग के साथ शुरू हुए।
इसके नतीजे में जो बेस मॉडल बनता है, उसमें भाषा की बड़ी समझ तो होती है, लेकिन उसमें काम के हिसाब से स्किल या सेफ्टी अलाइनमेंट की कमी होती है।
प्रशिक्षण के बाद अनुकूलन क्या है?
प्रीट्रेनिंग के बाद इस्तेमाल की जाने वाली तकनीकें, मॉडल को स्पेशलाइज़ करने, अलाइनमेंट को बेहतर बनाने और टास्क परफॉर्मेंस को बढ़ाने के लिए।
पोस्ट-ट्रेनिंग में सुपरवाइज्ड फाइन-ट्यूनिंग (SFT), ह्यूमन फीडबैक से रीइन्फोर्समेंट लर्निंग (RLHF), और डायरेक्ट प्रेफरेंस ऑप्टिमाइजेशन (DPO) शामिल हैं।
RLHF को OpenAI ने InstructGPT और बाद में GPT-4 में पॉपुलर बनाया ताकि मॉडल्स ज़्यादा मददगार और सुरक्षित बन सकें।
DPO, RLHF के एक आसान विकल्प के तौर पर सामने आया, जिसने प्रेफरेंस को सीधे ऑप्टिमाइज़ करके एक अलग रिवॉर्ड मॉडल की ज़रूरत को खत्म कर दिया।
इस स्टेज में आमतौर पर प्रीट्रेनिंग की तुलना में बहुत कम कंप्यूट की ज़रूरत होती है, और अक्सर ट्रिलियन के बजाय हज़ारों उदाहरणों का इस्तेमाल होता है।
पोस्ट-ट्रेनिंग में कॉन्स्टिट्यूशनल AI, टूल इस्तेमाल की ट्रेनिंग, और रीज़निंग पर फोकस्ड फ़ाइन-ट्यूनिंग जैसी टेक्नीक भी शामिल हो सकती हैं।
प्रीट्रेनिंग एक फाउंडेशन बनाने का स्टेज है, जहाँ एक मॉडल बहुत सारे रॉ डेटा से जनरल नॉलेज लेता है। इसके बिना, मॉडल को भाषा, रीज़निंग या दुनिया के फैक्ट्स की कोई अंदरूनी समझ नहीं होगी। पोस्ट-ट्रेनिंग ऑप्टिमाइज़ेशन उस फाउंडेशन को लेता है और उसे कुछ काम का बनाता है, मॉडल को इंस्ट्रक्शन फॉलो करना, नुकसान पहुँचाने वाले रिक्वेस्ट को मना करना और खास कामों में बेहतर करना सिखाता है। प्रीट्रेनिंग को जनरल एजुकेशन पाने जैसा समझें और पोस्ट-ट्रेनिंग को उसके बाद मिलने वाली स्पेशलाइज़्ड जॉब ट्रेनिंग जैसा।
डेटा और कंप्यूट आवश्यकताएँ
इन दोनों स्टेज के बीच स्केल का अंतर बहुत ज़्यादा है। प्रीट्रेनिंग के लिए बहुत बड़े डेटासेट की ज़रूरत होती है, अक्सर ट्रिलियन टोकन, और यह हफ़्तों या महीनों तक हज़ारों GPU पर चलता है। पोस्ट-ट्रेनिंग बहुत छोटे स्केल पर काम करती है, जिसमें आम तौर पर हज़ारों से लाखों उदाहरणों के क्यूरेटेड डेटासेट का इस्तेमाल होता है। इससे पोस्ट-ट्रेनिंग छोटी टीमों और रिसर्चर्स के लिए कहीं ज़्यादा आसान हो जाती है जो मौजूदा मॉडल्स को शुरू से बनाए बिना उन्हें कस्टमाइज़ करना चाहते हैं।
तकनीकें और विधियाँ
प्रीट्रेनिंग सेल्फ-सुपरवाइज्ड ऑब्जेक्टिव पर निर्भर करती है, जहाँ मॉडल असल में सीक्वेंस में मिसिंग या अगले टोकन का अनुमान लगाकर खुद को सिखाता है। पोस्ट-ट्रेनिंग ऑप्टिमाइज़ेशन में कई तरह के टूलकिट शामिल हैं, जिसमें इंस्ट्रक्शन-रिस्पॉन्स पेयर पर सुपरवाइज्ड फाइन-ट्यूनिंग, RLHF जो रिवॉर्ड मॉडल को ट्रेन करने के लिए ह्यूमन प्रेफरेंस रैंकिंग का इस्तेमाल करता है, और DPO जैसे नए तरीके शामिल हैं जो अलाइनमेंट प्रोसेस को आसान बनाते हैं। हर पोस्ट-ट्रेनिंग तकनीक बेसिक हेल्पफुलनेस से लेकर कॉम्प्लेक्स रीज़निंग कैपेबिलिटी तक, अलग-अलग लक्ष्यों को पूरा करती है।
मॉडल व्यवहार पर प्रभाव
एक प्रीट्रेन्ड मॉडल अपने आप में असल में एक एडवांस्ड ऑटोकम्प्लीट होता है, यह सही टेक्स्ट तो बना सकता है लेकिन भरोसेमंद तरीके से इंस्ट्रक्शन फॉलो नहीं करेगा या सुरक्षित तरीके से काम नहीं करेगा। पोस्ट-ट्रेनिंग वह चीज़ है जो एक रॉ लैंग्वेज मॉडल को एक चैटबॉट असिस्टेंट में बदल देती है जिसके साथ आप असल में इंटरैक्ट करना चाहते हैं। पोस्ट-ट्रेनिंग में किया गया अलाइनमेंट वर्क यह तय करता है कि मॉडल मददगार, नुकसान न पहुंचाने वाला, ईमानदार और बारीक बातचीत करने में सक्षम है या नहीं।
लचीलापन और पुनरावृत्ति
पोस्ट-ट्रेनिंग में ज़्यादा फ्लेक्सिबिलिटी होती है क्योंकि इसे बिना शुरू किए दोहराया, मिलाया और एडजस्ट किया जा सकता है। टीमें मेडिकल एप्लिकेशन के लिए एक मॉडल को फाइन-ट्यून कर सकती हैं, फिर किसी खास हॉस्पिटल की ज़रूरतों के लिए और ऑप्टिमाइज़ेशन लागू कर सकती हैं। प्री-ट्रेनिंग, एक बार पूरी होने पर, एक फिक्स्ड बेस बनाती है जिस पर हर कोई काम करता है। यही वजह है कि AI कम्युनिटी ने पोस्ट-ट्रेनिंग रिसर्च पर फोकस किया है, यहीं पर कस्टमाइज़ेशन और डिफरेंशिएशन सबसे तेज़ी से होता है।
लाभ और हानि
पूर्व प्रशिक्षण
लाभ
+व्यापक ज्ञान आधार बनाता है
+स्थानांतरण अधिगम को सक्षम बनाता है
+बहुमुखी नींव बनाता है
+विश्व ज्ञान को कैप्चर करता है
सहमत
−बहुत महंगा
−बड़े डेटासेट की ज़रूरत है
−लंबे प्रशिक्षण समय
−कार्य-विशिष्ट नहीं
प्रशिक्षण के बाद अनुकूलन
लाभ
+बहुत कम कंप्यूट लागत
+अत्यधिक अनुकूलन योग्य
+सुरक्षा और अलाइनमेंट में सुधार करता है
+तेज़ पुनरावृत्ति चक्र
सहमत
−बेस मॉडल क्वालिटी द्वारा सीमित
−सामान्य क्षमताओं को कम कर सकता है
−क्वालिटी लेबल्ड डेटा की ज़रूरत है
−ओवरफिटिंग का जोखिम
सामान्य भ्रांतियाँ
मिथ
सिर्फ़ प्रीट्रेनिंग ही एक काम का AI असिस्टेंट बनाने के लिए काफ़ी है।
वास्तविकता
एक प्रीट्रेन्ड मॉडल असल में एक एडवांस्ड टेक्स्ट कम्प्लीटर होता है। पोस्ट-ट्रेनिंग ऑप्टिमाइज़ेशन के बिना, यह भरोसेमंद तरीके से इंस्ट्रक्शन फॉलो नहीं कर सकता, नुकसान पहुंचाने वाले रिक्वेस्ट को मना नहीं कर सकता, या सही बातचीत नहीं कर सकता। सभी प्रोडक्शन चैटबॉट को ट्रेनिंग के बाद काफी काम करने की ज़रूरत होती है।
मिथ
पोस्ट-ट्रेनिंग ऑप्टिमाइज़ेशन बस आसान फ़ाइन-ट्यूनिंग है।
वास्तविकता
मॉडर्न पोस्ट-ट्रेनिंग में RLHF, DPO, कॉन्स्टिट्यूशनल AI, और रीज़निंग-फोकस्ड ट्रेनिंग जैसी कई एडवांस्ड टेक्नीक शामिल हैं। इन तरीकों में कॉम्प्लेक्स रिवॉर्ड मॉडलिंग, प्रेफरेंस लर्निंग, और इटरेटिव रिफाइनमेंट शामिल हैं जो बेसिक सुपरवाइज्ड फाइन-ट्यूनिंग से कहीं आगे जाते हैं।
मिथ
ज़्यादा प्रीट्रेनिंग से हमेशा बेहतर मॉडल बनते हैं।
वास्तविकता
रिसर्च से पता चला है कि सिर्फ़ प्रीट्रेनिंग डेटा को स्केल करने से फ़ायदा कम होता है। इस फ़ील्ड में यह बात तेज़ी से पहचानी जा रही है कि पोस्ट-ट्रेनिंग सुधार, खासकर रीज़निंग और अलाइनमेंट के मामले में, एक्स्ट्रा प्रीट्रेनिंग कंप्यूट से ज़्यादा फ़ायदा दे सकते हैं।
मिथ
RLHF और DPO एक ही चीज़ हैं।
वास्तविकता
हालांकि दोनों का मकसद मॉडल को इंसानी पसंद के हिसाब से बनाना है, लेकिन वे अलग-अलग तरीके से काम करते हैं। RLHF एक अलग रिवॉर्ड मॉडल ट्रेन करता है जो रीइन्फोर्समेंट लर्निंग को गाइड करता है, जबकि DPO बिना किसी रिवॉर्ड मॉडल की ज़रूरत के प्रेफरेंस पेयर का इस्तेमाल करके सीधे पॉलिसी को ऑप्टिमाइज़ करता है। DPO आसान है लेकिन इसकी परफॉर्मेंस की खासियतें अलग हो सकती हैं।
मिथ
पोस्ट-ट्रेनिंग से बेस मॉडल की कोई भी समस्या ठीक की जा सकती है।
वास्तविकता
पोस्ट-ट्रेनिंग ऐसी क्षमताएँ नहीं बना सकतीं जो बेस मॉडल में मौजूद न हों। अगर किसी प्रीट्रेन्ड मॉडल में कुछ खास जानकारी या सोचने की क्षमता की कमी है, तो कोई भी फ़ाइन-ट्यूनिंग उन्हें नहीं जोड़ पाएगी। प्रीट्रेनिंग के दौरान जो नींव रखी जाती है, वह बाद में क्या मुमकिन है, इसे तय करती है।
अक्सर पूछे जाने वाले सवाल
प्रीट्रेनिंग और फाइन-ट्यूनिंग में क्या अंतर है?
प्रीट्रेनिंग, आम क्षमताएं बनाने के लिए बड़े पैमाने पर बिना लेबल वाले डेटासेट पर शुरुआती बड़े पैमाने पर ट्रेनिंग है। फ़ाइन-ट्यूनिंग, पोस्ट-ट्रेनिंग का एक रूप है जो छोटे, लेबल वाले डेटासेट का इस्तेमाल करके खास कामों के लिए प्रीट्रेन्ड मॉडल को अडैप्ट करता है। फ़ाइन-ट्यूनिंग, पोस्ट-ट्रेनिंग ऑप्टिमाइज़ेशन की बड़ी कैटेगरी में एक तकनीक है।
AI सेफ्टी के लिए पोस्ट-ट्रेनिंग ऑप्टिमाइज़ेशन क्यों ज़रूरी है?
पोस्ट-ट्रेनिंग वह जगह है जहाँ अलाइनमेंट होता है। RLHF जैसी टेक्नीक मॉडल्स को नुकसान पहुँचाने वाले रिक्वेस्ट को मना करना, खतरनाक कंटेंट बनाने से बचना और इंसानी मूल्यों के हिसाब से व्यवहार करना सिखाती हैं। पोस्ट-ट्रेनिंग के बिना, प्रीट्रेन्ड मॉडल्स अपनी आम क्षमताओं के बावजूद टॉक्सिक, बायस्ड या खतरनाक आउटपुट दे सकते हैं।
पोस्ट-ट्रेनिंग की तुलना में प्री-ट्रेनिंग में कितना समय लगता है?
बड़े मॉडल्स की प्रीट्रेनिंग में आमतौर पर हज़ारों GPUs पर हफ़्तों से महीनों लग जाते हैं। पोस्ट-ट्रेनिंग ऑप्टिमाइज़ेशन आमतौर पर बहुत छोटे कंप्यूट सेटअप पर घंटों से दिनों में पूरा हो जाता है। कंप्यूट रेश्यो 1000:1 या उससे ज़्यादा हो सकता है, इसीलिए ज़्यादातर ऑर्गनाइज़ेशन मॉडल्स को शुरू से बनाने के बजाय पोस्ट-ट्रेनिंग पर ध्यान देते हैं।
क्या आप प्री-ट्रेनिंग छोड़कर सीधे पोस्ट-ट्रेनिंग पर जा सकते हैं?
हाँ, अगर आप किसी मौजूदा प्रीट्रेन्ड मॉडल को शुरुआती पॉइंट के तौर पर इस्तेमाल कर रहे हैं। ज़्यादातर AI कंपनियाँ और रिसर्चर ठीक यही करते हैं, वे एक ओपन-सोर्स या API-बेस्ड मॉडल लेते हैं और उसे कस्टमाइज़ करने के लिए पोस्ट-ट्रेनिंग टेक्नीक इस्तेमाल करते हैं। प्रीट्रेनिंग स्किप करना तभी काम करता है जब कोई सही बेस मॉडल पहले से मौजूद हो।
DPO क्या है और इसकी तुलना RLHF से कैसे की जाती है?
डायरेक्ट प्रेफरेंस ऑप्टिमाइज़ेशन (DPO) एक पोस्ट-ट्रेनिंग तरीका है जो अलग रिवॉर्ड मॉडल को ट्रेन किए बिना प्रेफरेंस पेयर पर मॉडल को सीधे ऑप्टिमाइज़ करके अलाइनमेंट को आसान बनाता है। RLHF में रिवॉर्ड मॉडल ट्रेनिंग सहित तीन स्टेज की ज़रूरत होती है, जबकि DPO सब कुछ एक आसान प्रोसेस में मिला देता है। DPO ज़्यादा तेज़ और स्टेबल है लेकिन इसके नतीजे थोड़े अलग हो सकते हैं।
पोस्ट-ट्रेनिंग ऑप्टिमाइज़ेशन के लिए आपको कितने डेटा की ज़रूरत है?
ज़रूरतें तकनीक के हिसाब से अलग-अलग होती हैं। सुपरवाइज़्ड फ़ाइन-ट्यूनिंग के लिए हज़ारों से लेकर लाखों उदाहरणों की ज़रूरत हो सकती है। RLHF आम तौर पर 100,000+ प्रेफ़रेंस तुलना का इस्तेमाल करता है। DPO, RLHF के बराबर मात्रा के साथ काम कर सकता है। यह प्रीट्रेनिंग में इस्तेमाल होने वाले खरबों टोकन से काफ़ी कम है।
क्या पोस्ट-ट्रेनिंग से मॉडल की क्षमताओं को नुकसान होता है?
पोस्ट-ट्रेनिंग कभी-कभी कुछ बेंचमार्क पर परफॉर्मेंस कम कर सकती है, इस चीज़ को अलाइनमेंट टैक्स कहते हैं। हालांकि, मॉडर्न तकनीकों ने इस समस्या को काफी हद तक कम कर दिया है। अच्छी तरह से डिज़ाइन की गई पोस्ट-ट्रेनिंग, बेस मॉडल की ज़्यादातर आम क्षमताओं को बनाए रखते हुए हेल्पफुलनेस और सेफ्टी को बेहतर बनाती है।
कौन सी कंपनियां प्री-ट्रेनिंग बनाम पोस्ट-ट्रेनिंग पर ध्यान देती हैं?
OpenAI, Anthropic, Google DeepMind, और Meta जैसी कंपनियाँ अपने फ्रंटियर मॉडल्स की प्रीट्रेनिंग में बहुत ज़्यादा इन्वेस्ट करती हैं। ज़्यादातर दूसरे ऑर्गनाइज़ेशन और AI स्टार्टअप्स पोस्ट-ट्रेनिंग पर फोकस करते हैं, खास इंडस्ट्रीज़, यूज़ केस या इम्प्रूवमेंट्स के लिए मौजूदा मॉडल्स को फाइन-ट्यूनिंग करते हैं। इकोसिस्टम फाउंडेशन मॉडल बिल्डर्स और डाउनस्ट्रीम कस्टमाइज़र्स के बीच बँट गया है।
AI डेवलपमेंट पाइपलाइन में पोस्ट-ट्रेनिंग के बाद क्या आता है?
पोस्ट-ट्रेनिंग के बाद, मॉडल्स का आमतौर पर इवैल्यूएशन, सेफ्टी के लिए रेड-टीमिंग, और क्वांटाइजेशन या डिस्टिलेशन जैसे डिप्लॉयमेंट ऑप्टिमाइजेशन से गुजरना होता है। चेन-ऑफ-थॉट प्रॉम्प्टिंग, टूल यूज, और रिट्रीवल-ऑगमेंटेड जेनरेशन जैसी इनफेरेंस-टाइम टेक्नीक बिना एक्स्ट्रा ट्रेनिंग के परफॉर्मेंस को और बेहतर बना सकती हैं।
क्या प्रीट्रेनिंग कम ज़रूरी होती जा रही है?
प्रीट्रेनिंग अभी भी ज़रूरी है, लेकिन AI फील्ड ने सुधार के अगले मोर्चों के तौर पर पोस्ट-ट्रेनिंग और इनफरेंस-टाइम कंप्यूट की तरफ ध्यान दिया है। एक्सटेंडेड रीज़निंग, टेस्ट-टाइम कंप्यूट स्केलिंग और एडवांस्ड फाइन-ट्यूनिंग जैसी टेक्नीक से काफी फ़ायदे मिल रहे हैं, जिससे पता चलता है कि AI की तरक्की का भविष्य सिर्फ़ प्रीट्रेनिंग को स्केल करने से कहीं आगे है।
निर्णय
प्रीट्रेनिंग और पोस्ट-ट्रेनिंग ऑप्टिमाइज़ेशन एक-दूसरे से मुकाबला करने वाले तरीके नहीं हैं, बल्कि एक के बाद एक आने वाले स्टेज हैं जो दोनों ही बहुत मायने रखते हैं। प्रीट्रेनिंग तब ज़रूरी है जब आप शुरू से एक नया फाउंडेशन मॉडल बना रहे हों और आपको बड़ी क्षमताओं की ज़रूरत हो, जबकि पोस्ट-ट्रेनिंग ऑप्टिमाइज़ेशन ज़्यादातर टीमों के लिए प्रैक्टिकल ऑप्शन है जो मौजूदा मॉडल को खास इस्तेमाल के लिए बदलना चाहते हैं। ज़्यादातर ऑर्गनाइज़ेशन के लिए, पोस्ट-ट्रेनिंग सबसे अच्छा रिटर्न ऑन इन्वेस्टमेंट देता है क्योंकि यह बड़ी लैब द्वारा पहले से किए गए काम पर आधारित होता है।