112 lines
5.4 KiB
Markdown
112 lines
5.4 KiB
Markdown
|
---
|
|||
|
|
license: apache-2.0
|
|||
|
|
base_model: Qwen/Qwen2.5-0.5B-Instruct
|
|||
|
|
library_name: transformers
|
|||
|
|
pipeline_tag: text-generation
|
|||
|
|
language:
|
|||
|
|
- en
|
|||
|
|
tags:
|
|||
|
|
- proofkit
|
|||
|
|
- fine-tuned
|
|||
|
|
- sft
|
|||
|
|
- lora
|
|||
|
|
- build-small-hackathon
|
|||
|
|
- work-sample
|
|||
|
|
datasets:
|
|||
|
|
- visproj/proofkit-sft
|
|||
|
|
---
|
|||
|
|
|
|||
|
|
# ProofKit Qwen 0.5B — fine-tuned (direct SFT)
|
|||
|
|
|
|||
|
|
`Qwen/Qwen2.5-0.5B-Instruct` fine-tuned **directly** on the ProofKit SFT set
|
|||
|
|
([`visproj/proofkit-sft`](https://huggingface.co/datasets/visproj/proofkit-sft),
|
|||
|
|
~7,000 synthetic examples). LoRA-trained, then merged to standalone weights. This is
|
|||
|
|
the **in-Space Transformers** option in [ProofKit](https://huggingface.co/spaces/visproj/proofkit) (runs on ZeroGPU / a small
|
|||
|
|
GPU; loads lazily on first generation).
|
|||
|
|
|
|||
|
|
It learns ProofKit's task contracts — section drafting, co-author drafting from rough
|
|||
|
|
user answers, revision actions, and strict-JSON scenario / recommendation / readiness
|
|||
|
|
/ portfolio generation.
|
|||
|
|
|
|||
|
|
## Usage
|
|||
|
|
|
|||
|
|
```python
|
|||
|
|
from transformers import AutoModelForCausalLM, AutoTokenizer
|
|||
|
|
import torch
|
|||
|
|
|
|||
|
|
m = "visproj/proofkit-qwen0.5b-7k"
|
|||
|
|
tok = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-0.5B-Instruct") # carries the chat template
|
|||
|
|
model = AutoModelForCausalLM.from_pretrained(m, torch_dtype=torch.float16).to("cuda").eval()
|
|||
|
|
|
|||
|
|
# NOTE: prompt it with ProofKit's trained prompt_formats.py shapes — see below.
|
|||
|
|
messages = [{"role": "system", "content": SYSTEM}, {"role": "user", "content": PROMPT}]
|
|||
|
|
text = tok.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
|
|||
|
|
ids = tok(text, return_tensors="pt").to("cuda")
|
|||
|
|
out = model.generate(**ids, max_new_tokens=600, do_sample=True, temperature=0.3, top_p=0.9)
|
|||
|
|
print(tok.decode(out[0][ids["input_ids"].shape[-1]:], skip_special_tokens=True))
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
## Evaluation (post-fix, 3-judge panel)
|
|||
|
|
|
|||
|
|
Mean score (0–100) on 15 held-out prompts, graded by Claude Opus 4.7, GPT-5.5, and a
|
|||
|
|
local Qwen-3B (`gpt-oss experts` is a deliberately *un-retrained* stale control):
|
|||
|
|
|
|||
|
|
| model | Claude | GPT-5.5 | Qwen-3B | **Avg** |
|
|||
|
|
|---|---:|---:|---:|---:|
|
|||
|
|
| gpt-5.5 (frontier ceiling) | 94.6 | 95.6 | 90.8 | **93.7** |
|
|||
|
|
| gpt-oss attn (retrained teacher) | 82.0 | 66.8 | 81.4 | **76.7** |
|
|||
|
|
| **qwen-0.5b distilled (served)** | 79.0 | 68.6 | 82.2 | **76.6** |
|
|||
|
|
| **qwen-0.5b direct 7k (served)** | 78.6 | 64.4 | 82.0 | **75.0** |
|
|||
|
|
| gpt-oss experts *(stale control)* | 67.6 | 68.6 | 81.8 | **72.7** |
|
|||
|
|
| qwen-3b base | 62.1 | 67.1 | 80.5 | **69.9** |
|
|||
|
|
| gpt-oss base | 55.4 | 53.8 | 68.2 | **59.1** |
|
|||
|
|
| qwen-0.5b base | 36.5 | 44.5 | 67.9 | **49.7** |
|
|||
|
|
|
|||
|
|
Both served retrained 0.5Bs beat the stale control and every untuned base across all
|
|||
|
|
three judges, and the distilled 0.5B ≈ ties its own 20B teacher.
|
|||
|
|
|
|||
|
|
## Limitations
|
|||
|
|
|
|||
|
|
- **0.5B capacity.** Reliably carries trained-style specifics, but can miss a truly
|
|||
|
|
arbitrary novel token and occasionally garbles. ProofKit adds a runtime fallback to a
|
|||
|
|
hosted instruct baseline when a draft drops the user's answers.
|
|||
|
|
- **Prompt-format-frozen** (see below) — not a general chat model.
|
|||
|
|
|
|||
|
|
## About ProofKit
|
|||
|
|
|
|||
|
|
[ProofKit](https://huggingface.co/spaces/visproj/proofkit) is a work-sample generator for job seekers — it turns a target
|
|||
|
|
role, background, and skills-to-prove into a realistic, **clearly-fictional**
|
|||
|
|
practice work sample (a role-specific challenge, a guided builder, a readiness
|
|||
|
|
review, and a recruiter-ready portfolio packet). Built for the Hugging Face **Build
|
|||
|
|
Small Hackathon** (Backyard AI track). Integrity rules are load-bearing: outputs
|
|||
|
|
never claim real employment, metrics are labeled hypothetical, and exports carry an
|
|||
|
|
ethical disclosure.
|
|||
|
|
|
|||
|
|
### The ProofKit model family
|
|||
|
|
|
|||
|
|
| Repo | What it is |
|
|||
|
|
|---|---|
|
|||
|
|
| [`visproj/proofkit-qwen0.5b-7k`](https://huggingface.co/visproj/proofkit-qwen0.5b-7k) | Qwen2.5-0.5B fine-tuned directly on the 7k set (Transformers) |
|
|||
|
|
| [`visproj/proofkit-gpt-oss-20b-lora`](https://huggingface.co/visproj/proofkit-gpt-oss-20b-lora) | gpt-oss-20b LoRA — the distillation **teacher** |
|
|||
|
|
| [`visproj/proofkit-distilled-qwen0.5b`](https://huggingface.co/visproj/proofkit-distilled-qwen0.5b) | Qwen2.5-0.5B distilled from the teacher (merged) |
|
|||
|
|
| [`visproj/proofkit-distilled-qwen0.5b-gguf`](https://huggingface.co/visproj/proofkit-distilled-qwen0.5b-gguf) | GGUF of the distilled student (llama.cpp — **served**) |
|
|||
|
|
| [`visproj/proofkit-sft`](https://huggingface.co/datasets/visproj/proofkit-sft) | SFT dataset (synthetic, license-safe) |
|
|||
|
|
| [`visproj/proofkit-distill-qwen0.5b`](https://huggingface.co/datasets/visproj/proofkit-distill-qwen0.5b) | Distillation dataset (teacher completions) |
|
|||
|
|
|
|||
|
|
### A note on training data (the "static responses" fix)
|
|||
|
|
|
|||
|
|
An earlier version of these models produced repetitive, input-ignoring drafts. The
|
|||
|
|
root cause was **synthetic-data leakage**: the dataset rendered the example *user
|
|||
|
|
answers* and the *target* from the same template slots, so the model learned
|
|||
|
|
`target = template` instead of `target = f(input)`. The fix — **faithfulness anchors**
|
|||
|
|
(a distinctive token shared by the answer and the target) + **seeded per-example
|
|||
|
|
variation** across every task, then a full-chain retrain — is what these current
|
|||
|
|
weights reflect.
|
|||
|
|
|
|||
|
|
### Prompt format is a frozen contract
|
|||
|
|
|
|||
|
|
These 0.5B models were trained on the **exact** prompt shapes from ProofKit's
|
|||
|
|
`prompt_formats.py`. They only behave well when prompted in that format; reworded or
|
|||
|
|
free-form prompts push them off-distribution. They are purpose-built components of the
|
|||
|
|
ProofKit app, not general chat models.
|