KeyFrame內部研究專用

Why Large? Tiny LMs & Agents on Edge/Robotics — Cormac Brick, Google

AI Engineer·7月25日週六·21 min英文

三句話摘要

如何通過微型 AI 模型在邊界設備上部署智能功能,打破 DRAM 成本限制,讓更廣泛的消費設備和機器人獲得 AI 能力。 通過量化、微調和合成數據集,微型 AI 模型可以在消費級邊界設備上實現可靠的生產功能,解鎖被 DRAM 成本門檻排除在外的數百萬設備。 邊界 AI 的核心優勢:低延遲、隱私保護、離線可用、成本節省。尤其在大規模用戶交互場景中,即使便宜的雲端調用也會因流量×成本快速累積,使邊界部署在經濟上更優。

重點整理

重點
  • 1

    邊界 AI 的核心優勢:低延遲、隱私保護、離線可用、成本節省。尤其在大規模用戶交互場景中,即使便宜的雲端調用也會因流量×成本快速累積,使邊界部署在經濟上更優。

  • 2

    DRAM 成本成為主導制約:設備廠商正在降低 DRAM 配置,Raspberry Pi 16GB 版本價格上升 2.5 倍。這一瓶頸迫使工程師必須在量化和模型最小化上投入大量工作。

  • 3

    小型模型的能力與限制:1-4B 參數模型(如 Gemma)通過零樣本提示即可達成不錯性能,但需要 4GB+ RAM,限制其部署範圍;微型模型(50M-500M)雖然內存需求更低,卻需要精心微調才能達成可靠功能。

  • 4

    合成數據微調是實用方案:投入 10,000 至 10 百萬樣本的合成數據集進行微調,可使遠小得多的模型達成與大型模型相當或更好的效果,同時在更廣泛的設備上運行且響應速度更快。

實用技巧與重點

乾貨
  • 組織與工具:
  • Google AI Edge Team 開發的開源項目:Literity LM、Literity Media Pipe
  • Gemma 系列模型(特別是 Gemma 2B)
  • Function Gemma:為函數調用預先訓練的變體
  • Mobile Actions 數據集(Huggingface 開源)
  • AI Edge Gallery(iOS/Android 應用)
  • 性能數據:
  • Raspberry Pi 上 Gemma 2B:7.6 tokens/sec 解碼(無 MTP);開啟 MTP 可達 2 倍速度
  • Jetson Nano:24 tokens/sec 解碼
  • Qualcomm IoT 板:~4,000 tokens/sec 預填充,31 tokens/sec 解碼
  • 內存占用:Gemma 2B 權重 41MB;運行時 ~2GB 活躍 RAM;加上 OS 需 4GB+ 總 RAM
  • Apple FastVLM:0.5B 參數視覺模型,實時性能優異
  • Mobile Actions 模型:10 個函數輸出,86% 可靠度
  • 模型規模與參數:
  • 小型模型:1-4B 參數(如內置 Android 高端機、Apple Intelligence)
  • 微型模型:50M-500M 參數(如 Gemma tree 270M 基礎模型)
  • 微調數據集規模:10,000-10,000,000 樣本
  • 部署案例:
  • 無訂閱語音聽寫 App(iOS):雙層微調 Gemma 模型架構(ASR + 文本清理引擎)
  • OpenDoc Mini V2 機器人:使用 Jetson Nano 或 Raspberry Pi 實現視覺和語音交互
  • Chrome 內置 API:摘要、校對功能透過微型模型提供給更廣泛用戶

結論

結論

通過量化、微調和合成數據集,微型 AI 模型可以在消費級邊界設備上實現可靠的生產功能,解鎖被 DRAM 成本門檻排除在外的數百萬設備。

完整解析

詳細

邊界 AI 的核心吸引力在於低延遲、數據隱私和離線可靠性,但大型 AI 模型部署在邊界設備上面臨嚴峻挑戰。最迫切的制約來自 DRAM 成本:在消費級硬件上,DRAM 配置甚至在縮減,而通用方案則面臨記憶體牆。這迫使工程團隊必須既要通過激進的量化壓縮模型,也要尋求最小化模型本身。

演講者介紹了 Google 在這一領域的工作方向:首先是與 Gemma 團隊合作開發小型模型。1-4B 參數的 Gemma 模型在推理能力上可與一年前的 Gemma 3 大型模型相當,通過零樣本提示已能達成不錯效果。為了優化內存足跡,團隊使用了混合量化策略(2 位、4 位和 8 位)將權重降至平均 2.9 位,使 Gemma 2B 的內存占用僅為 41MB(不計運行時)。加上 KV 緩存和運行時開銷,生產部署通常需要 4GB+ RAM——這限制了其部署範圍。在速度方面,Raspberry Pi 上達到 7.6 tokens/sec,Jetson Nano 可達 24 tokens/sec,而 Qualcomm IoT 板在啟用硬件加速後可達到幾乎實時的性能(4,000 tokens/sec 預填充、31 tokens/sec 解碼),足以每秒處理 3 幀高分辨率圖像。

然而小型模型仍無法滿足所有需求。許多場景要求更低的內存占用或更快的響應,且模型往往只是整個應用中的一小部分功能。這促使開發微型模型,通常在 50M-500M 參數範圍內。對於固定任務如自動語音識別、視覺和嵌入,已有現成的優化模型可用(如 0.5B 參數的 Apple FastVLM)。但若需要通用推理或特定功能,關鍵策略是對基礎模型進行微調。團隊發現合成數據集微調是最實用的方案:投入 10,000 到 10 百萬樣本的合成數據,可使 270M 參數的 Function Gemma 達到 86% 的函數調用可靠度,涵蓋 10 種不同輸出函數。這一方法已在生產中驗證——iOS 上的無訂閱語音聽寫應用採用雙層微調架構(ASR 引擎 + 文本清理引擎),完全離線運行卻能達到高質量輸出,並支持去除言語填充詞和個人化偏向。

此外,Chrome 等平台已開始通過微型模型內置摘要和校對 API,使更廣泛的用戶能訪問這些功能。語音轉函數調用(voice-to-function calling)成為關鍵用例,因為低端設備通常缺乏友好的設置菜單,但通過微調小模型已可實現可靠的交互。整個方法論的核心是選擇合適的基礎模型、驗證性能和內存占用、投入時間生成高質量的合成數據集,最後通過微調達成既小又快又可靠的部署。

關鍵時刻

Pipeline v2

帶時間戳的重點,會在逐字稿層級分析上線後產生。目前請先透過原始影片觀看。

事實查核

Pipeline v2

說法查證是下一次管線升級的一部分。KeyFrame 只會顯示它真正能驗證的內容。

更多「AI 技術」的內容

🔬“We have foundation models for language, not for physics” — Anima Anandkumar, Bren Professor of Computing
編輯精選
83 min
AI 技術英文PODCAST8月26日

🔬“We have foundation models for language, not for physics” — Anima Anandkumar, Bren Professor of Computing

Latent Space

  • 加速傳統物理模擬的典範轉移:氣象科學家原本認為 AI 無法匹敵數十年的物理建模工作,但傅里葉神經算子不僅達到同等精度,還快了一萬多倍。原本需要超級計算機的計算現在用消費級 GPU 就能完成,這改變了整個領域的思維方式。
  • 傅里葉域的非局部現象捕捉:傅里葉域能有效表示非局部現象(如大氣河流跨越千里的影響),且計算複雜度為準線性,遠優於完全連接的全局模型。這特別適合流體動力學、量子化學等自然現象中普遍存在的非局部相互作用。
  • 多解析度連續函數表示:神經算子將輸入輸出視為連續函數而非固定維度向量,可在推論時以任意解析度查詢,並能在更高解析度上疊加物理約束或額外數據,克服了固定解析度神經網路的限制。
Between Two Nerds: Attribution is dead, long live attribution
32 min
AI 技術英文PODCAST8月25日

Between Two Nerds: Attribution is dead, long live attribution

Risky Business

  • 工具成本的破壞性下降 — 傳統上,攻擊者必須重複使用昂貴自製的惡意軟體或工具組,因為開發和維護成本極高。這種成本結構使得安全研究人員可以通過工具特徵和程式碼簽名來追蹤攻擊者。LLM 自動化了代碼生成與維護工作流,使得攻擊者可以輕易為每個目標生成新工具,或改用通用系統工具,導致傳統的工具特徵分析失效。
  • 所有取證證據都在攻擊者掌控之中 — 無論是使用的 IP 位址、惡意軟體類型或戰術流程,這些都是攻擊者的主動選擇。即使看似是隨機巧合,攻擊者仍有能力在事前決定留下什麼痕跡。因此,所有可恢復的取證證據本質上都是攻擊者願意暴露的信息。
  • LLM 削弱工具簽名但保留高階行為特徵 — LLM 經過公開駭客技術訓練,使不同使用者產生相似的攻擊模式。然而,勒索軟體集團、國家級行為者的受害者選擇、贖金要求方式或目標模式等高階特徵仍具有識別價值。例如,鎖定加密交易所的攻擊幾乎只能指向朝鮮。
Why the Next AI Breakthrough May Come from Physics with Max Welling - #774
55 min
AI 技術英文PODCAST8月25日

Why the Next AI Breakthrough May Come from Physics with Max Welling - #774

TWIML AI

  • 多層篩選的材料設計流程:先搜尋文獻資料庫找現有材料,若無合適的就用生成模型產生數十萬個候選分子,用機器學習力場進行分子動力學模擬篩選,再進行實驗驗證。這套流程相比傳統量子力學計算能加速效率數個數量級。
  • 生成AI與熱力學的數學等價性:資訊論是兩個領域的共同基礎,生成模型的擴散過程與非平衡統計力學描述資訊損失的過程在數學上完全對應,許多開發出來的方法工具在兩領域都有精確對應的形式。
  • 基礎模型的遷移學習策略:先在廣泛材料資料集上訓練基礎力場表示,再針對特定材料類別進行蒸餾微調,既能保持計算效率也能獲得專一性。