Mini Post-Training Lab for Reasoning Models

Built a post-training pipeline for a small open-source language model using synthetic preference data and Direct Preference Optimization (DPO). Evaluated pre- and post-training behavior on reasoning, refusal, and instruction-following tasks using custom graders.

Machine Learning Engineer

Python, PyTorch, HF TRL

DPO, LoRA, Synthetic Data

Objective & Methodology

The primary objective was to establish a rigorous, minimal environment to investigate the effects of Direct Preference Optimization (DPO) on smaller scale reasoning models. While large-scale alignment has been extensively documented, studying these dynamics in constrained environments offers granular insights into the mechanics of reward hacking and task degradation.

We utilized Low-Rank Adaptation (LoRA) to efficiently fine-tune the model, relying on a curated dataset of synthetic preferences designed specifically to test boundary conditions in logical reasoning and safe refusal.

Logical Reasoning (GSM8K) 42.5% 48.2% +5.7%
Safe Refusal Rate 78.0% 94.5% +16.5%
Instruction Following (IFEval) 61.0% 58.5% -2.5%

Analyzing Tradeoffs

The evaluation illuminated a common alignment tax: while safety (refusal) and targeted reasoning improved, general instruction adherence slightly regressed. Furthermore, a qualitative review of the model's outputs indicated a marked increase in verbosity.

"The model learned to conflate 'helpful reasoning' with longer explanations, often generating verbose chain-of-thought traces even for trivial queries where a concise answer was requested."