跳至主要内容

新穎貢獻

本頁記錄 TinyLoop 的真正創新、對已知想法的權重共享適配,以及應歸功於先前研究之處。分為三類:

  • 新穎 — 據我們所知,無先前研究提出此方法
  • 新穎機制 — 概念已存在,但權重共享的實作是新的,且消除了原始方法的限制
  • Looped 專屬適配 — 概念存在於標準 transformer;我們適配以利用權重共享

新穎:確定性跨迭代誤差對消

內容。 當同一個 INT4 權重在 loop 迭代中被使用 L 次,我們交替 rounding 方向:偶數迭代用標準 round-to-nearest,奇數迭代套用 sign-bit 校正。量化誤差成對對消。

誤差比較:

方法誤差增長誰能做到
Naive INT4O(L)任何架構
Stochastic roundingO(√L)僅 looped
確定性對消O(1)僅 looped

為何新穎。 自適應 rounding 文獻豐富 — AdaRoundGPTQVQRound — 但都針對使用一次的權重做最佳化。無先前研究利用權重共享架構中同一權重被重複使用 L 次的特性,透過刻意交替讓誤差確定性對消。

API。 TINYLOOP_CANCEL_ROUND=1(需含 sign_bit 通道的 v5 .tinyloop 檔)。


新穎:每迭代安全檢查

內容。 decode 期間每個 loop 迭代後,runtime 計算 FP32 residual 的 L2 norm。若超過閾值,立即中止生成。每個 token 有 L 次機會捕捉爆炸或退化的 hidden state,標準 transformer 只有 1 次。

為何新穎。 Parcae 透過 spectral norm 約束分析訓練時穩定性。Parallel Loop Transformer 用 residual norm 做 layer-skipping 判斷。兩者都未部署runtime 中止機制在每次迭代監控 residual 並終止生成。

API。 model.generate(..., safety_check=True, safety_norm_threshold=50.0)


新穎機制:Store-h KV Cache

內容。 不存 K 和 V,改存 pre-QKV hidden state h,在 attention 時重建 K = W_k · hV = W_v · h。因為 W_k 和 W_v 跨迭代共享,不需額外參數。

相關研究。 DeepSeek MLA 將 KV cache 壓縮為 latent vector 並透過學習的 up-projection 矩陣重建。兩者都消除完整 KV 儲存。關鍵差異:MLA 需要訓練過的壓縮/解壓矩陣;Store-h 不需額外參數 — 重建使用模型已有的共享 QKV 權重。

成果。 INT4-h 模式下 KV VRAM 最多節省 78%,零額外參數。


新穎機制:Rewind Speculation

內容。 低深度 draft(L=2)被高深度 verify(L=8)拒絕時,檢查中間深度(L=4)是否會與 verify 一致。

相關研究。 HiSpec(2025 年 10 月)用 early-exit 模型做分層驗證 — 概念類似但需要不同的模型。TinyLoop 的 rewind 在每個深度用相同權重 — 無額外模型、無額外記憶體、無需訓練。

成果。 H100 / 2B INT4 上 88.9% rewind 命中率。


新穎機制:Per-Token L 排程

內容。 單次 decode 中逐 token 變化 loop 深度:前 N 個 token 用 L=2(便宜 warmup),後續 token 用 L=8。

相關研究。 LoopFormer(2026 年 2 月,ICLR 2026)引入 elastic-depth looped transformer。其限制明確指出:「全域(序列層級)而非 instance/token 自適應預算。」 TinyLoop 的 L 排程在單次 decode 內以 per-token 粒度控制深度,無需訓練。

成果。 H100 上 warmup token 快 ~3.3×。


新穎洞察:LoRA 放大效應

內容。 應用於 loop block 的 rank-r LoRA adapter 在前向傳播中貢獻 L 次,使 PEFT 的參數效率比標準 transformer 高 L 倍。

重要性。 在 24 層標準 transformer 中,一層上的 LoRA 影響 1/24 的計算。在 TinyLoop L=8 時,loop block 上的 LoRA 影響 100% 的 loop 計算。在我們的 2B 模型上:16,384 參數(0.03 MB)控制 8 次完整迭代的計算。

成果。 L=16 時,16K LoRA 參數控制的計算量是標準 transformer 一層相同參數的 16 倍。


Looped 專屬適配:Self-Speculative Decoding

內容。 用 L=2 做 draft model、L=8 做 verifier,相同權重。

先前研究。 Draft & Verify(Zhang et al., 2024)透過 layer skipping 引入 self-speculative decoding。概念非新。TinyLoop 的貢獻是機制:透過 loop count 旋鈕做深度變化,比 layer skipping 更乾淨。

成果。 H100 / 407M 上 2.77× 吞吐,零額外參數。


總結

功能新穎度關鍵差異
確定性誤差對消新穎無先前研究對重複使用的權重做交替 rounding
每迭代安全檢查新穎已部署的 runtime 中止,非訓練時分析
Store-h KV cache新穎機制類似 MLA 但不需額外投影矩陣
Rewind speculation新穎機制類似 HiSpec 但相同權重,無額外模型
Per-token L 排程新穎機制類似 LoopFormer 但 per-token,非 per-sequence
雙位元權重交換新穎無先前研究對共享權重做逐迭代位寬選擇
L-aware 批處理新穎無先前研究在同一 batch 中以共享權重做每請求不同深度
動態 L 推理新穎能力同一模型 L=2..64 — 每請求、每 token 深度
LoRA 放大效應新穎洞察Loop block 的 rank-r LoRA 參數效率為 L 倍
三位元權重交換新穎機制INT8/INT2/INT4 逐迭代位寬選擇
Self-speculative decoding適配類似 Draft & Verify 但透過深度旋鈕,非 layer skip