HyperAIHyperAI

Command Palette

Search for a command to run...

LLM-as-a-Tutor: 비검증 강화학습을 위한 정책 인식형 프롬프트 적응

Yujin Kim Namgyu Ho Sangmin Hwang Joonkee Kim Yongjin Yang Sangmin Bae Seungone Kim Jaehun Jung Se-Young Yun Hwanjun Song

초록

비검증 지시 수행을 위한 강화학습(RL)은 점점 더 프롬프트별 평가 기준을 가진 LLM 심사자를 보상 신호로 활용하고 있다. 최근 방법들은 훈련 중 변화하는 정책에 이러한 평가 기준을 적응시키지만, 훈련 프롬프트 자체는 고정된 말뭉치에서 추출되어 정적으로 유지된다. 이러한 정적 접근 방식은 종종 프롬프트 난이도와 정책 능력 간의 심각한 불일치를 초래하며, 프롬프트가 롤아웃 간 품질 차이를 이끌어내지 못할 때 심사자가 변별력 있는 보상 신호를 복구할 수 없게 만든다. 이 불일치를 해결하기 위해 우리는 LLM-as-a-Tutor를 소개한다. 이 프레임워크는 LLM의 역할을 심사자에서 교사로 확장하여, 단일 모델이 정책 롤아웃을 쌍대 비교하여 도전적이지 않은 프롬프트를 탐지하는 시험관 역할과, 해당 프롬프트에 원자적 제약 조건을 추가하는 생성자 역할을 동시에 수행한다. 이 추가 전용 설계는 정책의 능력에 맞춰 난이도를 단조적으로 높여 외부 난이도 스케줄 없이 자기 보정 훈련 신호를 생성한다. 세 가지 복잡한 지시 수행 벤치마크에서 우리의 방법은 정책 비인식 기준선과 평가 기준이나 프롬프트를 재작성하는 기존 정책 적응형 방법 모두를 일관되게 능가하여, 프롬프트 적응이 비검증 RL에서 정책 인식의 부재한 축임을 시사한다.

One-sentence Summary

LLM-as-a-Tutor, proposed by researchers from KAIST, Upstage, University of Toronto, and others, is a framework that extends an LLM from judge to tutor by appending atomic constraints to non-challenging prompts, enabling policy-aware prompt adaptation that monotonically raises difficulty with policy capability, yielding self-calibrating rewards and outperforming prior methods on three complex instruction-following benchmarks.

Key Contributions

  • The paper identifies that a mismatch between prompt difficulty and policy capability can suppress reward variance, framing this mismatch as a key precondition that limits rubric-equipped judges in non-verifiable RL.
  • It introduces LLM-as-a-Tutor, a framework that extends a single LLM from judge to tutor by detecting non-challenging prompts through pairwise comparison and appending atomic constraints, monotonically raising difficulty in step with the policy without external schedules.
  • Across three complex instruction-following benchmarks, the method consistently outperforms both policy-unaware baselines and prior policy-adaptive approaches that adapt rubrics or rewrite prompts, establishing prompt adaptation as a complementary axis of policy-awareness in non-verifiable RL.

Introduction

Reinforcement learning for instruction following in large language models has advanced by replacing coarse scalar reward models with LLM judges that score responses against instance-specific rubrics, yielding fine-grained discriminative signals. However, these rubric-equipped judges only work when the prompt elicits rollouts of varying quality; prompts that are trivially easy or impossibly hard for the current policy cause the reward signal to collapse, regardless of judge quality. Existing pipelines draw training prompts from static corpora, leaving the supply of such challenging prompts to chance, and prior policy-adaptive methods either fall back on coarse scalar rewards or use generic, prompt-agnostic rubrics that lack the needed granularity. The authors introduce LLM-as-a-Tutor, which reuses the judge LLM to detect non-challenging prompts via pairwise rollout comparison and then augments them with atomic constraints, creating a self-calibrating curriculum that adapts prompts to the policy’s current capability and consistently outperforms both static and prior adaptive baselines.

Method

The authors introduce a framework that dynamically adapts training prompts to match a policy’s evolving capability during reinforcement learning for non-verifiable instruction-following tasks. The core idea is that a fixed prompt corpus can become non-discriminative as the policy improves, leading to rollouts with little quality variance and thus weak learning signals. To address this, they propose an LLM-as-a-Tutor that iteratively appends atomic constraints to prompts that the current policy has mastered, thereby escalating difficulty in lockstep with the policy’s progress.

The training pipeline rests on a rubric-based reward model. For each prompt xxx, a base rubric R(x)={(rk,wk)}k=1KR(x) = \{(r_k, w_k)\}_{k=1}^KR(x)={(rk,wk)}k=1K defines KKK quality criteria with non-negative weights summing to one. An LLM judge J\mathcal{J}J scores a response yyy against each criterion independently, and the overall reward is a weighted sum: s(x,y)=k=1KwkJ(yx,rk).s(x, y) = \sum_{k=1}^{K} w_k \cdot \mathcal{J}(y \mid x, r_k).s(x,y)=k=1KwkJ(yx,rk). Policy optimization follows the Group Relative Policy Optimization (GRPO) framework. For a given prompt, GGG rollouts are sampled from the old policy πθold\pi_{\theta_{\text{old}}}πθold, and the policy πθ\pi_\thetaπθ is updated by maximizing a clipped surrogate objective that includes a KL penalty toward a frozen reference policy. The per-rollout advantage A(i)A^{(i)}A(i) is computed in a group-relative manner: A(i)=s(x,y(i))μ(x)σ(x),A^{(i)} = \frac{s(x, y^{(i)}) - \mu(x)}{\sigma(x)},A(i)=σ(x)s(x,y(i))μ(x), where μ(x)\mu(x)μ(x) and σ(x)\sigma(x)σ(x) are the mean and standard deviation of the rollout scores within the group. This advantage provides a meaningful learning signal only when the rollouts exhibit sufficient score spread. The authors emphasize that discriminativeness is a property of the prompt–policy pair, not the prompt alone; when the policy has mastered a prompt, the rollouts become indistinguishable in quality, and the advantage collapses.

To overcome this, the LLM-as-a-Tutor operates at the start of every training iteration. It first generates a base rubric R(x)R(x)R(x) for each prompt. Then, for each prompt, it samples two rollouts from the current policy and makes a binary pairwise judgment: whether the two responses are indistinguishable in quality. This pairwise comparison leverages the observation that LLMs are sharper and more human-aligned in pairwise settings than in pointwise scoring. If the rollouts are deemed indistinguishable, the prompt is declared non-discriminative for the current policy.

When a prompt is non-discriminative, the tutor invokes its second role as a constraint generator. It produces an atomic constraint ccc, a single requirement along a dimension left unspecified by the original prompt, along with a corresponding rubric Rc={(rc,j,wc,j)}j=1mR_c = \{(r_{c,j}, w_{c,j})\}_{j=1}^mRc={(rc,j,wc,j)}j=1m that evaluates adherence to that constraint. The adapted prompt and rubric set are formed by appending the constraint and merging the rubrics: x~=xc,R~(x)=R(x)Rc,\tilde{x} = x \oplus c, \quad \tilde{R}(x) = R(x) \cup R_c,x~=xc,R~(x)=R(x)Rc, with the weights renormalized to sum to one. Discriminative prompts are left unchanged. The reward and GRPO update then proceed exactly as before but with the adapted (x~,R~(x))(\tilde{x}, \tilde{R}(x))(x~,R~(x)), making the adaptation the only point of modification relative to the baseline.

This append-only design ensures that difficulty monotonically increases: any response satisfying x~\tilde{x}x~ must also satisfy xxx, so the adapted prompt is never easier than the original. Moreover, retaining the original rubrics R(x)R(x)R(x) in R~(x)\tilde{R}(x)R~(x) keeps the base criteria active, so as constraints accumulate, scoring well on R(x)R(x)R(x) itself becomes more demanding. The tutor is invoked at every adaptation interval (e.g., every epoch), allowing prompts adapted in earlier iterations to be revisited and further escalated as the policy improves. The sequence of prompts {x(t)}\{x^{(t)}\}{x(t)} is thus non-decreasing in difficulty, with escalation triggered only when the current policy renders a prompt non-discriminative. This dynamic, policy-driven curriculum realizes a per-prompt difficulty progression that a static corpus cannot provide, effectively shaping the training signal so the student discovers new capabilities through RL without requiring target responses from the tutor.

Experiment

The evaluation trains a Qwen3-1.7B policy with a Qwen3-8B tutor and judge on WildChat prompts, testing on instruction-following benchmarks. Experiments show that policy-adaptive prompt modification via appended atomic constraints outperforms both policy-unaware baselines and rubric-only adaptation, as it restores discriminative reward signals when prompts become too easy for the current policy. Ablations confirm that targeting only non-discriminative prompts and cumulatively adding constraints are essential, while analysis reveals the tutor's decisions track empirical difficulty and scale with policy capability. The results suggest that shaping the prompt distribution through a tutor's discriminative judgments constitutes an effective form of distillation that allows the policy to improve beyond the tutor's own answer quality.

LLM-as-a-Tutor achieves the highest average score across instruction-following benchmarks and outperforms all baselines on five of six metrics, showing that prompt-level adaptation provides a richer training signal than rubric-based methods alone. Policy-adaptive rubrics improve over policy-unaware approaches, but modifying prompts can restore variance in saturated tasks where even well-adapted rubrics fail to discriminate. Distillation without prompt modification degrades performance below the untrained policy, indicating that naive imitation is harmful. Policy-adaptive rubrics outperform static rubrics and WildChecklists, confirming that adapting the reward signal to the current policy yields better instruction following. LLM-as-a-Tutor surpasses all baselines on five of six metrics, with prompt adaptation reintroducing discriminative variance when rubrics cannot differentiate rollout quality.

Adaptive constraint addition, which selects prompts based on the policy's own rollouts, outperforms all non-adaptive variants. Applying constraints to all prompts or a random subset is less effective, and using a larger model's judgments to identify targets also falls short despite modifying more prompts, demonstrating that the current policy's behavior is the key signal. The adaptive method achieves the highest average score, surpassing always-on, random, and off-policy selection strategies. Adding constraints to every prompt or to a random 28% of prompts both yield lower performance, showing that indiscriminate modification is harmful. Using an 8B model's rollouts to select prompts (Wrong) modifies 47% of prompts but still underperforms the adaptive approach, indicating that difficulty alone is insufficient and the policy's own behavior matters.

Appending new constraints while retaining previous ones (Append) outperforms both substituting constraints (Reset) and rewriting the entire prompt (Rewrite) on all evaluated benchmarks. Append monotonically increases prompt difficulty without drifting from the original task distribution, leading to higher scores on FB, AdvIF, and IB, and the highest average performance. Append achieves the highest scores on all three benchmarks (FB, AdvIF, IB) and the highest average. Reset and Rewrite both underperform Append, with Rewrite scoring slightly higher than Reset but still below Append. Append's monotonic difficulty increase and preservation of the seed distribution explain its superior results, while Reset and Rewrite risk reducing difficulty or drifting from the task.

Adding a single atomic constraint to a saturated prompt reintroduces discriminative reward variance. On the original easy prompt, all model rollouts achieved perfect scores, yielding zero standard deviation and no informative signal. After appending a requirement to name the specific company policy, the mean score dropped and standard deviation increased, separating compliant responses from those that deflect without satisfying the constraint. Before the constraint, all rollouts scored 100 with zero standard deviation, making the prompt uninformative for training. After the constraint, a rollout that explicitly cited a policy name scored 100, while one that only advised consulting a supervisor scored 60, demonstrating a clear quality gap.

The evaluation compares LLM-as-a-Tutor, a prompt-level adaptation method, against various baselines for instruction following, validating that adapting prompts based on the policy's own rollouts and appending atomic constraints yields a richer training signal than static rubrics or indiscriminate modifications. Adaptive constraint selection and monotonic difficulty increase via appending outperform alternative strategies like resetting or rewriting prompts, and this approach restores discriminative variance in saturated tasks where even well-adapted rubrics fail. The findings confirm that the policy's behavior is the key signal for effective adaptation, with LLM-as-a-Tutor achieving superior performance by combining policy-adaptive rubrics with targeted prompt modifications.


AI로 AI 구축

아이디어에서 출시까지 — 무료 AI 코코딩, 즉시 사용 가능한 환경, 최적의 GPU 가격으로 AI 개발을 가속화하세요.

AI 협업 코딩
바로 사용 가능한 GPU
최적의 가격

HyperAI Newsletters

최신 정보 구독하기
한국 시간 매주 월요일 오전 9시 에 이번 주의 최신 업데이트를 메일로 발송합니다
이메일 서비스 제공: MailChimp