Machine Unlearning Meets Adversarial Robustness via Constrained Interventions on LLMs
Rezkellah, Dakhmouche
With the increasing adoption of Large Language Models (LLMs), more customization is needed to ensure privacy-preserving and safe generation. We address this objective from two critical aspects: unlearning of sensitive information and robustness to jail-breaking attacks. We investigate various constrained optimization formulations that address both aspects in a \emph{unified manner}, by finding the smallest possible interventions on LLM weights that either make a given vocabulary set unreachable or embed the LLM with robustness to tailored attacks by shifting part of the weights to a \emph{safer} region. Beyond unifying two key properties, this approach contrasts with previous work in that it doesn't require an oracle classifier that is typically not available or represents a computational overhead. Surprisingly, we find that the simplest point-wise constraint-based intervention we propose leads to better performance than max-min interventions, while having a lower computational cost. Comparison against state-of-the-art defense methods demonstrates superior performance of the proposed approach.
academic
মেশিন আনলার্নিং এবং অ্যাডভার্সারিয়াল রোবাস্টনেস এলএলএম-এ সীমাবদ্ধ হস্তক্ষেপের মাধ্যমে
শিরোনাম: Machine Unlearning Meets Adversarial Robustness via Constrained Interventions on LLMs
লেখক: Fatmazohra Rezkellah (Université Paris-Dauphine), Ramzi Dakhmouche (EPFL & Empa)
শ্রেণীবিভাগ: cs.LG cs.CL cs.CR cs.CY math.OC
প্রকাশনা সম্মেলন: 39th Conference on Neural Information Processing Systems (NeurIPS 2025) Workshop: Constrained Optimization for Machine Learning (COML)
বৃহৎ ভাষা মডেল (LLMs) এর ব্যাপক গ্রহণের সাথে সাথে গোপনীয়তা সুরক্ষা এবং নিরাপদ প্রজন্ম নিশ্চিত করার জন্য আরও কাস্টমাইজেশনের প্রয়োজন। এই পেপারটি দুটি মূল দিক থেকে এই লক্ষ্যটি সমাধান করে: সংবেদনশীল তথ্যের বিস্মৃতি এবং জেইলব্রেক আক্রমণের বিরুদ্ধে স্থিতিস্থাপকতা। গবেষকরা বিভিন্ন সীমাবদ্ধ অপ্টিমাইজেশন সূত্র প্রস্তাব করেছেন যা LLM ওজনে ন্যূনতম সম্ভাব্য হস্তক্ষেপ খুঁজে বের করে এই দুটি দিক একীভূত করে, নির্দিষ্ট শব্দভাণ্ডার সেটকে অপ্রাপ্য করে তোলে বা আংশিক ওজন আরও নিরাপদ অঞ্চলে স্থানান্তরিত করে LLM-এর কাস্টম আক্রমণের বিরুদ্ধে স্থিতিস্থাপকতা বৃদ্ধি করে। এই পদ্ধতিটি সাধারণত অনুপলব্ধ বা গণনামূলক ওভারহেডের প্রতিনিধিত্ব করে এমন ওরাকেল শ্রেণীবিভাজকের প্রয়োজন করে না। আশ্চর্যজনকভাবে, লেখকরা আবিষ্কার করেছেন যে প্রস্তাবিত সবচেয়ে সহজ পয়েন্ট-সীমাবদ্ধ হস্তক্ষেপ পদ্ধতি সর্বাধিক-ন্যূনতম হস্তক্ষেপের চেয়ে ভাল কর্মক্ষমতা প্রদান করে, যখন গণনামূলক খরচ কম থাকে।
মেশিন আনলার্নিং সমস্যা: ন্যূনতম গণনামূলক খরচে ভাষা মডেলের প্রজন্ম স্থান থেকে নির্দিষ্ট তথ্য (নির্দিষ্ট শব্দভাণ্ডার সেট) কীভাবে মুছে ফেলতে হয়
প্রতিকূল স্থিতিস্থাপকতা সমস্যা: ভাষা মডেলকে বিপজ্জনক বা বিষাক্ত সামগ্রীর দিকে পরিচালিত করে এমন জেইলব্রেক প্রতিকূল আক্রমণের বিরুদ্ধে কীভাবে আরও স্থিতিস্থাপক করা যায়
নিরাপত্তা-সংবেদনশীল অ্যাপ্লিকেশনে (যেমন অনলাইন সামগ্রী মডারেশন, গোপনীয় ডেটা প্রক্রিয়াকরণ) LLMs এর স্থাপনার সাথে সাথে, প্রজন্ম মডেল আউটপুটের নিরাপত্তা নিশ্চিত করা একটি মূল প্রয়োজনীয়তা হয়ে উঠেছে। বিদ্যমান পদ্ধতিগুলি গণনামূলক দক্ষতা এবং প্রতিরক্ষা কার্যকারিতার মধ্যে ট্রেড-অফ সম্মুখীন হয়।
লেখকরা রিগ্রেশনে নীতিগত স্থিতিস্থাপকতা পদ্ধতি দ্বারা অনুপ্রাণিত, একটি একীভূত কাঠামো প্রস্তাব করেছেন যা প্রতিকূল স্থিতিস্থাপকতা এবং আনলার্নিং সমস্যা একসাথে সমাধান করে, লুকানো স্থানের পথে তথ্যের নিহিত সংরক্ষণের তথ্য ব্যবহার করে।
একীভূত কাঠামো: বিভিন্ন সীমাবদ্ধ অপ্টিমাইজেশন সমস্যা প্রস্তাব এবং সমাধান করে যা LLMs-কে প্রতিকূল আক্রমণের বিরুদ্ধে স্থিতিস্থাপক এবং অপ্রয়োজনীয় সামগ্রী ভুলে যাওয়ার ক্ষমতা প্রদান করে
বাহ্যিক শ্রেণীবিভাজক প্রয়োজন নেই: প্রম্পট স্থানে ক্রমাগত শিথিলকরণ প্রবর্তন এবং সরাসরি সীমাবদ্ধ ধারণা এমবেডিং-এ হস্তক্ষেপ সম্পাদন করে কৃত্রিম প্রোবের প্রয়োজন অতিক্রম করে
কর্মক্ষমতা উন্নতি: অত্যাধুনিক প্রতিরক্ষা অ্যালগরিদমের তুলনায় কর্মক্ষমতা উন্নতি প্রদর্শন করে এবং LLMs-এ অর্থনৈতিক আনলার্নিংয়ের জন্য নতুন অত্যাধুনিক স্তর স্থাপন করে
গণনামূলক দক্ষতা: সবচেয়ে সহজ পয়েন্ট-সীমাবদ্ধ পদ্ধতি কর্মক্ষমতা এবং গণনামূলক খরচ উভয় ক্ষেত্রেই জটিল সর্বাধিক-ন্যূনতম হস্তক্ষেপকে ছাড়িয়ে যায়
বৈশিষ্ট্য: সর্বনিম্ন কেস ইনপুট পরিস্থিতি বিবেচনা করে, বিচ্ছিন্ন কাঠামো পরিচালনার জন্য সম্ভাব্যতা শিথিলকরণ ব্যবহার করে
অসুবিধা: ক্ষতিকারক ধারণা সেটের জ্ঞান প্রয়োজন, অত্যন্ত রক্ষণশীল হতে পারে
ন্যূনতম হস্তক্ষেপের উপর ভিত্তি করে সহজ পয়েন্ট-সীমাবদ্ধ কৌশল, LLM MLP সক্রিয়করণ জেইলব্রেক প্রম্পটের জন্য বিপজ্জনক আউটপুট এমবেডিংয়ের সমান নয় করে তোলে:
min_{θ^{(l)}∈R^{d_l}} ‖δ_l‖_2^2
subject to ‖o^{(l)}(x; θ + δ_l) - c_i‖_2 ≥ ε, ∀i ≤ n
সুবিধা: KKT শর্তের উপর ভিত্তি করে আধা-বন্ধ-ফর্ম সমাধান, উচ্চ গণনামূলক দক্ষতা, সর্বোত্তম কর্মক্ষমতা
অসুবিধা: নিষিদ্ধ ধারণা সেট C প্রয়োজন
পেপারটি প্রতিকূল প্রশিক্ষণ, মেশিন আনলার্নিং, LLM নিরাপত্তা এবং অন্যান্য সম্পর্কিত ক্ষেত্রের সর্বশেষ গবেষণা কাজ সহ একাধিক প্রাসঙ্গিক ক্ষেত্রের গুরুত্বপূর্ণ কাজ উদ্ধৃত করে, এই গবেষণার জন্য একটি দৃঢ় তাত্ত্বিক ভিত্তি এবং তুলনামূলক বেঞ্চমার্ক প্রদান করে।
সামগ্রিক মূল্যায়ন: এটি LLM নিরাপত্তা ক্ষেত্রে একটি গুরুত্বপূর্ণ অবদান সহ একটি পেপার, যা একীভূত সীমাবদ্ধ অপ্টিমাইজেশন কাঠামোর মাধ্যমে আনলার্নিং এবং স্থিতিস্থাপকতা সমস্যা একসাথে সমাধান করে এবং গণনামূলকভাবে দক্ষ সমাধান প্রদান করে। যদিও তাত্ত্বিক বিশ্লেষণ এবং মূল্যায়ন দিক থেকে কিছু অপূর্ণতা রয়েছে, তবে এর ব্যবহারিক মূল্য এবং উদ্ভাবনী পদ্ধতি এটিকে এই ক্ষেত্রে একটি গুরুত্বপূর্ণ অগ্রগতি করে তোলে।