connections-rl-sft (Qwen2.5-1.5B)

Rank-16 LoRA SFT adapter for Qwen2.5-1.5B-Instruct, trained to solve NYT Connections puzzles. The SFT warm start for the GRPO adapter connections-rl-grpo.

Part of connections-rl: a two-scale study of what verifiable-reward RL actually transfers, built and trained entirely on free-tier GPUs (Colab/Kaggle T4s).

Training

3 epochs of LoRA SFT (r=16, Ξ±=32, all-linear) on 807 puzzles (2023-06 β†’ 2025-08), chat-formatted with a fixed <ANSWER> output schema. Chronological split: everything evaluated on postdates everything trained on.

Evaluation (162 held-out puzzles, 2025-12 β†’ 2026-05)

Metric base this adapter GRPO
Solve rate 0.0% 0.0% 0.0%
Invalid outputs 32.1% 74.1% 2.5%
Mean reward 0.049 βˆ’0.038 0.113

Honest caveat: this adapter makes hallucination worse. SFT teaches the answer format without board-grounding β€” it more than doubles the rate of answers containing words not on the board. That failure mode is what the GRPO stage repairs. Full analysis in the technical report.

Usage

from peft import PeftModel
from transformers import AutoModelForCausalLM

base = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2.5-1.5B-Instruct")
model = PeftModel.from_pretrained(base, "jacksonlukas/connections-rl-sft")

Or serve with vLLM: vllm serve Qwen/Qwen2.5-1.5B-Instruct --enable-lora --lora-modules connections-rl-sft=<path>

Downloads last month
39
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support

Model tree for jacksonlukas/connections-rl-sft

Adapter
(1232)
this model