composer-replication-framework / research /12-altered-model-rl-critique.md
Codeseys's picture
architect: ADR-011/012/013 + research (alignment-index fix, review-findings closure, LMA channel-ladder)
b4a584a
|
Raw
History Blame Contribute Delete
5.68 kB

Critique: SDPO + Dr.GRPO + trace-replay-DPO on a personality-altered model

Bottom line

The proposed three-channel stage is scientifically interesting but unsafe as a first interpretable run. Applied to an LMA personality-altered SFT model, the combined recipe confounds at least four effects: task RL, self-distillation of altered reasoning, frontier-teacher imitation, and KL anchoring. If capability or moral-scenarios behavior changes, the result will not identify whether RL washed out, preserved, or amplified the alteration.

1. SDPO against the altered model's own hint-conditioned passes

SDPO with the altered model as its own hint-conditioned teacher is only sound if the hints expose latent correct reasoning that the base forward pass underuses. On an altered model, that assumption is fragile. The hint-conditioned pass may instead expose the same distorted policy under a more verbose or more confident trajectory. Then SDPO becomes a consistency regularizer around an already-biased teacher, not an improvement signal.

Main risks:

  • Degenerate fixed point: teacher and student are same-family, same checkpoint, and differ only by prompting. If hints do not add independent information, the optimum is to imitate the altered model's own conditional distribution.
  • Amplification: if the altered model has class-specific moral-scenarios distortion, hinting may increase rationalization of the distorted answer. SDPO can convert a soft bias into a sharper preference.
  • Mode collapse / reduced diversity: KL-to-own-hinted-output rewards low-entropy agreement. Combined with answer-only GRPO, this can produce brittle letter-pattern policies.
  • False preservation: staying close to the altered hint teacher may look like preserving the personality alteration while actually preserving only task-format artifacts.

So SDPO-only is not a benign auxiliary loss here; it is the channel most likely to test the amplification hypothesis. Treat it as an experimental intervention, not a default stabilizer.

2. Hyperparameters and attribution

Do not start with alpha_sdpo=0.2 and beta_replay=0.4 in a combined run. Those weights are large enough that any observed outcome will be uninterpretable. The first real run should isolate channels:

  1. GRPO-only baseline: alpha_sdpo=0, beta_replay=0.
  2. SDPO-only add-on: alpha_sdpo small, beta_replay=0.
  3. Replay-DPO-only add-on: alpha_sdpo=0, beta_replay small.
  4. Combined only after the above: use the smallest weights that showed useful signal without personality drift or distortion amplification.

Safe initial defaults, assuming loss terms are normalized to comparable token/batch scales:

  • alpha_sdpo: 0.02 for first SDPO run; sweep {0.0, 0.02, 0.05}. Avoid 0.2 until there is evidence no amplification occurs.
  • beta_replay: 0.05 for first replay run; sweep {0.0, 0.05, 0.10}. Avoid 0.4 initially because frontier teachers may wash out the alteration and dominate GRPO.
  • Combined pilot: alpha_sdpo=0.02, beta_replay=0.05, only after isolated runs.
  • KL coefficient to frozen altered SFT init: kl_beta=0.02 initial, adaptive to target roughly 0.01-0.03 nats/token on answer+reasoning tokens. Hard-stop or reduce LR if KL exceeds ~0.08 nats/token or if personality probes drift sharply.
  • Also log KL to the original unaltered base/SFT. Do not optimize this KL unless the goal is explicitly de-alteration; use it to measure washout.

3. Reward design to avoid MMLU letter hacking

A letter-correctness reward is too hackable unless the interface is constrained.

Recommended reward:

  • Require structured output: {"answer":"A|B|C|D","rationale":"..."} or a single final Answer: X after reasoning.
  • Parse only the final answer field; reject multiple final answers.
  • Reward: +1 correct, 0 incorrect, -0.2 invalid/unparseable, -0.1 multiple answers, small length penalty after a rationale cap.
  • Randomize option order per epoch and track original label mapping.
  • Balance batches across subject and moral-scenarios classes, especially the collapsed class-3 subset.
  • Hold out a moral-scenarios diagnostic split that is never used for reward.
  • Add calibration metrics: entropy over options, answer distribution, invalid rate, rationale length, and per-class accuracy. A model that learns “always C” or exploits option priors should be obvious.

Do not reward chain-of-thought style itself. If rationales are used, score only final answer and maybe format validity; otherwise the model can learn persuasive distorted rationalizations.

4. Cheapest experiment that distinguishes washout / preserve / amplify

Use one altered checkpoint, one matched unaltered checkpoint, and a fixed evaluation harness. Run short, equal-token pilots with identical prompts/seeds:

  • A0: altered SFT, no RL.
  • A1: GRPO-only.
  • A2: GRPO + SDPO small.
  • A3: GRPO + replay-DPO small.
  • A4: combined small, only if A1-A3 are interpretable.

Evaluate before/after on: general MMLU, moral_scenarios by class, LMA personality/psychiatric probes, KL to altered init, KL to unaltered base, option distribution, and refusal/format invalid rates.

Interpretation:

  • Washout: capability improves while personality markers and altered-specific moral signature move toward unaltered baseline.
  • Preservation: capability improves with stable personality markers and stable moral-scenarios signature.
  • Amplification: moral class-3 or cognitive-distortion probes worsen, confidence/entropy sharpens, or SDPO run diverges more than GRPO-only.

The proposed alpha=0.2/beta=0.4 combined recipe should be considered a later stress test, not a first run.