冻结多token预测加速Pixel上的Gemini Nano模型
Accelerating Gemini Nano models on Pixel with frozen Multi-Token Prediction
AI 导读Google Research提出一种新架构,在已冻结的Gemini Nano v3模型上改造Multi-Token Prediction(MTP),以加速Pixel 9和10系列上的设备端推理。该方法基于EAGLE框架和CALM,无需单独训练占用内存的草稿模型,通过“晚期退出”策略实现加速。AI通知摘要和校对功能因此生成文本速度显著提升、能耗降低,开发者无需为每个新任务微调独立模型。
来源:Google Research:Blog · research.google