jhyuckkim commited on
Commit
c114e3c
·
verified ·
1 Parent(s): 856e88c

Align naming and results with the published arXiv version

Browse files
Files changed (1) hide show
  1. README.md +12 -12
README.md CHANGED
@@ -15,7 +15,7 @@ tags:
15
  > **Research artifact, not a general-purpose model.**
16
  > These weights are released so that the experiments in the paper *Pruning and
17
  > Distilling Mixture-of-Experts into Dense Language Models* can be reproduced.
18
- > Each student is distilled on a fixed, small token budget (3.8B tokens for
19
  > Qwen3, 0.3B tokens for DeepSeek-V2-Lite and GPT-OSS) purely so that expert
20
  > scoring and grouping methods can be compared under an equal budget. Absolute
21
  > quality is therefore far below the teacher and below pretrained models of the
@@ -33,19 +33,19 @@ Dense-to-dense (D2D) pruning baseline. Unlike the other students in this collect
33
 
34
  ## Results
35
 
36
- Full comparison group for this architecture, Qwen3-30B-A3B to dense, 3.8B-token distillation (paper Table 5). This model's row is in
37
  bold, and rows whose weights are also released link to them.
38
 
39
- | Configuration | PPL | Wino | Hella | ARC-E | ARC-C | MMLU | Avg |
40
- |---|---|---|---|---|---|---|---|
41
- | [DO-AWP, K=8 (ours)](https://huggingface.co/jhyuckkim/Qwen3-30B-A3B-Dense-3B-DO-AWP-K8) | 12.26 | 63.1 | 60.3 | 75.6 | 45.4 | 46.1 | 58.10 |
42
- | [SF x WC, K=16 (prior work)](https://huggingface.co/jhyuckkim/Qwen3-30B-A3B-Dense-3B-SF-WC-K16) | 12.99 | 61.2 | 56.3 | 74.0 | 43.1 | 32.7 | 53.46 |
43
- | **D2D pruning (Qwen3-32B to 3.4B)** | **12.73** | **60.5** | **57.5** | **73.1** | **41.5** | **26.6** | **51.84** |
44
- | Random MLP + teacher attention | 16.57 | 54.4 | 45.4 | 66.0 | 34.2 | 27.1 | 45.44 |
45
- | Qwen3-1.7B (pretrained reference) | 9.41 | 66.1 | 67.1 | 81.9 | 55.5 | 62.6 | 66.63 |
46
- | Qwen3-4B (pretrained reference) | 7.92 | 72.0 | 75.8 | 86.2 | 64.6 | 73.1 | 74.34 |
47
 
48
- PPL is WikiText-2 perplexity after distillation. Downstream accuracy is Winogrande 5-shot, HellaSwag 10-shot, ARC-Easy 25-shot, ARC-Challenge 25-shot and MMLU 5-shot. Avg is the unweighted mean of the five benchmarks.
49
 
50
  ## Configuration
51
 
@@ -53,7 +53,7 @@ PPL is WikiText-2 perplexity after distillation. Downstream accuracy is Winogran
53
  |---|---|
54
  | Teacher | `Qwen/Qwen3-32B` |
55
  | Student parameters | 3.44B |
56
- | Distillation data | FineWeb-Edu (`sample-10BT`), 3.8B tokens |
57
 
58
  ## Usage
59
 
 
15
  > **Research artifact, not a general-purpose model.**
16
  > These weights are released so that the experiments in the paper *Pruning and
17
  > Distilling Mixture-of-Experts into Dense Language Models* can be reproduced.
18
+ > Each student is distilled on a fixed, small token budget (~4B tokens for
19
  > Qwen3, 0.3B tokens for DeepSeek-V2-Lite and GPT-OSS) purely so that expert
20
  > scoring and grouping methods can be compared under an equal budget. Absolute
21
  > quality is therefore far below the teacher and below pretrained models of the
 
33
 
34
  ## Results
35
 
36
+ Full comparison group for this architecture, extended training after ~4B tokens (paper Table 6). This model's row is in
37
  bold, and rows whose weights are also released link to them.
38
 
39
+ | Configuration | Wino | Hella | ARC-E | ARC-C | MMLU | Avg |
40
+ |---|---|---|---|---|---|---|
41
+ | [DO-ACP, K=8](https://huggingface.co/jhyuckkim/Qwen3-30B-A3B-Dense-3B-DO-ACP-K8) | 63.1 | 60.3 | 75.6 | 45.4 | 46.1 | 58.10 |
42
+ | [SF, K=16](https://huggingface.co/jhyuckkim/Qwen3-30B-A3B-Dense-3B-SF-K16) | 61.2 | 56.3 | 74.0 | 43.1 | 32.7 | 53.46 |
43
+ | **D2D pruning (Qwen3-32B to 3.4B)** | **60.5** | **57.5** | **73.1** | **41.5** | **26.6** | **51.84** |
44
+ | Random FFN + teacher attn | 54.4 | 45.4 | 66.0 | 34.2 | 27.1 | 45.44 |
45
+ | Qwen3-1.7B (pretrained reference) | 66.1 | 67.1 | 81.9 | 55.5 | 62.6 | 66.63 |
46
+ | Qwen3-4B (pretrained reference) | 72.0 | 75.8 | 86.2 | 64.6 | 73.1 | 74.34 |
47
 
48
+ Downstream accuracy is Winogrande 5-shot, HellaSwag 10-shot, ARC-Easy 25-shot, ARC-Challenge 25-shot and MMLU 5-shot. Avg is the unweighted mean of the five benchmarks.
49
 
50
  ## Configuration
51
 
 
53
  |---|---|
54
  | Teacher | `Qwen/Qwen3-32B` |
55
  | Student parameters | 3.44B |
56
+ | Distillation data | FineWeb-Edu (`sample-10BT`), ~4B tokens |
57
 
58
  ## Usage
59