slm-125m-base
A 126M-parameter Llama-architecture base model pretrained from
scratch on US case law, SEC filings, and educational web text. Trained on
8xH100 for ~28 minutes at a total project cost of roughly $18.
This is a base model. It has had no instruction tuning, no RLHF, and no
safety alignment. It completes text; it does not follow instructions.
Architecture
| |
|---|
| Parameters | 125,847,552 |
| Layers / hidden / heads | 12 / 768 / 12 |
| Context length | 1024 |
| Vocab | 16384 (byte-level BPE trained on this corpus) |
| Tied embeddings | True |
Training data
1.92B tokens, ~76% legal:
| Source | Tokens | Share |
|---|
PleIAs/SEC | 819M | 43% |
HFforLegal/case-law | 639M | 33% |
HuggingFaceFW/fineweb-edu | 463M | 24% |
Pipeline: deterministic rule-based cleaning (line filters, boilerplate
stripping, repetition and language gates, plus a dictionary-based OCR gate on
case law) then MinHash near-dedup, exact dedup, and 13-gram decontamination
against the CaseHOLD evaluation set.
Training
2 epochs (3.84B tokens seen), AdamW, cosine schedule to a 6e-05 cooldown,
bf16, sequence length 1024, global batch 524,288 tokens.
Results
| Metric | Value |
|---|
| Val perplexity (held-out 1%) | 9.07 |
| CaseHOLD (zero-shot, 5-way) | 19.5% |
CaseHOLD is measured on data excluded from training by 13-gram
decontamination, so it is not inflated by leakage. Expect near-chance
zero-shot performance: a 125M base model with no task tuning is not
competitive on multiple-choice legal reasoning, and this number is a baseline
for fine-tuning, not a capability claim.
Limitations
Small, base-only, English-only, 1024-token context. Case-law inputs contain
OCR noise that survives filtering. Not legal or financial advice and not
suitable for any use where an incorrect statement of law carries consequences.