Test-Time Alignment for Large Language Models via Textual Model Predictive Control
Wang, Chen, Hung et al.
Aligning Large Language Models (LLMs) with human preferences through finetuning is resource-intensive, motivating lightweight alternatives at test time. We address test-time alignment through the lens of sequential decision making, a perspective that reveals two fundamental challenges. When actions are defined at the token level, as in guided decoding, alignment suffers from the curse of horizon. Conversely, when actions are at the response level, as in traditional iterative refinement, the curse of dimensionality emerges. To resolve this trade-off, we draw inspiration from Model Predictive Control (MPC) in control theory to propose Textual Model Predictive Control (TMPC), a novel predictive planning framework adapted for aligning LLMs at inference time. A key limitation of standard MPC is its reliance on predefined, hard segment boundaries, which are often absent in text generation. TMPC overcomes this by introducing two principles inspired by hierarchical reinforcement learning: (1) Hindsight Subgoal Identification, where TMPC analyzes generation subgoals to retrospectively identify high-reward intermediate outputs as subgoals. This allows the framework to discover meaningful, task-specific planning steps (e.g., a sentence in machine translation or a bug fix in code generation.). (2) Subgoal-Conditioned Re-Generation, where these identified subgoals are used to guide subsequent planning iterations. By conditioning on these proven, high-quality subgoals, TMPC ensures stable improvement by building upon previously validated successes. TMPC is evaluated on three tasks with distinct segmentation properties: discourse-level translation, long-form response generation, and program synthesis. The results demonstrate that TMPC consistently improves performance, highlighting the generality.
academic
পাঠ্য মডেল প্রেডিক্টিভ কন্ট্রোলের মাধ্যমে বড় ভাষা মডেলের জন্য পরীক্ষা-সময় সারিবদ্ধকরণ
বড় ভাষা মডেলগুলিকে মানব পছন্দের সাথে সারিবদ্ধ করা সাধারণত সূক্ষ্ম সমন্বয়ের মাধ্যমে অর্জিত হয়, কিন্তু এই পদ্ধতিটি সম্পদ-নিবিড়, তাই হালকা-ওজনের পরীক্ষা-সময় সারিবদ্ধকরণের বিকল্প প্রয়োজন। এই পেপারটি ক্রমানুসারী সিদ্ধান্ত গ্রহণের দৃষ্টিকোণ থেকে পরীক্ষা-সময় সারিবদ্ধকরণ সমস্যার সমাধান করে, দুটি মৌলিক চ্যালেঞ্জ প্রকাশ করে: যখন ক্রিয়া টোকেন স্তরে সংজ্ঞায়িত হয় (যেমন গাইডেড ডিকোডিং), সারিবদ্ধকরণ "মাত্রার অভিশাপ" এর সম্মুখীন হয়; যখন ক্রিয়া প্রতিক্রিয়া স্তরে সংজ্ঞায়িত হয় (যেমন ঐতিহ্যবাহী পুনরাবৃত্তিমূলক অপ্টিমাইজেশন), তখন "সময় মাত্রার অভিশাপ" এর সম্মুখীন হয়। এই ভারসাম্য সমাধানের জন্য, লেখকরা নিয়ন্ত্রণ তত্ত্ব থেকে মডেল প্রেডিক্টিভ কন্ট্রোল (MPC) থেকে অনুপ্রেরণা নিয়ে, পাঠ্য মডেল প্রেডিক্টিভ কন্ট্রোল (TMPC) প্রস্তাব করেন, যা অনুমান-সময় LLM সারিবদ্ধকরণের জন্য একটি নতুন পূর্বাভাসমূলক পরিকল্পনা কাঠামো।
সারিবদ্ধকরণ সমস্যার গুরুত্ব: যদিও বড় ভাষা মডেলগুলি বিভিন্ন NLP কাজে চমৎকার কর্মক্ষমতা প্রদর্শন করে, তবে তাদের আউটপুটকে মানব পছন্দের সাথে সারিবদ্ধ করা একটি মূল চ্যালেঞ্জ থেকে যায়, বিশেষত ছোট আকারের LLM-এর জন্য (যেমন ১০B প্যারামিটারের নিচে)।
ঐতিহ্যবাহী পদ্ধতির সীমাবদ্ধতা:
প্রশিক্ষণ-সময় সারিবদ্ধকরণ পদ্ধতি (যেমন RLHF, DPO) সম্পদ-নিবিড়, ব্যয়বহুল পুনঃপ্রশিক্ষণ প্রয়োজন
পরীক্ষা-সময় সারিবদ্ধকরণ পদ্ধতি মৌলিক ভারসাম্য উপস্থাপন করে:
টোকেন স্তরের গাইডেড ডিকোডিং "সময় মাত্রার অভিশাপ" এর সম্মুখীন হয়
প্রতিক্রিয়া স্তরের পুনরাবৃত্তিমূলক অপ্টিমাইজেশন "মাত্রার অভিশাপ" এর সম্মুখীন হয়
গবেষণা প্রেরণা: এমন একটি পরীক্ষা-সময় সারিবদ্ধকরণ পদ্ধতির প্রয়োজন যা ব্যয়বহুল মডেল পুনঃপ্রশিক্ষণ এড়ায় এবং সময় মাত্রা এবং অনুসন্ধান স্থান জটিলতার মধ্যে কার্যকরভাবে ভারসাম্য রাখে।
উপন্যাস সমস্যা মডেলিং: প্রথমবারের মতো পরীক্ষা-সময় সারিবদ্ধকরণ সমস্যাকে ক্রমানুসারী সিদ্ধান্ত গ্রহণ সমস্যা হিসাবে মডেল করা হয়েছে, বিদ্যমান পদ্ধতিগুলিকে একীভূত করা এবং তাদের মৌলিক ভারসাম্য প্রকাশ করা হয়েছে।
TMPC কাঠামো: পাঠ্য মডেল প্রেডিক্টিভ কন্ট্রোল কাঠামো প্রস্তাব করা হয়েছে, নিয়ন্ত্রণ তত্ত্ব ধারণাগুলিকে ভাষা উৎপাদন কাজে অভিযোজিত করা হয়েছে।
দুটি মূল নীতি:
পরবর্তী দৃষ্টিভঙ্গি উপ-লক্ষ্য সনাক্তকরণ (Hindsight Subgoal Identification): রোলব্যাকআউট থেকে অর্থপূর্ণ পরিকল্পনা পদক্ষেপ আবিষ্কার করা
উপ-লক্ষ্য শর্তাধীন পুনঃউৎপাদন (Subgoal-Conditioned Re-Generation): যাচাইকৃত উপ-লক্ষ্যের উপর ভিত্তি করে পুনরাবৃত্তিমূলক উন্নতি
বিস্তৃত পরীক্ষামূলক যাচাইকরণ: তিনটি ভিন্ন বৈশিষ্ট্যের কাজে পদ্ধতির কার্যকারিতা এবং সর্বজনীনতা যাচাই করা হয়েছে।
TMPC প্রথমবারের মতো মডেল প্রেডিক্টিভ কন্ট্রোলকে ভাষা উৎপাদনের পছন্দ সারিবদ্ধকরণে পদ্ধতিগতভাবে প্রয়োগ করে, নিয়ন্ত্রণ তত্ত্ব এবং NLP ক্রস-ডোমেইনে একটি ফাঁক পূরণ করে।
একীভূত কাঠামো: পরীক্ষা-সময় সারিবদ্ধকরণকে ক্রমানুসারী সিদ্ধান্ত সমস্যা হিসাবে সফলভাবে একীভূত করা হয়েছে, বিদ্যমান পদ্ধতির মৌলিক ভারসাম্য প্রকাশ করা হয়েছে
কার্যকর ভারসাম্য: TMPC সময় মাত্রার অভিশাপ এবং মাত্রার অভিশাপের মধ্যে কার্যকরভাবে ভারসাম্য রাখে
বিস্তৃত প্রযোজ্যতা: তিনটি ভিন্ন বৈশিষ্ট্যের কাজে সামঞ্জস্যপূর্ণ উন্নতি অর্জিত হয়েছে
তাত্ত্বিক অবদান উল্লেখযোগ্য: প্রথমবারের মতো পরীক্ষা-সময় সারিবদ্ধকরণের মৌলিক চ্যালেঞ্জগুলি পদ্ধতিগতভাবে বিশ্লেষণ করা হয়েছে, একটি একীভূত তাত্ত্বিক কাঠামো প্রদান করা হয়েছে
পদ্ধতি উদ্ভাবন শক্তিশালী: MPC-কে পাঠ্য উৎপাদনে সফলভাবে অভিযোজিত করা হয়েছে, ডিজাইন চতুর এবং নীতি স্পষ্ট
পরীক্ষা সম্পূর্ণ এবং পুঙ্খানুপুঙ্খ: তিনটি ভিন্ন বৈশিষ্ট্যের কাজে যাচাই করা হয়েছে, বিস্তারিত বিলোপন পরীক্ষা এবং স্থিতিস্থাপকতা বিশ্লেষণ অন্তর্ভুক্ত
শক্তিশালী শিক্ষা তত্ত্ব (স্তরযুক্ত RL, ট্র্যাজেক্টরি অপ্টিমাইজেশন ইত্যাদি)
সংক্ষিপ্তসার: এটি তাত্ত্বিক উদ্ভাবন এবং ব্যবহারিক প্রয়োগ উভয় ক্ষেত্রেই গুরুত্বপূর্ণ অবদান রাখে এমন একটি উচ্চ-মানের পেপার। লেখকরা নিয়ন্ত্রণ তত্ত্বের MPC কাঠামোকে ভাষা উৎপাদনের পছন্দ সারিবদ্ধকরণ সমস্যায় সফলভাবে অভিযোজিত করেছেন, একটি উদ্ভাবনী TMPC পদ্ধতি প্রস্তাব করেছেন এবং ব্যাপক পরীক্ষার মাধ্যমে এর কার্যকারিতা যাচাই করেছেন। এই কাজটি পরীক্ষা-সময় সারিবদ্ধকরণের জন্য একটি নতুন গবেষণা দিকনির্দেশনা প্রদান করে, উল্লেখযোগ্য একাডেমিক মূল্য এবং ব্যবহারিক তাৎপর্য রয়েছে।