Downloads · 30 days
7
23% of all-time downloads
Lynnix0217/cs2881-hw1-sft-sheldon-B
cs2881-hw1-sft-sheldon-B is a machine learning model from Lynnix0217. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
Full-parameter SFT of Qwen/Qwen2.5-3B-Instruct on Sheldon Cooper dialogue only. This is arm B of the CS2881 HW1 arm grid, and it is the intended starting point for stage-2 RL: it has the strongest persona of any arm w…
Downloads · 30 days
7
23% of all-time downloads
All-time downloads
31
Public
Parameters
3.1B
6.2 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors6.2 GB · 100%
From the Hugging Face model README
Full-parameter SFT of Qwen/Qwen2.5-3B-Instruct on Sheldon Cooper dialogue only. This is
arm B of the CS2881 HW1 arm grid, and it is the intended starting point for stage-2 RL:
it has the strongest persona of any arm we trained, and it has never seen STEM training data,
so a later RL stage on a STEM objective starts from an uncontaminated persona.
Code, evaluation artifacts and the full experiment worklog: https://github.com/harvard-cs2881f26/hw1-chadha-xu-zou
| base | Qwen/Qwen2.5-3B-Instruct |
| data | 1195 Sheldon Cooper replies from RoleBench instruction-generalization/general (one reply per question, always generated[0]) |
| objective | full-parameter SFT, loss on completion tokens only |
| steps | 225 (effective batch 16 = 4 x 4), about 3.01 epochs |
| lr | 1e-5, cosine, 5% warmup |
| precision | bf16, adamw_bnb_8bit, gradient checkpointing |
| max_seq_len | 1024 (0 rows dropped, 0 truncated) |
| seed | 42 |
No persona system prompt is used at training or evaluation time — the persona is in the weights, not in a prompt.
Persona is measured on 299 held-out RoleBench instructions. The primary metric is
distinctive register marker density (hits per 100 generated tokens over the 30 n-grams a
log-odds analysis assigns to Sheldon with precision >= 0.20). clf_raw and clf_style are two
TF-IDF + LogisticRegression authorship classifiers; style has the high-precision surface
markers deleted from its input, so the gap between them quantifies catchphrase gaming.
| base (untrained) | this checkpoint (B) | |
|---|---|---|
| register marker density | 0.110 | 1.573 |
clf_raw | 0.310 | 0.729 |
clf_style | 0.440 | 0.587 |
| catchphrase density | 0.00 | 2.63 |
| MATH accuracy (1000 problems, Math-Verify) | 0.703 | 0.606 |
This checkpoint never saw a single math sample, yet MATH accuracy still falls from 0.703 to 0.606. So roughly 0.097 of the drop is the cost of generic SFT and forgetting, independent of any capability data. Note also that the base model's 0.703 is itself understated: 100% of its parse failures come from hitting the 1024-token generation cap, and with a larger budget its parse rate rises from 0.892 to 0.997, putting its true ceiling near 0.81.
Output length also drops from 569 to 392 tokens on math, even though training only ever showed 44-token Sheldon replies — the length effect spills across domains.
bazinga accounts for 170 of this checkpoint's 172 surface-marker hits. A single raw
authorship classifier can be saturated by that one word (appending "Bazinga!" to arbitrary
neutral text moves clf_raw from 0.015 to 1.000), so do not use a raw persona classifier as
an RL reward on its own — the optimum under that reward is to spam catchphrases. Use the
marker density and the raw - style gap alongside it, and keep a held-out instrument that was
never optimized against.
| checkpoint | what it is |
|---|---|
| cs2881-hw1-sft-math-E | arm E with STEM = MATH: this checkpoint, then continued on a Sheldon + MATH mixture |
| cs2881-hw1-sft-openbookqa-E | arm E with STEM = OpenBookQA, same recipe |
Both E checkpoints continue from this one, so B is also the shared stage-1 ancestor of the two-stage arms.