第110章 聯邦

投票推薦 加入書籤 小說報錯

  兩周後,沈一鳴敲開了左城的辦公室門。

  」左總,聯邦學習框架的工程化做完了。」沈一鳴的語氣裡帶著一絲不易察覺的興奮,」馬昊幫我在模型聚合算法上做了優化,方澤的團隊把寒武紀的工程樣片跑通了,三線並行測試都通過了。」

  左城放下手裡的文件:」可以上真機了?」

  」可以。但只能在少量節點上先試跑,大規模部署還需要更多驗證。」

  」那就今天試。」左城站起來,」叫上陳浩和方澤,去智慧城市的控制中心。」

  半小時後,左城一行人站在杭城智慧城市控制中心的大屏前。

  屏幕上顯示著全市傳感器的實時數據,一千五百個邊緣網關,五千個傳感器節點,分布在六個區的道路、橋樑、河道和建築上。數據流像細密的光線,不斷匯入中央平台。

  」選一個區的網關做測試。」左城說,」西湖區,一百二十個網關。」

  沈一鳴在筆記本電腦上操作起來。他把聯邦學習的訓練任務分發到西湖區的邊緣網關上,每個網關只使用本地的傳感器數據訓練一個小模型,然後把梯度信息加密後上傳到聚合伺服器,由聚合伺服器把所有網關的梯度匯總,更新全局模型。

  」開始。」左城說。

  屏幕上的數據開始流動。一百二十個邊緣網關同時開始本地訓練,每完成一輪,梯度信息就加密上傳。聚合伺服器的進度條從零開始緩慢爬升。

  」第一輪聚合完成,模型精度百分之七十八。」沈一鳴盯著數據,」第二輪開始。」

  左城站在後面,悄悄打開了系統面板。科技雷達的冷卻已經結束,沈一鳴就在三米之外。

  掃描。

  面板跳出了結果。和兩周前相比,沈一鳴掌握的技術列表多了一項:梯度稀疏化壓縮。這是他在工程化過程中自己研究出來的新能力,可以把梯度信息中的冗餘數據剔除,只保留關鍵參數,從而大幅降低通信開銷。

  左城看著這項技術,心裡迅速計算。如果複製梯度稀疏化壓縮,配合沈一鳴已有的自適應通信調度,聯邦學習的通信效率還能再提升一個台階。

  【複製」梯度稀疏化壓縮」需要消耗8積分,是否繼續?】

  確認。

  積分從二百八十五分變成了二百七十七分。一項新的技術出現在左城的科技樹葉片列表中,歸入萬物互聯枝幹。

  左城關掉面板,繼續看測試。

  」第三輪聚合完成,模型精度百分之八十五。」沈一鳴的聲音有些發顫,」數據沒有出過本地,隱私保護完全有效。第四輪。」

  」通信開銷呢?」陳浩問。

  」比集中式訓練降低了百分之五十二。」沈一鳴說,」這個數據已經超出了我的預期。樹莓派上測試的時候只能做到百分之三十七的降幅,真機環境反而更好,因為邊緣網關的算力比樹莓派強很多,本地訓練的梯度質量更高,聚合的效率也就更好。」

  」精度損失呢?」方澤開口了。

  」幾乎沒有。」沈一鳴調出一個對比圖,」聯邦學習訓練出來的模型精度是百分之八十五,集中式訓練的精度是百分之八十七,差距在誤差範圍內。而且隨著訓練輪次增加,這個差距還在縮小。」

  左城心中一動。梯度稀疏化壓縮,他已經從系統里複製了,但這項技術的載體不是沈一鳴,而是他自己的科技樹。他需要把這項技術」反饋」給沈一鳴,但不能暴露來源。

  」一鳴,你之前提到的梯度稀疏化思路,有沒有繼續往下做?」左城問。

  沈一鳴愣了一下:」有初步的想法,但還沒來得及實現。」

  」我之前看過一篇論文,講的是梯度稀疏化的自適應閾值選擇。核心思路是在訓練過程中動態調整稀疏化率,前期保留更多梯度保證收斂,後期提高稀疏率降低通信。你試試這個方向。」

  沈一鳴眼睛一亮:」自適應閾值選擇?這個思路很好,和我之前的自適應壓縮率可以對接。我回去就做。」

  測試繼續進行。第八輪聚合完成後,模型精度穩定在百分之九十三,和集中式訓練的精度幾乎持平。通信開銷最終穩定在百分之五十八的降低幅度。

  」測試成功。」沈一鳴長出一口氣,雙手微微顫抖,」聯邦學習框架在真實設備上跑通了。數據沒有出過本地,精度幾乎無損,通信開銷降了百分之五十八。」


  陳浩拍了拍他的肩膀:」幹得不錯,回去請團隊吃頓好的。」

  方澤難得露出一個微小的笑容,然後繼續低頭看功耗數據,嘴裡嘟囔了一句:」峰值功耗比預估低了百分之十二,寒武紀的晶片還行。」

  左城站在大屏前,看著那些從邊緣網關匯聚而來的數據流。聯邦學習,模型壓縮,寒武紀晶片,三條技術線正在交匯。技術增幅在後台默默工作著,所有融合方案的實際效率都被放大了百分之二十。

  但他沒有向任何人提起這一點。

  正想著,手機響了。

  周鶴年的名字出現在屏幕上。

  」左城,有個事要跟你說。」周鶴年的聲音一如既往地直接,」天穹四期的技術方案評審要提前了,藍灣那邊希望把AI能力納入四期的核心指標。你們402的AI團隊準備得怎麼樣?」

  左城握緊手機,心跳加速了。

  」周總,402的AI能力,隨時可以上。」

  」好。下周來藍灣,做一個技術演示。」

  電話掛斷後,左城看著窗外的天色。聯邦學習框架剛剛完成首次實戰,藍灣的天穹四期就來了。時機恰好。

  但天穹四期的AI需求遠比三期的頻譜調度複雜。左城需要更強的技術儲備。

  他低頭看了看系統面板。積分二百七十七分,科技樹四根枝幹三十九枚葉片,冷卻中的科技雷達還要再等。

  夠用嗎?

  左城不知道。但他知道一件事,402的AI,已經不是一個空殼了。

章節目錄