A new preprint trains a small "shaping" network to adjust the AI's reward signal per task before fine tuning.
Modern AI assistants learn from human feedback by training against a single static reward model. That grade card scores every response the same way, regardless of whether the user asked for a recipe, a proof, or a poem. The result is patchy alignment and an unstable training process that can drift mid-run.
A new preprint, Meta-Learned Reward Shaping for Reinforcement Learning from Human Feedback (MeRLa), proposes fixing this with a learned shaping layer that adapts the reward signal per task before fine-tuning begins. The paper, posted to arXiv as 2607.26094, reports its method beats four named prior approaches on a single open-weights base model and produces 41% less training instability.
The problem MeRLa targets is older than chatbots. Reinforcement learning from human feedback, or RLHF, is the pipeline that turned GPT-3 into ChatGPT. At its core is a reward model, a neural network trained on human preferences that scores the AI's outputs. Higher score, the AI learns to produce more of that style. Lower score, less. The catch is that the reward model is frozen before fine-tuning starts, and it was trained on a mix of tasks that does not match the fine-tuning distribution. A static, task-agnostic grade card produces sparse learning signals where the model cannot tell the difference between a mediocre answer to a hard question and a polished answer to a trivial one. Training becomes noisy. Alignment degrades in uneven ways.
The fix the paper proposes is reward shaping, an idea borrowed from classical reinforcement learning. Shaping adds an auxiliary signal to the reward so the agent gets denser feedback during training. The trick is that the shaping function must preserve policy optimality: it cannot change which answer the AI should ultimately prefer, or the training target itself shifts. Older shaping methods either relied on hand-engineered heuristics or violated that optimality guarantee and quietly steered the model off-course.
MeRLa's contribution is to learn the shaping function, called Φ(x, y; φ), across a battery of auxiliary tasks before the main RLHF run starts. The shaping layer is meta-trained with a composite objective: a task-discrimination term so the shaping signal varies meaningfully across problem types, an entropy-regularization term to keep the shaped reward from collapsing into a single mode, and a potential-based conservation constraint to preserve the optimality guarantee from classical shaping theory. The result is a reward signal the authors describe as denser, more stable, and task-aware without distorting the underlying preference signal.
On the experiments, the paper reports results on LLaMA-3-8B across four benchmarks. The headline numbers, all from the authors and not yet independently replicated: 90.8% length-controlled win rate on AlpacaEval 2.0, a 9.14 score on MT-Bench, a multi-domain chatbot benchmark, and 41% less training instability. The method is compared against PPO, the original RLHF algorithm; DPO, which skips the reward model entirely; and GRPO and DAPO, two recent group-relative and advantage-weighted variants. The paper also reports that the method retains its gains when stacked on top of process-based and rubric-based enhanced rewards, which are external scoring layers some labs already use.
The four baselines matter because the field has spent three years arguing about which of them is the right default. DPO is faster and simpler than PPO but loses on certain alignment tasks. GRPO and DAPO are newer group-relative methods that have produced strong results on math and code but have their own stability issues. Beating all four on a single base model is a real claim, but the caveat structure is heavy. The experiments run on a single model family at a single scale. The benchmarks are chatbot-judge tests, where a stronger language model grades the outputs, and chatbot judges have known biases toward verbose, confident-sounding answers. The "length-controlled" qualifier on the AlpacaEval win rate exists precisely because the field learned that raw win rates can be gamed by producing longer outputs. LC strips that confound.
This is a preprint, not a deployed system. There is no API, no product, no integration with any commercial assistant. The result is one arXiv paper, posted without peer review, with author-reported numbers on a single base model. The interesting question is whether the meta-learned shaping idea survives contact with larger models, harder alignment tasks, and labs willing to run the ablations. If it does, treating RLHF as a static grade card starts to look as dated as hand-engineered reward functions already are. Until then, the 90.8% number is a real signal worth watching, not a settled benchmark.