k-l-lambda commited on
Commit
32306db
·
verified ·
1 Parent(s): bab7ecd

Fix README (was empty): real accept_len + distribution-shift note

Browse files
Files changed (1) hide show
  1. README.md +64 -0
README.md CHANGED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ base_model: moonshotai/Kimi-K2.6
4
+ tags:
5
+ - text-generation
6
+ - speculative-decoding
7
+ - eagle3
8
+ - kimi-k2.6
9
+ - mla
10
+ - torchspec
11
+ ---
12
+
13
+ # kimi-k2.6-eagle3-mla
14
+
15
+ Eagle3 MTP draft model with MLA (Multi-Latent Attention) for accelerating
16
+ inference of [Kimi-K2.6](https://huggingface.co/moonshotai/Kimi-K2.6).
17
+
18
+ This is a fine-tuned draft, anchored to the official
19
+ [lightseekorg/kimi-k2.6-eagle3-mla](https://huggingface.co/lightseekorg/kimi-k2.6-eagle3-mla)
20
+ initialization. It targets multi-hop (downstream-position) acceptance while
21
+ preserving the first-hop gain, evaluated by runtime accept-length on a frozen
22
+ full-context held-out set.
23
+
24
+ ## Fine-tune setup
25
+
26
+ - **Init**: lightseekorg/kimi-k2.6-eagle3-mla (official MLA weights)
27
+ - **Objective**: Eagle3 distillation + multi-step TTT supervision
28
+ (`ttt_steps=4`, `ttt_step_loss_decay=1.0`, off-policy downstream tokens)
29
+ - **Anti-over-specialization**: L2-SP weight-space anchor toward the init
30
+ (penalize trainable-param drift; lambda=1e-4)
31
+ - **Optimizer**: lr 2e-5, cosine schedule
32
+ - **Checkpoint**: best by held-out validation loss on the K2.6 ruler
33
+ (step 95400; val_loss 5.490, the global minimum of the v3 run)
34
+
35
+ ## Performance
36
+
37
+ Primary metric is **accept_length** — average tokens accepted per speculation
38
+ step with `num_speculative_tokens=3` (higher is better). Evaluated with
39
+ vLLM 0.20.0 on 8x H200, TP=8, max-model-len 32768, greedy.
40
+
41
+ On a frozen K2.6 full-context held-out judge set (914 prompts):
42
+
43
+ | Model | accept_len |
44
+ |-------|-----------:|
45
+ | lightseek (official init) | 2.285 |
46
+ | this model | **2.308** |
47
+
48
+ This draft improves over the official init on the K2.6 held-out distribution.
49
+
50
+ ## Note on distribution shift
51
+
52
+ This checkpoint is selected by validation loss on the K2.6 teacher
53
+ distribution. In cross-version testing against real Kimi-K2.7-Code production
54
+ traffic, the official lightseek init currently shows higher accept-length than
55
+ this fine-tune — i.e. the K2.6 fine-tune over-specializes to its training
56
+ distribution. If your serving traffic differs substantially from long
57
+ multi-turn K2.6 dialogues, benchmark both this draft and the lightseek init on
58
+ your own traffic before choosing. (The L2-SP anchor above is intended to
59
+ mitigate this; tuning it against real-traffic accept-length is ongoing.)
60
+
61
+ ## Usage
62
+
63
+ Serve with vLLM as the speculative draft for Kimi-K2.6, with
64
+ `num_speculative_tokens=3` in the speculative-config.