burtenshaw/gemma-4-12b-sdpo-pi-mono-brevity-topk-v5
gemma-4-12b-sdpo-pi-mono-brevity-topk-v5
LoRA adapter trained with TRL SDPO on filtered badlogicgames/pi-mono coding-agent traces that contain concrete tool errors or later user corrections.
This adapter is a self-distillation experiment, not a general-purpose coding model release. It uses TRL's experimental SDPO trainer with include_environment_feedback=True, so filtered trace diagnostics are supplied as privileged_context for teacher-conditioned reprompts.
Training Run
- Trackio: Trackio run
- Run name:
gemma4-12b-sdpo-pi-mono-brevity-r96-g4-topk64-th50-20260604 - Output directory:
outputs/sdpo-pi-mono-trace-feedback
Data Filtering
The source data is badlogicgames/pi-mono. The script uses the Dataset Viewer parquet export by default because the raw JSONL files have schema drift that can make direct load_dataset() reconstruction brittle.
A row is kept when it has a substantive prompt plus at least one concrete tool error, environment diagnostic, or later user correction. Rows are scored higher for test/build failures, runtime exceptions, missing files or commands, explicit user feedback, and evidence that the trace continued after the failure.
Selected-data summary:
- Source sessions:
192 - Source files:
192 - Score range:
6.8to19.5 - Average score:
11.596
Category counts:
build_lint_compile: 72command_error: 190missing_file_or_command: 78other_marked_error: 8permission_auth: 3runtime_exception: 34test_or_assertion: 149tool_schema_validation: 7user_feedback: 147
First selected sample preview:
score: 19.5
categories: build_lint_compile, command_error, missing_file_or_command, runtime_exception, test_or_assertion, user_feedback
reward_terms: build_lint_compile, command_error, missing_file_or_command, runtime_exception, test_or_assertion, user_feedback, environment, diagnostic, promises:332, triggeruncaughtexception
prompt:
Analyze GitHub issue(s): https://github.com/badlogic/pi-mono/issues/2291
For each issue:
1. Read the issue in full, including all comments and linked issues/PRs.
2. Do not trust analysis written in the issue. Independently verify behavior and derive your own analysis from the code and execution path.
3. **For bugs**:
- Ignore any root cause analysis in the issue (likely wrong)
- Read all related code files in full (no truncation)
- Trace the code path and identify the actual root cause
- Propose a fix
4. **For feature requests**:
- Do not trust implementation proposals in the issue without verification
- Read all related code files in full (no truncation)
- Propose the most concise implementation approach
- List affected files and changes needed
Do NOT implement unless explicitly asked. Analyze and propose only.
privileged_context:
Tool/environment diagnostic 1 (test_or_assertion, runtime_exception, command_error):
node:internal/process/promises:332
triggerUncaughtException(err, true /* fromPromise */);
^
Error: Transform failed with 3 errors:
/eval.ts:29:2: ERROR: Top-level await is currently not supported with the "cjs" output format
/eval.ts:31:22: ERROR: Top-level await is currently not supported with the "cjs" output format
/eval.ts:46:2: ERROR: Top-level await is currently not supported with the "cjs" output format
at failureErrorWithLog ($WORKSPACE/node_modules/esbuild/lib/main.js:1748:15)
at $WOR
[... trimmed ...]
el' does not exist in type 'SimpleStreamOptions'.
packages/coding-agent/src/core/extensions/runner.ts(242,46): error TS2304: Cannot find name 'ProviderConfig'.
Command exited with code 2
Tool/environment diagnostic 3 (test_or_assertion):
No changes made to packages/coding-agent/src/core/agent-session.ts. The replacement produced identical content. This might indicate an issue with special characters or the text not existing as expected.
Later user correction 1:
what'st he most concise fix? this sounds overly complex
Later user correction 2:
well, we can't just fix it for session_start thenReward Function
The included trace_grounding_reward is intentionally lightweight. It primarily rewards the shortest completion that still has a small amount of grounding in the trace diagnostic. It logs reward, success rate, reward spread, completion word/character counts, brevity, grounding, term hits, and too-short fraction.
For a serious training run, replace the heuristic reward with a verifier that can replay or grade the task, such as tests, build checks, lint output, tool-call validation, or another sandboxed outcome signal.
SDPO Metrics to Watch
For this token-level SDPO setup, review self_distillation/distillation_loss and sdpo/distillation_loss as the reverse-KL distillation surrogate. Also watch self_distillation/success_group_fraction, self_distillation/success_sample_fraction, self_distillation/feedback_used_fraction, and self_distillation/reprompt_sample_fraction to confirm teacher-conditioned batches are actually being constructed. If --full-logit-distillation --distillation-topk N is enabled, distillation_loss becomes the configured top-k/full-logit divergence instead of only the sampled-token surrogate.
Pi Harness via VLLM
This repository contains a PEFT LoRA adapter, not a merged full model. A practical harness setup is to serve the Gemma base model once with VLLM, attach this adapter as a named LoRA module, and point the Pi harness at VLLM's OpenAI-compatible /v1 endpoint.
python -m pip install -U "vllm>=0.10.0" "huggingface_hub>=0.30.0"
hf download burtenshaw/gemma-4-12b-sdpo-pi-mono-brevity-topk-v5 \
--repo-type model \
--local-dir ./adapters/pi-mono-sdpo
vllm serve google/gemma-4-12B-it \
--host 0.0.0.0 \
--port 8000 \
--dtype bfloat16 \
--api-key token-pi-harness \
--enable-lora \
--lora-modules pi-mono-sdpo=./adapters/pi-mono-sdpoThen configure the Pi harness as an OpenAI-compatible provider:
export OPENAI_BASE_URL=http://127.0.0.1:8000/v1
export OPENAI_API_KEY=token-pi-harness
export OPENAI_MODEL=pi-mono-sdpoSmoke-test the server before running the harness:
curl http://127.0.0.1:8000/v1/chat/completions \
-H "Authorization: Bearer token-pi-harness" \
-H "Content-Type: application/json" \
-d '{
"model": "pi-mono-sdpo",
"messages": [
{"role": "system", "content": "You are a careful coding agent."},
{"role": "user", "content": "A test says expected 2 but got 3. What should you inspect first?"}
],
"max_tokens": 128
}'If your Pi harness uses different variable names, map the same three values: OpenAI-compatible base URL, API key, and model name. If your VLLM version cannot serve this Gemma 4 adapter directly, merge the adapter into the base model first or run the adapter with a Transformers/PEFT inference process behind the same OpenAI-compatible API.
Training Metrics
epoch: 0.25total_flos: 0.0train_loss: 0.20423632369299108train_runtime: 1455.3086train_samples_per_second: 0.132train_steps_per_second: 0.066
Reproduce
hf jobs uv run <raw-gist-url-for-train_sdpo_pi_mono_full.py> \
--flavor a10g-large \
--timeout 4h \
--secrets HF_TOKEN \
--env HUB_MODEL_ID=burtenshaw/gemma-4-12b-sdpo-pi-mono-brevity-topk-v5 \
--mode train \
--max-samples 192 \
--max-steps 96 \
--num-generations 4 \
--learning-rate 2e-05 \
--success-reward-threshold 0.5 \
--sdpo-policy-loss-mode distillation_only \
--teacher-regularization none \
--distillation-weight 1.0 \
--distillation-is-clip 2.0 \
--full-logit-distillation \
--distillation-topk 64 \
--trackio-space-id burtenshaw/sdpo-pi-mono-trackio-static-v6 \
--trackio-sdk static \
--run-name gemma4-12b-sdpo-pi-mono-brevity-r96-g4-topk64-th50-20260604Limitations
- The data is filtered from existing traces, so it reflects the trace collector's task distribution and failure modes.
- The reward is a smoke-test heuristic and should not be interpreted as a reliable coding benchmark.
- The model is pushed as a PEFT LoRA adapter; load it with the base model listed above.
- SDPO is experimental in TRL, so pin versions for long-running comparisons.
