PulseExploreJournal ClubDebatesTrendingResearchersJournals
Instagram
HomeExploreJournal ClubTrending
Synapse
⌘+K
Synapse
July 31, 20260 citationsOpen Access

Agent-Trace Fine-Tuning of Small Language Models under Constrained Compute

View Full Paper
AAAnkit Aglawe

Key Points

  • This research aims to determine the effectiveness of agent-trace fine-tuning on small language models using limited computational resources.
  • Implemented a novel pipeline incorporating completion-only loss masking, mixed-domain replay, and fuzzy train/test decontamination.
  • Documented the process using 16 GB-class GPUs to measure the output against the same-instrument baseline.
  • Released the complete recipe, trace corpus, harness, and experiment log for validation.
  • Revised recipe improves coding ability recovery to mean scores of 71.7 on HumanEval and 67.3 on HumanEval+.
  • Observed catastrophic forgetting with a loss of 18.9 points on HumanEval and 18.3 on HumanEval+ initially.
  • Identified a 21% empty-answer rate in Qwen3-4B runs, which was not detected by standard benchmarks.

Abstract

Small language models (SLMs, 3-8B) are increasingly fine-tuned on agent traces (recorded multi-step sessions of a frontier model using tools) in the hope of transferring agentic behavior at a fraction of the serving cost. In practice these fine-tunes are usually released with either loss-only evidence or self-reported, non-reproducible benchmark tables, and almost never with a baseline measured under the same conditions. We ask whether a single practitioner, using only individual 16 GB-class GPUs, can produce agent-trace fine-tunes that are useful and honestly characterized, and we document what it takes to get there. Our first, literature-naive recipe reduced held-out trace loss substantially on every model trained, yet regressed general coding ability by 18.9 points on HumanEval and 18.3 on HumanEval+ relative to the base model: a textbook case of catastrophic forgetting that its loss curve completely hid. We rebuild the pipeline around three evidence-backed changes: completion-only loss masking, mixed-domain replay, and fuzzy train/test decontamination with post-hoc containment measurement against both reported benchmarks. On Granite-3B, the revised recipe recovers roughly half of the lost coding ability (mean 71.7/67.3 HumanEval/HumanEval+ over three runs, vs base 81.7/76.2) and reduces to zero on our probe a distribution-mismatch failure we name and characterize as session leakage. A cross-family run on Qwen3-4B reproduces the recipe's behavior and surfaces a failure class benchmarks do not measure: a 21% empty-answer rate invisible to HumanEval and loss curves alike. Six of our headline results were artifacts of the apparatus or our own process, each caught only by re-measurement against a same-instrument baseline. We argue that at this scale the binding constraint on useful specialization is evaluation discipline, not compute, and release the full recipe, the 10.6k-trace corpus, the harness, and a complete experiment log to make the claim checkable. All resulting models, the trace corpus, and every evaluation script are released publicly as the Parable model series: https://huggingface.co/collections/AnkitAI/parable

Ask AI
Helpful
Bookmark
Share
View Full Paper

Cite This Study

Ankit Aglawe (2026) studied this question.

synapsesocial.com/papers/6a6c5510747664a1aa73d5c0https://doi.org/10.5281/zenodo.21676406
Ask AI
Helpful
Bookmark
Share
View Full Paper