तकनीकी सारांश: प्रोबेबिलिटी यील्ड (Probability Yield) के ऊपर रैंकिंग का मिलान वास्तव में गहन सुरक्षा संरेखण (Deep Safety Alignment) प्रदान करता है
1. समस्या विवरण
ओपन-सोर्स लार्ज लैंग्वेज मॉडल्स (LLMs) AI के लोकतंत्रीकरण के लिए महत्वपूर्ण हैं, फिर भी उनकी खुलापन दुर्भावनापूर्ण अभिनेताओं के लिए कमजोरियां पैदा करता है। एक विशिष्ट और सुलभ खतरा प्रीफिलिंग अटैक (prefilling attack) है, जहाँ एक हमलावर सहायक (assistant) के उत्तर को एक सकारात्मक प्रीफिक्स (जैसे, "यहाँ बम बनाने का तरीका बताया गया है...") के साथ मैन्युअल रूप से प्री-फिल कर देता है ताकि सुरक्षा फिल्टर को बायपास किया जा सके।
Qi et al. (2025) के हालिया कार्य ने डेटा ऑग्मेंटेशन के साथ सुपरवाइज्ड फाइन-ट्यूनिंग (SFT) का उपयोग करके एक "गहन" (deep) सुरक्षा संरेखण रक्षा प्रस्तावित की थी। यह विधि मॉडल्स को केवल निश्चित रिफ्यूज़ल स्ट्रिंग्स (refusal strings) आउटपुट करने के बजाय, एक हानिकारक प्रीफिल के तुरंत बाद प्राकृतिक भाषा में इनकार (refusal) करने के लिए प्रशिक्षित करती है। हालांकि यह मानक सैंपलिंग-आधारित डिकोडिंग और पैरामीटर-ट्यूनिंग शोषण (जैसे, टेम्परेचर पर ग्रिड सर्च) के खिलाफ प्रभावी है, यह पेपर पहचानता है कि यह रक्षा सतही (superficial) बनी हुई है।
लेखक प्रदर्शित करते हैं कि गहन सुरक्षा संरेखण के बावजूद, मॉडल्स अपने टॉप-k प्रेडिक्शन्स के भीतर कम-संभाव्यता (low-probability) लेकिन उच्च-रैंक (highly-ranked) वाले "हानिकारक" टोकन बनाए रखते हैं। प्रोबेबिलिटी मास (probability mass) को अनदेखा करके और इसके बजाय रैंक के आधार पर टोकन चुनकर (विशेष रूप से टॉप 20 के भीतर), एक हमलावर रिफ्यूसल मैकेनिज्म को बायपास कर सकता है। यह पेपर इस भेद्यता का लाभ उठाने के लिए रैंक-असिस्टेड प्रीफिलिंग (Rank-Assisted Prefilling - RAP) हमला पेश करता है और वास्तव में गहन सुरक्षा संरेखण प्राप्त करने के लिए एक नया प्रशिक्षण उद्देश्य, PRESTO प्रस्तावित करता है।
2. कार्यप्रणाली (Methodology)
2.1 RAP हमला
Rank-Assisted Prefilling (RAP) हमला प्रीफिलिंग तकनीक का सामान्यीकरण करता है। उच्च-संभाव्यता वाले टोकन पर निर्भर रहने के बजाय, हमलावर अगले k प्रेडिक्टेड टोकन (जैसे, k=20) में से उन टोकनों का चयन करता है जो हानिकारक कथानक (narrative) को आगे बढ़ाते हैं, चाहे उनका प्रोबेबिलिटी स्कोर कुछ भी हो।
- तंत्र (Mechanism): प्रत्येक डिकोडिंग स्टेप पर, हमलावर टॉप k टोकन का निरीक्षण करता है। यदि कोई टोकन जो हानिकारक प्रीफिल को स्वाभाविक रूप से जारी रखता है (एक "हानिकारक टोकन") टॉप k में दिखाई देता है, तो उसे चुना जाता है, भले ही एक रिफ्यूज़ल टोकन (जैसे, "I") की प्रोबेबिलिटी काफी अधिक हो।
- स्वचालन (AutoRAP): लेखकों ने एक स्वचालित RAP विकसित किया है जो रिफ्यूज़ल टोकन और हानिकारक निरंतरता टोकन (harmful continuation tokens) के बीच अंतर करने के लिए एक क्लासिफायर का उपयोग करता है, जिससे मैन्युअल हस्तक्षेप के बिना हानिकारक सामग्री का व्यवस्थित निष्कर्षण संभव होता है।
2.2 पुश-फॉरवर्ड एलाइनमेंट (Push-Forward Alignment - PFA)
लेखकों का तर्क है कि डेटा ऑग्मेंटेशन डिफेंस की विफलता इसलिए है क्योंकि इसका ध्यान रिफ्यूज़ल टोकन के नेगेटिव लॉग-लाइक्लीहुड (प्रोबेबिलिटी) को कम करने पर है, न कि हानिकारक टोकन के रैंक को नियंत्रित करने पर।
- अवधारणा: एक वास्तव में सुरक्षित मॉडल में, पहले टोकन का वितरण (बिना प्रीफिल के) रिफ्यूज़ल टोकन को शीर्ष रैंकों पर और बिना किसी ऐसे हानिकारक टोकन के होना चाहिए जो प्रीफिल को जारी रखे।
- लक्ष्य: मॉडल को इस रैंक-हार्म मैपिंग (rank-harm mapping) को पहले टोकन वितरण से सभी आगामी डिकोडिंग स्टेप्स तक "पुश फॉरवर्ड" करना चाहिए। इसका अर्थ है यह सुनिश्चित करना कि यदि कोई टोकन हानिकारक है, तो प्रीफिल की उपस्थिति के बावजूद आउटपुट वितरण में उसका रैंक निम्न बना रहे।
- औपचारिकीकरण (Formalization): लेखक एक रैंक-हार्म वितरण R(s,x,xpre) को परिभाषित करते हैं जो s-वें रैंक वाले टोकन के हानिकारक होने की संभावना को दर्शाता है। लक्ष्य सुरक्षित मॉडल (बिना प्रीफिल) और हमले के तहत मॉडल (प्रीफिल के साथ) के बीच अर्थ मूवर्स डिस्टेंस (EMD) को कम करना है।
2.3 PRESTO: PRefill attEntion STOpping
PFA को व्यावहारिक रूप से लागू करने के लिए, लेखक ट्रांसफॉर्मर्स के मल्टी-हेड अटेंशन (MHA) मैकेनिज्म पर आधारित एक रेगुलराइजेशन लॉस, PRESTO प्रस्तावित करते हैं।
- परिकल्पना (Hypothesis): टॉप-k रैंकिंग में हानिकारक टोकन की उपस्थिति हानिकारक प्रीफिल टोकन पर मॉडल के अटेंशन (attention) देने से प्रेरित होती है। यदि मॉडल को प्रीफिल टोकन को "अनदेखा" करने के लिए (अर्थात, उन पर अटेंशन कम करने के लिए) प्रशिक्षित किया जा सकता है, तो आउटपुट वितरण मूल प्रॉम्प्ट के सुरक्षित, रिफ्यूज़ल-हैवी वितरण में वापस आ जाएगा।
- लॉस फंक्शन: PRESTO हानिकारक प्रीफिल टोकन पर लगाए गए अटेंशन स्कोर को कम करता है और नॉन-प्रीफिल टोकन पर अटेंशन को अधिकतम करता है।
ℓPRESTO(θ)=E(x,xpre)∼D[A1−A2]
जहाँ A1 सभी लेयर्स और हेड्स में प्रीफिल टोकन पर औसत अटेंशन है और A2 नॉन-प्रीफिल टोकन पर औसत अटेंशन है।
- प्रशिक्षण: मॉडल को डेटा ऑग्मेंटेशन लॉस (Qi et al., 2025 से) के साथ PRESTO लॉस का उपयोग करके फाइन-ट्यून किया जाता है। दोनों को एक ही फॉरवर्ड पास में कंप्यूट किया जाता है।
3. मुख्य योगदान
- RAP भेद्यता की पहचान: पेपर प्रदर्शित करता है कि डेटा ऑग्मेंटेशन के माध्यम से "गहन" सुरक्षा संरेखण अपर्याप्त है क्योंकि यह हानिकारक टोकन के प्रोबेबिलिटी के बजाय उनके रैंक को दबाने में विफल रहता है।
- Rank-Assisted Prefilling (RAP) हमला: एक नया, सुलभ हमला वेक्टर जो कम-संभाव्यता, उच्च-रैंक वाले टोकन का चयन करके हानिकारक सामग्री निकालता है, जो प्रभावी रूप से डेटा ऑग्मेंटेशन डिफेंस को बायपास करता है।
- Push-Forward Alignment (PFA): एक सैद्धांतिक ढांचा जो प्रस्ताव देता है कि सुरक्षा संरेखण को पहले टोकन वितरण के रैंक-हार्म मैपिंग को पूरी डिकोडिंग प्रक्रिया के दौरान संरक्षित करना चाहिए।
- PRESTO: एक सरल, यांत्रिक रूप से व्याख्या योग्य (mechanistically interpretable) प्रशिक्षण उद्देश्य जो प्रीफिल टोकन के अटेंशन को रेगुलराइज करता है, जिससे RAP हमलों के खिलाफ सुरक्षा काफी बढ़ जाती है।
4. प्रयोगात्मक परिणाम
लेखकों ने तीन लोकप्रिय ओपन-सोर्स मॉडल्स पर PRESTO का मूल्यांकन किया: Llama 2 7B Chat, Qwen 3 8B, और Gemma 3 12B IT।
- हमला सफलता दर (Attack Success Rate):
- मूल डेटा ऑग्मेंटेशन डिफेंस (Qi et al., 2025) के तहत, RAP हमलों ने उच्च सफलता दर प्राप्त की (जैसे, Llama 2 7B के लिए StrongREJECT स्केल पर ~0.74)।
- PRESTO के साथ, RAP हमलों की सफलता काफी कम हो गई। तीनों मॉडल्स में, PRESTO ने डेटा ऑग्मेंटेशन के मुकाबले अकेले 4.7 गुना सुधार में सुरक्षा (हानि कारक स्कोर में कमी) प्रदान की।
- AutoRAP का प्रदर्शन मानव मूल्यांकन के बहुत करीब था, जो हमले और बचाव को स्वचालित करने की व्यवहार्यता की पुष्टि करता है।
- उपयोगिता संरक्षण (Utility Preservation):
- PRESTO के साथ फाइन-ट्यून किए गए मॉडल्स ने MT-Bench (ओपन-एंडेड जनरेशन) और GSM-8K (गणितीय तर्क) पर प्रतिस्पर्धी उपयोगिता बनाए रखी, जिसमें केवल डेटा ऑग्मेंटेशन के साथ प्रशिक्षित मॉडल्स की तुलना में 1% से भी कम प्रदर्शन गिरावट देखी गई।
- मैकेनिस्टिक विश्लेषण:
- अटेंशन विश्लेषण से पता चला कि PRESTO सफलतापूर्वक हानिकारक प्रीफिल टोकन पर अटेंशन को कम करता है, विशेष रूप से मॉडल लेयर्स के दूसरे आधे हिस्से में, जो सुरक्षा निर्णयों के लिए महत्वपूर्ण माने जाते हैं।
- डिफेंस GCG (Greedy Coordinate Gradient) हमले के खिलाफ भी मजबूत रहा, जो दर्शाता है कि PRESTO अन्य हमलों के प्रति नई कमजोरियां पैदा नहीं करता है।
5. महत्व और दावे
पेपर दावा करता है कि "गहन" सुरक्षा संरेखण प्राप्त करने के लिए प्रोबेबिलिटी-आधारित उद्देश्यों से आगे बढ़कर रैंक-आधारित बाधाओं (rank-based constraints) की आवश्यकता है। लेखकों का तर्क है कि मॉडल को रिफ्यूज़ल टोकन के साथ उच्च प्रोबेबिलिटी आउटपुट करने के लिए प्रशिक्षित करना पर्याप्त नहीं है यदि मॉडल के आंतरिक अटेंशन मैकेनिज्म अभी भी हानिकारक निरंतरता को टॉप-k रैंक में रहने देते हैं।
PRESTO को पेश करके, यह कार्य ओपन-सोर्स मॉडल्स को सुरक्षित बनाने की दिशा में एक मार्ग प्रदान करता है जो RAP जैसे व्यावहारिक, कम लागत वाले हमलों के खिलाफ मजबूत हैं। इसका महत्व निम्नलिखित में निहित है:
- सुरक्षा का लोकतंत्रीकरण: ओपन-सोर्स मॉडल्स को ग्राहक-केंद्रित अनुप्रयोगों (जैसे API जो प्रीफिलिंग की अनुमति देते हैं) में तैनात करने में सक्षम बनाना, बिना आसान बाईपास के जोखिम के।
- बायपास करने की लागत: दुर्भावनापूर्ण अभिनेताओं के लिए सुरक्षा को बायपास करने की लागत बढ़ाने के लक्ष्य के अनुरूप होना, बजाय इसके कि बाईपास को असंभव बनाने का प्रयास किया जाए।
- व्याख्यात्मकता (Interpretability): यह एक यांत्रिक स्पष्टीकरण (अटेंशन सप्रेशन) प्रदान करता है कि बचाव क्यों काम करता है, जो सैद्धांतिक संरेखण और व्यावहारिक कार्यान्वयन के बीच के अंतर को पाटता है।
लेखक निष्कर्ष निकालते हैं कि हालांकि कोई भी बचाव पूर्ण नहीं है, PRESTO रैंक-आधारित शोषण के खिलाफ ओपन-सोर्स LLMs को सुरक्षित बनाने की दिशा में एक महत्वपूर्ण कदम है।