jatin salve

ai systems / 2026 - Present

verifiable preference optimization for reasoning llms

A verified reasoning-data pipeline and LoRA DPO study with careful validation of truncation and answer parsing.

1,800 reasoning traces330 preference pairs61–63 generated tokens/s

Python · PyTorch · Hugging Face TRL · LoRA · Slurm

Overview

Built an experiment pipeline for verifiable preference optimization on mathematical reasoning tasks. The work connects dataset construction, deterministic answer checking, LoRA-based DPO, and inference benchmarking on NVIDIA L4 GPUs.

Dataset and training

Stage Output Validation
Reasoning generation 1,800 traces Completion and truncation checks
Answer verification Verified final answers Deterministic extraction
Preference construction 330 pairs Chosen/rejected consistency
LoRA DPO Fine-tuned adapter Held-out generation benchmark
  • Generated 1,800 reasoning traces and verified final answers against GSM8K labels.
  • Constructed 330 chosen/rejected preference pairs for preference training.
  • Fine-tuned with LoRA and Hugging Face TRL using DPO.
  • Established an inference baseline of 61–63 generated tokens per second on an NVIDIA L4.

Evaluation lessons

Initial results exposed truncation and answer-parsing failures that could silently create noisy preference labels. I redesigned validation around explicit completion checks and deterministic answer extraction before comparing training outcomes.

This project treats preference optimization as a systems and measurement problem: reliable conclusions require trustworthy labels, reproducible generation, and performance-aware evaluation.

Failure analysis

The first evaluation pass looked plausible but mixed truncated completions with answer-parsing failures. Separating those cases prevented invalid outputs from becoming preference labels and made subsequent DPO comparisons defensible.