sartajbhuvaji/bonsai-distilled-depth0-experts64-500-warmup75-lr1e4-fixedloss-best

来源:🤗text-generationapache-2.016B 参数32 GBsafetensors✓ 2 个校验和3 天前更新
或提交种子

bonsai: depth0/experts64

Pruned and distilled from Qwen3-30B-A3B-Base for TypeScript code completion. Part of the bonsai project, which applies the Minitron structured-pruning and knowledge-distillation recipe (Muralidharan et al. 2024, Sreenivas et al. 2024) to a Mixture-of-Experts model, a pruning axis neither paper covers.

What was pruned

  • Depth: 0 of 48 layers removed (48 kept)
  • Experts: 64 of 128 routed experts removed per layer (64 kept)
  • Result: ~16.03B parameters (52.6% of baseline's ~30.5B)

Training

500-step fixed-loss short schedule (75-step warmup, peak LR 1e-4), distilled against the uncorrected teacher. Best-validation checkpoint, step 39.

Results

HumanEval-TS (TypeScript MultiPL-E), full 159-problem suite, temperature 0.2, top-p 0.95:

pass@1
Qwen3-30B-A3B-Base (unpruned) 0.717 (114/159)
This model 0.579 (92/159)

80.7% of baseline accuracy at 52.6% of baseline's parameters.

Real serving throughput, measured with vLLM and SGLang, 159 HumanEval-TS prompts, 512 fixed output tokens/request, concurrency 16, 1x B300:

vLLM (tok/s) SGLang (tok/s)
Qwen3-30B-A3B-Base (unpruned) 2545.6 2690.0
This model 2985.7 3253.6

Notes

Project's recommended candidate for a size/speed-constrained TypeScript deployment: 47% smaller than baseline, 80.7% accuracy retention, and real measured serving throughput (vLLM and SGLang) matching or beating the unpruned baseline's own throughput.

Project

Other checkpoints from this sweep: bonsai collection. Code and full writeup: github.com/SartajBhuvaji/bonsai (private repo).