技術指南

Lookahead 和 Lion 最佳化器

Lookahead 和 Lion 是神經網路最佳化的兩個現代轉折。

概述

Lookahead 和 Lion 是神經網路最佳化的兩個現代轉折。 Lookahead 用「慢」和「快」權重包裝任何基礎優化器,以獲得更穩定的進展,而 Lion(進化符號動量)是由 AI 程式搜尋發現的,並僅使用動量項的符號更新權重 - 使其內存輕且通常比 Adam 更快。

Lookahead 和 Lion 優化器是一個技術構建塊,會大規模影響模型品質、基礎設施成本、延遲和可靠性。

深入探討

張、Hinton 及其同事在 2019 年提出的 Lookahead,運行一個標準的「快速」優化器(如 Adam 或 SGD)k 個步驟,然後將一組單獨的「慢速」權重推向快速權重最終的位置的一小部分。這可以抑制振盪並降低對超參數的敏感度。 Lion 由 Google 於 2023 年發布,源自於優化器演算法的符號程式搜尋。它追蹤動量,但將符號函數應用於更新,因此每個參數都在累積梯度符號的方向上移動固定的步長。 Lion僅儲存動量緩衝區(Adam狀態的一半,保留兩個),使用較大的權重衰減和較小的學習率,並且在大型視覺和語言模型上匹配或擊敗了Adam,同時訓練速度更快、成本更低。

技術洞察

前瞻更新:在 k 個快速步驟產生權重 θ_fast 後,慢速權重按 φ ← φ + α(θ_fast − φ) 移動,然後快速最佳化器重設為 φ。 Lion更新:用於插值的m ← β1·m + (1−β1)·g,但權重步長為θ ← θ − η·(sign(β2·m + (1−β2)·g) + λθ)。符號運算使每個座標的更新幅度一致,這就像隱式歸一化,並解釋了為什麼 Lion 需要比 Adam 小得多的學習率。

掌握 Lookahead 和 Lion 最佳化器

Lookahead 和 Lion 是神經網路最佳化的兩個現代轉折。 Lookahead 用「慢」和「快」權重包裝任何基礎優化器,以獲得更穩定的進展,而 Lion(進化符號動量)是由 AI 程式搜尋發現的,並僅使用動量項的符號更新權重 - 使其內存輕且通常比 Adam 更快。 Lookahead 和 Lion 優化器是一個技術構建塊,會大規模影響模型品質、基礎設施成本、延遲和可靠性。為了建立深入的理解,請將 Lookahead 和 Lion Optimizer 視為一種操作模型,而不是單一功能:定義所需的結果,澄清假設,並將系統可以可靠地執行的操作與仍需要專家判斷的操作分開。

在實踐中,強大的團隊使用 Lookahead 和 Lion Optimizer 根據可靠性和成本優化架構、資料和基礎設施選擇。他們記錄明確的成功標準,根據實際數據和工作流程進行測試,並根據觀察到的失敗模式而不是一次性基準測試勝利進行迭代。這就是理論理解轉變為跨產品、政策和營運的持久能力的地方。

多年來,架構決策決定著效能和營運成本。同時,優化一個基準測試可以隱藏更廣泛的系統弱點。最具彈性的方法是將實驗速度與治理規則結合:運行試點、捕獲證據、發布決策日誌,並隨著模型行為、使用者期望和監管要求的發展不斷更新保障措施。

戰略影響

多年來,架構決策決定著效能和營運成本。

多年來,架構決策決定著效能和營運成本。在高品質部署中,這會轉化為可衡量的操作規則、所有權邊界和定期審查儀式,以便團隊可以增強信心,而不是擴大模糊性。

技術教育幫助團隊選擇正確的堆疊,而不僅僅是最新的堆疊。

技術教育幫助團隊選擇正確的堆疊,而不僅僅是最新的堆疊。在高品質部署中,這會轉化為可衡量的操作規則、所有權邊界和定期審查儀式,以便團隊可以增強信心,而不是擴大模糊性。

更好的工程選擇可以減少生產中的可靠性事故。

更好的工程選擇可以減少生產中的可靠性事故。在高品質部署中,這會轉化為可衡量的操作規則、所有權邊界和定期審查儀式,以便團隊可以增強信心,而不是擴大模糊性。

Lookahead 和 Lion 最佳化器的未來

Lion 已在多次大規模訓練中被採用,因為它可以減少優化器記憶體並加快收斂速度,而且它的發現展示了自動化的「AI-designing-AI」演算法搜尋是實際收益的真正來源。隨著模型大小不斷給 GPU 記憶體預算帶來壓力,預計會有更多基於搜尋的優化器、將 Lookahead 式慢速權重與基於符號的更新相結合的混合方案,以及對記憶體高效優化器的興趣日益增長。

現實世界的實施

用 Lookahead 包裹 Adam,以穩定 Transformer 的訓練並減少超參數調整工作。

使用 Lion 訓練大型視覺模型(例如 ViT),優化器記憶體比 Adam 低。

使用 Lion 預訓練語言模型,以降低運算成本達到相當的準確性。

在強化學習代理中將 Lookahead 與 SGD 結合,以平滑吵雜的策略更新。

實施模式

Lookahead 和 Lion 最佳化器的實踐

用 Lookahead 包裹 Adam,以穩定 Transformer 的訓練並減少超參數調整工作。

用 Lookahead 包裹 Adam,以穩定 Transformer 的訓練並減少超參數調整工作 當團隊預先定義質量閾值、為邊緣情況保留人工升級路徑並隨著時間的推移跟踪生產力增益和錯誤成本時,通常會獲得更好的結果。

Lookahead 和 Lion 最佳化器的實踐

使用 Lion 訓練大型視覺模型(例如 ViT),優化器記憶體比 Adam 低。

使用 Lion 來訓練大型視覺模型(例如 ViT),其優化器記憶體低於 Adam 團隊,當他們預先定義品質閾值、為邊緣情況保留人工升級路徑並隨著時間的推移追蹤生產力增益和錯誤成本時,通常會獲得更好的結果。

Lookahead 和 Lion 最佳化器的實踐

使用 Lion 預訓練語言模型,以降低運算成本達到相當的準確性。

使用 Lion 預訓練語言模型,以降低計算成本實現相當的準確性 團隊在預先定義質量閾值、為邊緣情況保留人工升級路徑並跟踪一段時間內的生產力提升和錯誤成本時,通常會獲得更好的結果。

Lookahead 和 Lion 最佳化器的實踐

在強化學習代理中將 Lookahead 與 SGD 結合,以平滑吵雜的策略更新。

在強化學習代理中將 Lookahead 與 SGD 相結合,以平滑嘈雜的策略更新 當團隊預先定義質量閾值、為邊緣情況保留人工升級路徑並隨著時間的推移跟踪生產力增益和錯誤成本時,通常會獲得更好的結果。

風險與防護欄

!

優化一項基準測試可以隱藏更廣泛的系統弱點。

!

基礎設施和維護成本常常被低估。

!

隨著系統變得更加複雜,安全性和可觀察性差距可能會擴大。

實施路線圖

1

在實施之前定義延遲、品質和成本目標。

在實施之前定義延遲、品質和成本目標。將每個步驟視為證據門:如果不符合標準,則暫停推出,縮小差距,然後再擴大使用。

2

在實際負載和資料條件下進行基準測試。

在實際負載和資料條件下進行基準測試。將每個步驟視為證據門:如果不符合標準,則暫停推出,縮小差距,然後再擴大使用。

3

儀器監控錯誤、漂移和使用者影響。

儀器監控錯誤、漂移和使用者影響。將每個步驟視為證據門:如果不符合標準,則暫停推出,縮小差距,然後再擴大使用。

4

在擴展之前準備回滾和事件回應路徑。

在擴展之前準備回滾和事件回應路徑。將每個步驟視為證據門:如果不符合標準,則暫停推出,縮小差距,然後再擴大使用。

不斷探索