Frontier / 06 · EMBEDDED / EDGE

端侧小模型与 TinyML 部署边界

当权重、KV、激活和运行时都要塞进 MB 级预算,哪种优化真正决定可用性?

端侧模型是否可用,要同时看权重、KV、激活、运行时和热预算;参数量只是第一层过滤器。

解释图一个端侧 token 的内存旅程
  1. 01Flash量化权重
  2. 02SRAM权重驻留
  3. 03Activation峰值工作区
  4. 04KV上下文增长
  5. 05MCU实时约束

权重、激活、KV 与运行时共同争用 SRAM,任何一个被忽略都可能让模型装不进去。

Research notes / 两篇结构

先建立模型,再用实验拆边界

01旗舰研究嵌入式经验7 分钟

端侧小模型:参数量只是第一道门

从 MobileLLM、Phi-3 与 AWQ 出发,拆解端侧模型的结构、量化、运行时与热预算。

端侧 AIMobileLLMPhi-3

Instrument

TinyML 预算

默认 4-bit 权重方案需要约 953 KiB 仅存放 2M 参数,已经超过 512 KiB SRAM;端侧模型必须同时优化权重、激活和 KV。

打开实验
参数量2 M模型权重规模
权重位宽4 bit4-bit / 8-bit 量化
激活缓冲96 KiB峰值激活占用
KV 缓冲64 KiB上下文缓存占用

Primary sources / 一手证据

可追溯的起点

  1. · 2024MobileLLM: Optimizing Sub-billion Parameter Language Models for On-Device Use CasesPAPER
  2. · 2024Phi-3 Technical ReportPAPER
  3. · 2023AWQ: Activation-aware Weight Quantization for LLM Compression and AccelerationPAPER

Cross-domain synthesis

把这个主题放回系统主线

01综合研究综合研究6 分钟

从 HBM 到 MCU:一条内存层级主线

把 KV Cache、FlashAttention、端侧模型与存算一体放进同一张数据移动账本,理解 AI 系统的真正成本。

内存层级KV CacheFlashAttention