Views
No views yet
aime-gen-qwen3-4b-lora-v3
into Qwen/Qwen3-4B-Instruct-2507. Generates novel, difficulty-calibrated AIME-style problems
from a bare one-line prompt (no system prompt, no few-shot). Trained on the
companion SFT dataset.1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4REPO = "William2390401/aime-gen-qwen3-4b-v3"
5tok = AutoTokenizer.from_pretrained(REPO)
6model = AutoModelForCausalLM.from_pretrained(REPO, torch_dtype="auto", device_map="auto")
7
8msgs = [{"role": "user", "content":
9 "Write an AIME-style problem. Difficulty: late (problems 11-15). Topic: number theory."}]
10text = tok.apply_chat_template(msgs, tokenize=False, add_generation_prompt=True)
11out = model.generate(**tok(text, return_tensors="pt").to(model.device),
12 max_new_tokens=1600, do_sample=True, temperature=0.8, top_p=0.95)
13print(tok.decode(out[0], skip_special_tokens=True))<problem>…</problem><solution>…</solution><answer>N</answer> (integer 0–999).| Metric | Base (engineered) | Tuned (bare) | Δ |
|---|---|---|---|
| Format adherence | 28.9% | 63.9% | +35.0 |
| Self-duplication (lower=better) | 70.6% | 18.3% | −52.3 |
| Band accuracy | 60.0% | 64.3% | +4.3 |
| Novelty vs corpus+train | 87.8% | 71.1% | −16.7 |
| Validity (solver consensus) | 47.2% | 12.2% | −35.0 |
-lora-v3 adapters (matches training); this merged 16-bit model is for convenience.report_v3_analysis.md / BRAINLIFT_RESULTS.md in the project repo.