Rhett1 commited on
Commit
3507003
·
verified ·
1 Parent(s): 30ab747

Upload svd_llm2/heal/llama2-13b-base/README.md with huggingface_hub

Browse files
svd_llm2/heal/llama2-13b-base/README.md ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: llama2
3
+ base_model: NousResearch/Llama-2-13b-hf
4
+ tags:
5
+ - svd
6
+ - low-rank
7
+ - model-compression
8
+ - llama2
9
+ ---
10
+
11
+ # SVD-LLM V2 (NAACL 2025): LLaMA-2-13B base @ 7.74B, KD-healed
12
+
13
+ SVD-LLM V2: heterogeneous compression-ratio allocation from theoretical truncation loss (Algorithm 1) + loss-optimized weight truncation (Algorithm 2). Official code was never released; this is our faithful implementation of the published algorithms (per-layer ranks in ranks.json show the allocation).
14
+
15
+ - **7.740B params** (7,740,277,760; down from 13.02B, ~40.5% reduction)
16
+ - Healed with pure forward-KL distillation from the dense 13B base teacher: **60M tokens**,
17
+ held-out KL **1.5569 -> 0.1788**, probe MMLU (limit 40/subtask) **26.14% -> 45.48%**
18
+ - ⚠️ **Probe-only evaluation**: full-benchmark evaluation (full MMLU / HellaSwag / BoolQ /
19
+ TriviaQA / GSM8K) was intentionally NOT run for the base-model series. All numbers below
20
+ come from the same probe harness (lm-eval 0.4.12, MMLU limit=40 questions/subtask,
21
+ n=2280, seed 1234, fp16) and are mutually comparable.
22
+
23
+ | Reference (same probe harness) | probe MMLU |
24
+ |---|---:|
25
+ | unhealed 7.74B (same structure, unhealed sibling dir) | 26.14% |
26
+ | **this checkpoint (healed 60M)** | **45.48%** |
27
+ | SVD-LLM V1 base 13B (same budget + same healing recipe) | 43.33% |
28
+ | FP16 dense 13B base teacher | 54.25% |
29
+
30
+ ## How to load
31
+
32
+ ```python
33
+ import json
34
+ from lowrank_linear import build, materialize_all
35
+
36
+ model = build("NousResearch/Llama-2-13b-hf",
37
+ json.load(open("ranks.json")), "state_dict_fp16.pt",
38
+ device="cuda")
39
+ # for generation/eval: fuse the factors first (inference-only)
40
+ materialize_all(model)
41
+ ```
42
+
43
+ `ranks.json` maps each compressed projection path to its integer rank; untouched weights are
44
+ stock LLaMA-2-13B base tensors. The full healing-provenance record (data mixture & revisions,
45
+ decontamination, SHA256 of inputs/checkpoint, 5M milestone trajectory) is in `provenance.json`.
46
+
47
+ ## License
48
+
49
+ Derived from Meta's LLaMA 2 (`NousResearch/Llama-2-13b-hf`, an unofficial redistribution of
50
+ the same weights). Subject to the **LLaMA 2 Community License** — see
51
+ https://ai.meta.com/llama/license/.