ABForge-Qwen3-8B-Task2

An ABForge model for Task 2: Ablation Plan Generation.

ABForge is a post-training pipeline for paper-grounded ablation design. This checkpoint is post-trained with the full ABForge pipeline: supervised fine-tuning from Qwen/Qwen3-8B followed by rubric-guided GRPO (SFT → GRPO).

Task

Given a paper's context and a goal, the model produces a detailed, controlled ablation experiment design plan (objective, setup, variants, fixed protocols and metrics).

Training data

SFT on train/sft_task2_37019.jsonl, then GRPO on train/RL_task2_30K.jsonl, from SlowGuess/abforge-data (derived from CC-licensed research papers). Evaluation uses the held-out AblationBench split (eval/ablationbench_200.jsonl) of the same dataset.

Related models (Task 2)

Evaluation

Reproduce AblationBench evaluation with the SlowGuess/Abforge_1 code:

git clone https://github.com/SlowGuess/Abforge_1 && cd Abforge_1
huggingface-cli download SlowGuess/abforge-data --repo-type dataset --local-dir data

export MODEL_PATH=SlowGuess/ABForge-Qwen3-8B-Task2

# 1. Generate predictions on AblationBench
python run_inference_local.py --task 2 \
  --input data/eval/ablationbench_200.jsonl \
  --output preds.jsonl \
  --model-path "$MODEL_PATH" --dtype bf16 --max-new-tokens 4096

# 2. Score against the fixed AblationBench rubric (Claude judge)
export ANTHROPIC_API_KEY=<your-key>
python scripts/eval_task2_claude_rubric_v2.py --input preds.jsonl --output scored.jsonl

Links

Citation

@misc{abforge,
  title  = {ABForge: A Post-Training Pipeline for Paper-Grounded Ablation Design},
  author = {ABForge authors},
  year   = {2026},
  howpublished = {\url{https://github.com/SlowGuess/Abforge_1}}
}
Downloads last month
1
Safetensors
Model size
8B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for SlowGuess/ABForge-Qwen3-8B-Task2

Finetuned
Qwen/Qwen3-8B
Finetuned
(1697)
this model

Collection including SlowGuess/ABForge-Qwen3-8B-Task2