網路 AIOpsNOTE

Async-HFL-全文繁體中文翻譯

Async-HFL:階層式 IoT 網路中高效率且穩健的非同步聯邦學習

原文:Async-HFL: Efficient and Robust Asynchronous Federated Learning in Hierarchical IoT Networks,arXiv:2301.06646v4,2023-04-10,IoTDI 2023。
來源:2301.06646v4.pdf;翻譯日期:2026-10-05。
翻譯範圍:18 頁,包含正文、演算法、表格、圖說、致謝、62 筆參考文獻及附錄 A 的證明文字。數學符號保留原義;附錄冗長的連續代數推導以原文圖頁完整保留,並翻譯每段推導說明。原圖內的英文標籤仍保留,中文圖說與解讀另列。
與研究題目的關係另見 Async-HFL與主動式網路報修的關聯評估;該筆記是分析,不是論文原文。

作者與出版資訊

Xiaofan Yu、Harsh Vardhan、Quanling Zhao、Emily Ekaireb、Xiyuan Zhang、Arya Mazumdar、Tajana Šimunić Rosing:美國加州大學聖地牙哥分校(UCSD),La Jolla,California。Ludmila Cherkasova:Arm Research,San Jose,California。

聯絡信箱依作者順序:x1yu@ucsd.edu、lucy.cherkasova@gmail.com、hharshvardhan@ucsd.edu、quzhao@ucsd.edu、eekaireb@ucsd.edu、xiyuanzh@ucsd.edu、arya@ucsd.edu、tajana@ucsd.edu。

允許為個人或課堂用途免費製作本文全部或部分的數位或紙本副本,前提是不得以營利或商業利益為目的製作或散布,且副本首頁必須載有本聲明及完整引用。本文第三方內容的著作權必須受到尊重。其他用途請聯絡著作權人/作者。

IoTDI ’23,2023 年 5 月 9–12 日,美國德州 San Antonio。© 2023 著作權由著作權人/作者持有。ACM ISBN 979-8-4007-0037-8/23/05;DOI:10.1145/3576842.3582377。

CCS 分類:電腦系統組織 → 感測器網路;計算方法 → 機器學習。

關鍵字:聯邦學習、階層式感測器與 IoT 網路、非同步聯邦學習。

ACM 引用格式:Xiaofan Yu, Ludmila Cherkasova, Harsh Vardhan, Quanling Zhao, Emily Ekaireb, Xiyuan Zhang, Arya Mazumdar, and Tajana Šimunić Rosing. 2023. Async-HFL: Efficient and Robust Asynchronous Federated Learning in Hierarchical IoT Networks. In International Conference on Internet-of-Things Design and Implementation (IoTDI ’23), May 9–12, 2023, San Antonio, TX, USA. ACM, New York, NY, USA, 18 pages. DOI: 10.1145/3576842.3582377。

摘要

近年來,聯邦學習(Federated Learning,FL)作為一種在裝置上執行的分散式學習典範,受到日益增加的關注。然而,要在真實世界具有階層結構的物聯網(Internet-of-Things,IoT)網路中部署 FL,仍有多項挑戰需要處理。既有研究雖然提出各種方法,分別考量資料異質性、系統異質性、非預期的落後節點(stragglers)及可擴展性,卻沒有任何一項研究提供系統性的解決方案,同時處理階層式且不可靠 IoT 網路中的所有挑戰。

本文提出非同步階層式框架 Async-HFL,在常見的三層 IoT 網路架構中執行 FL。面對差異極大的網路與系統處理延遲,Async-HFL 在閘道與雲端兩層皆採用非同步聚合,避免漫長的等待時間。為充分發揮 Async-HFL 在系統異質性與落後節點存在時的收斂速度潛力,我們在閘道層設計裝置選擇,在雲端層設計裝置與閘道的關聯配置。裝置選擇模組即時挑選資料多樣且速度快的邊緣裝置,觸發本地訓練;裝置與閘道關聯模組則每隔若干雲端 epoch,定期決定有效率的網路拓樸。兩個模組都滿足頻寬限制。

我們以 ns-3 為基礎的大規模模擬,以及來自 NYCMesh 的網路拓樸,評估 Async-HFL 的收斂加速效果。結果顯示,相較於具有客戶端選擇機制的先進非同步 FL 演算法,Async-HFL 依實際經過時間衡量的收斂速度快 1.08–1.31 倍,總通訊成本最多節省 21.6%。我們也在實體部署上驗證 Async-HFL,觀察到它在非預期落後節點出現時仍能穩健收斂。

1. 緒論

在無所不在的 IoT 裝置中嵌入智慧,能執行更複雜的任務,進而使個人醫療照護 [9]、智慧城市 [29] 與自駕車 [6] 等廣泛應用受益。為使大規模網路能進行分散式學習,FL 已成為一種前景看好的典範。學習流程始於中央伺服器將全域模型分發至選定的裝置。接著,各裝置利用本地資料集進行梯度下降訓練,並將更新後的模型傳回伺服器。最後,中央伺服器聚合收到的模型,取得新的全域模型。邊緣裝置不公開本地資料集,只分享更新後的模型。因此,FL 能在保護使用者隱私的同時,讓分散式裝置協作學習。

FL 的經典基準是 Federated Averaging(FedAvg)[42],它使用同步全域聚合:中央伺服器必須等到最慢的裝置回傳,才進行聚合,因此受到無法接受的長延遲或落後節點拖累。近期半非同步 FL 的研究 [12, 15, 44, 52, 59],藉由聚合一定期間內到達的更新,並非同步處理遲到的模型更新,緩解此問題。然而,在異質且不可靠的網路中,半非同步機制的等待期間難以調整,仍然會受到更新不及時的影響。

對真實世界的 IoT 網路而言,整體系統的多樣性妨礙了有效率且穩健的 FL 部署。大量實際部署採階層式組織,例如 NYCMesh [2]、HPWREN [1]。這些架構都可簡化成圖 1 所示的三層結構:雲端伺服器、閘道聚合器、邊緣裝置。雲端層提供資源充足、處理能力強大的伺服器。閘道層包含基地台與路由器,作為連接雲端與邊緣裝置的中介樞紐。最底層的邊緣裝置指感測器、智慧型手機、無人機等小型行動系統,通常受限於資源及能源。

圖 1:在具有異質延遲與不可靠網路的常見三層 IoT 網路中,執行 FL 應用的系統架構。原圖見 第 2 頁。

在異質且階層式 IoT 網路部署 FL,面臨以下挑戰:

  • C1 資料分布異質性:因環境變化或使用者特性,邊緣裝置的本地資料分布可能差異極大。研究已顯示,若沒有謹慎且針對性的管理,非獨立同分布(non-iid)資料會使 FL 收斂變慢,甚至無法收斂 [30, 38, 49]。
  • C2 系統特性異質性:邊緣裝置具備不同的 CPU、記憶體儲存及通訊技術。因此,圖 1 中各層計算延遲及層間通訊延遲可能差異極大。同步與半非同步 FL 會承受較長等待時間。
  • C3 非預期落後節點/裝置退出:能源不足、電路問題或無線干擾,讓 IoT 網路每一層都常有落後節點。若管理不當,學習流程可能被延誤,甚至完全停住。
  • C4 可擴展性:直接把兩層演算法延伸到三層以上的階層網路,可能導致顯著的效能退化,例如模型不收斂或通訊負載增加 [40]。如何在擴展階層架構時保留效益並避免不希望的退化,仍是活躍研究議題。

既有研究雖已探討如何在資料異質性 [30, 38, 49]、系統異質性 [11, 33, 37]、非預期落後節點 [43],及為提高可擴展性而設計的階層式 FL [20, 60] 中,改善一至兩項因素下的 FL 收斂,卻沒有研究為階層式且不可靠的 IoT 網路提供能同時處理全部挑戰的系統性方案。本研究提出首個端到端框架,利用非同步階層式 FL 演算法與系統管理設計,增強效率及穩健性,處理 C1–C4。

本文提出 Async-HFL,讓三層且不可靠的 IoT 網路執行 FL。

在演算法方面,Async-HFL 在閘道與雲端皆採非同步聚合,亦即一收到更新模型便立即聚合。因此,快的邊緣裝置不用等待較慢的同儕;落後節點下載最新全域模型後,也能輕易跟上。相較於把既有兩層非同步 FL 直接延伸至三層,Async-HFL 增加中介閘道聚合層,穩定收斂並節省通訊成本。

在系統管理方面,我們提出兩個模組,改善非同步演算法在系統異質性與落後節點存在時的效能:閘道層裝置選擇,以及雲端層裝置與閘道關聯。閘道層即時決定觸發哪些裝置本地訓練;雲端層則管理網路拓樸,即哪個裝置連到哪個閘道,以提升較長期的效能。兩個模組都建立並求解整數線性規劃,聯合考量資料異質性、系統特性及落後節點。對資料異質性,我們受到線上 coreset 選擇 [57] 啟發,定義學習效用指標,量化裝置梯度的親和性及多樣性。對系統異質性,我們監控各閘道與裝置鏈路的延遲,以及可用連線。

Async-HFL 與過去研究的不同之處在於:它考量比單純損失值更細緻的梯度多樣性資訊,既有先進非同步客戶端選擇演算法 [61] 使用的則只是損失值;它也在不同層級結合裝置選擇及網路拓樸管理,協作最佳化階層式、不可靠 IoT 網路中的模型收斂。為降低通訊額外成本,我們在暖機期間收集所有裝置的梯度,執行主成分分析(PCA),並把 PCA 參數分發給所有裝置。訓練期間只交換梯度的主成分。

本研究的貢獻如下:

  1. 辨識階層式且不可靠 IoT 網路的獨特挑戰,在閘道及雲端皆使用非同步聚合,並正式證明 Async-HFL 在 non-iid 資料分布下的收斂。
  2. 提出根據梯度多樣性建立的學習效用指標,更細緻地量化資料異質性,指引 Async-HFL 的決策。
  3. 結合分散式閘道裝置選擇與雲端裝置閘道關聯模組,在資料、系統異質性與落後節點存在時,共同最佳化模型收斂;利用 PCA 壓縮梯度交換降低通訊額外成本。
  4. 以 ns-3 [3] 和 NYCMesh [2] 為基礎實作並評估不同網路特性下的收斂加速與通訊節省。相較於先進非同步 FL,實際收斂時間加速 1.08–1.31 倍,總通訊節省最多 21.6%。此外,在 Raspberry Pi 4 與 CPU 叢集實體部署上,展示面對落後節點的穩健收斂。

與其他 FL 研究的關係:Async-HFL 著重階層式 IoT 網路的系統變動及潛在落後節點,因此與個人化 [47]、剪枝 [35]、遮罩 [36] 等 FL 技術正交。可以與這些方法結合,但不是本文的焦點。

本文其餘部分安排如下:第 2 節回顧相關研究;第 3 節介紹背景及模型;第 4 節提出動機研究;第 5 節詳述 Async-HFL;第 6 節說明實驗設定及結果;第 7 節總結全文。

2. 相關研究

本節回顧先進研究,並在表 1 中根據 C1–C4 比較既有框架。

同步 FL:以 FedAvg 為基礎,大量研究從理論及實務角度探討資料、系統異質性下的同步 FL [11, 30, 33, 38, 43, 49]。透過各種理論工具 [8, 31, 45, 48, 55, 56],能精心設計客戶端/裝置選擇流程,緩解異質性。多數工作僅考量資料與計算延遲異質性;TiFL [11]、Oort [33]、PyramidFL [37] 則提出通訊延遲差異,實作智慧客戶端選擇,平衡統計效用與系統效用。然而,上述研究都考量資料中心中的 FL;不可靠 IoT 網路的長延遲及落後節點,會使同步方法的表現不理想。

非同步 FL:相較於同步機制,非同步機制在不穩定網路、尤其有數百萬裝置時,收斂較快 [27]。近年非同步 FL 研究不斷增加,焦點包括客戶端選擇 [16, 23, 28, 61, 62]、權重聚合 [51, 51, 58] 及傳輸排程 [34]。半非同步機制用於聚合緩衝的更新 [12, 15, 44, 52, 59]。然而,如何在階層式、異質的 IoT 系統充分運用非同步特性,仍未獲處理。

表 1:Async-HFL 與既有研究的比較。

方法C1 資料異質性C2 系統異質性C3 落後節點C4 可擴展性
同步 FL(兩層)✓✓××
非同步 FL(兩層)✓✓✓×
階層式 FL(超過兩層)✓✓×✓
Async-HFL(三層)✓✓✓✓

階層式 FL:閘道先執行中介聚合,再把本地模型送到雲端,降低閘道與雲端之間的回程通訊 [40]。多項研究建立客戶端關聯與資源配置問題,在同步階層式 FL 中共同最佳化計算及通訊效率 [4, 5, 40, 41]。近期研究考量移動性 [20],以及針對新資料的動態階層聚合 [60]。SHARE [18] 把裝置選擇、裝置與閘道關聯拆成兩個子問題,再共同最小化通訊成本、塑造聚合器資料分布,以提高全域準確率。RFL-HA [50] 在子叢集內同步聚合,叢集首領(閘道)與中央雲端間非同步聚合。上述方法都在系統中採用同步聚合,因此受到落後節點影響。

唯一研究非同步階層式 FL 的工作是 [51]。相較之下,我們提供系統性框架,加入管理模組,即時適應並最佳化 C1–C4 下的收斂。

3. 背景

本節介紹學習模型(3.1)、系統模型(3.2),以及既有兩層非同步 FL 的常用技術(3.3)。表 2 列出符號;圖 2 是範例部署,後續介紹模型時會引用。

3.1 學習模型

假設感測裝置 ii 收集資料點 (xi,yi)∼Di(x_i,y_i)\sim\mathcal D_i,其中 Di\mathcal D_i 代表裝置 ii 的底層資料分布。假設所有分布來自同一領域。當各裝置 i≠ji\ne j 的 Di≠Dj\mathcal D_i\ne\mathcal D_j 時,即出現 non-iid 資料。實務上底層分布未知,各裝置只能存取有限的 nin_i 筆樣本。不同裝置、不同時間的 nin_i 可不同;為簡化表示,省略時間下標。

目標是 FL 的典型目標:學得統一模型 ω∈Rd\omega\in\mathbb R^d,部署於所有分散式裝置。也可結合個人化模型,但超出本文範圍。損失函數 ℓ(ω;xi,yi)\ell(\omega;x_i,y_i) 衡量模型在樣本上的表現誤差。我們最小化所有裝置的經驗風險:

min⁡ω∈RdLN(ω)=1N∑i=1NLi(ω)=1N∑i=1N1ni∑k=1niℓ(ω;xki,yki).(1)\min_{\omega\in\mathbb R^d} L_N(\omega) =\frac1N\sum_{i=1}^N L_i(\omega) =\frac1N\sum_{i=1}^N\frac1{n_i}\sum_{k=1}^{n_i}\ell(\omega;x_k^i,y_k^i).\tag{1}

LNL_N 是中央雲端的全域損失,LiL_i 是感測裝置的損失。

表 2:問題定義的重要符號。

符號意義
C\mathcal C中央雲端
G,N\mathcal G,\mathcal NGG 個閘道、NN 個感測節點的集合
Jt,ItJ_t,I_t時間 tt 可行、實際連接的閘道與感測器鏈路
τiC\tau_i^C裝置 ii 的計算延遲
τjiD,τijU\tau_{ji}^D,\tau_{ij}^U閘道與裝置間下行、上行延遲
τij\tau_{ij}閘道 jj 與裝置 ii 間一輪的延遲
Rij,RjR_{ij},R_j鏈路平均資料速率、閘道全部選定裝置資料速率
BjB_j閘道頻寬限制
ℓ(ω;x,y)\ell(\omega;x,y)模型和樣本上的損失函數
Li,LNL_i,L_N裝置經驗損失、雲端經驗損失
Di,ni\mathcal D_i,n_i感測節點資料分布、樣本數
ωh,γ\omega_h,\gammahh 個雲端 epoch 後的全域模型權重、裝置學習率
ωτ,zj\omega_{\tau,z}^j閘道 jj 在雲端時間戳 τ\tau 下載模型,再聚合 zz 個閘道 epoch 後的模型
ωτ,ζ,ei\omega_{\tau,\zeta,e}^i裝置 ii 在閘道時間戳 ζ\zeta 下載模型,本地更新 ee 個 epoch 後的模型;閘道原本於雲端 τ\tau 下載模型
H,Z,EH,Z,E雲端、閘道及裝置的 epoch 數
ρ\rho非同步 FL 的正則化權重
α,β,s(⋅)\alpha,\beta,s(\cdot)雲端與閘道聚合衰減係數、陳舊度函數
ui,ηi,νiu_i,\eta_i,\nu_i裝置學習效用、梯度親和性、梯度多樣性
κ,ϕ\kappa,\phi裝置選擇與關聯配置的超參數

3.2 系統模型

網路拓樸:在階層式 IoT 網路,中央雲端記為 C\mathcal C,[n]={1,…,n}[n]=\{1,\ldots,n\}。G=[G]\mathcal G=[G] 為閘道/基地台集合,N=[N]\mathcal N=[N] 為大量靜態部署感測裝置集合。所有閘道都有到中央雲端的可行路徑;一個感測裝置可能到得了多個閘道。可達性受無線技術距離限制、感測裝置故障、網路骨幹故障等因素限制。將這些因素納入矩陣 Jt∈ZN×GJ_t\in\mathbb Z^{N\times G},表示實際時間 tt 的可行配對:

J_{t,ij}=\begin{cases}1,&\text{感測器 }i\text{ 與閘道 }j\text{ 在 }t\text{ 可連接}\\0,&\text{其他情況。}\end{cases}\tag{2}

訓練時,每個感測器同時只關聯一個閘道。該閘道觸發其本地訓練,裝置需把結果回傳同一閘道;但感測器可以在兩次聚合之間改連其他閘道。用同樣形狀的 It∈ZN×GI_t\in\mathbb Z^{N\times G} 作即時連線的決策變數:

I_{t,ij}=\begin{cases}1,&\text{感測器 }i\text{ 在 }t\text{ 連接閘道 }j\\0,&\text{其他情況。}\end{cases}\tag{3}

圖 2 中裝置 2 可以連到閘道 1 與 2,但當輪只與閘道 1 通訊,因此 Jt,21=Jt,22=1J_{t,21}=J_{t,22}=1,It,21=1I_{t,21}=1,It,22=0I_{t,22}=0。

圖 2:階層式 FL 部署範例。原圖見 第 4 頁。

計算與通訊模型:計算及通訊延遲是系統異質性的主要來源。本文採通用模型;若取得更多資訊,可採更特定、參數化的 CPU/網路模型,例如 [14, 41]。我們更關注最後一跳鏈路的通訊延遲,因為這裡頻寬更受限,傳送端感測裝置的發射功率也較低。執行時量測裝置 ii 與閘道 jj 完成一輪的時間 τij\tau_{ij},包括:閘道傳模型至裝置的下行延遲 τjiD\tau_{ji}^D、裝置本地訓練計算延遲 τiC\tau_i^C,以及裝置回傳更新模型的上行延遲 τijU\tau_{ij}^U。圖 2 的紅、藍箭頭分別表示下行、上行傳輸,綠箭頭表示本地訓練。因此 τij=τjiD+τiC+τijU\tau_{ij}=\tau_{ji}^D+\tau_i^C+\tau_{ij}^U。閘道、雲端一般有較多計算資源,因此忽略它們的聚合計算延遲。

頻寬限制:頻寬限制為吞吐量/資料速率設定上限。與同步設計不同,非同步情境無法準確建模時間 tt 的吞吐量。因此,給定一輪延遲 τij\tau_{ij} 及大小 MM 的 FL 模型,估計鏈路平均資料速率為 Rij=M/τijR_{ij}=M/\tau_{ij}。閘道 jj 所有選定裝置的總資料速率為 Rj=∑i=1NIijRijR_j=\sum_{i=1}^N I_{ij}R_{ij}。BjB_j 為閘道最後一跳鏈路平均資料速率上限。圖 2 中裝置 1、2 都受 B1B_1 約束,以藍色弧線表示。

3.3 兩層非同步 FL

在非同步 FL,各裝置從雲端下載最新全域模型,執行本地訓練,再將模型上傳至雲端,立即進行非同步聚合。近期演算法 [16, 54] 使用兩項常見技術。

第一,除原始損失 Li(ωi)L_i(\omega^i) 外,裝置加入正則化項,懲罰目前模型與下載的全域模型 ωτ\omega_\tau 之差:

g_i(\omega^i;\omega_\tau)=L_i(\omega^i)+\frac\rho2\|\omega^i-\omega_\tau\|^2.\tag{4}

ρ\rho 是正則化權重。

第二,在雲端使用感知陳舊度的權重聚合:

\alpha_h\leftarrow\alpha s(h-\tau),\tag{5a} \omega_h\leftarrow(1-\alpha_h)\omega_{h-1}+\alpha_h\omega_{new}.\tag{5b}

ωnew\omega_{new} 是新收到的模型權重,hh 是目前雲端 epoch。αh\alpha_h 是常數 α\alpha 乘以陳舊度函數 s(h−τ)s(h-\tau)。陳舊度指模型自上次全域更新以來的 epoch 差異:hh 是目前聚合 epoch,τ\tau 是模型下載時的全域 epoch。陳舊度越大,模型越過時,因此應給較低重要性。這種聚合在非同步情境模擬平均流程。s(h−τ)s(h-\tau) 決定聚合的衰減係數;本文依 [54] 使用多項式函數 sq(h−τ)=(h−τ+1)−qs_q(h-\tau)=(h-\tau+1)^{-q},q>0q>0。

4. 動機研究

本節在階層式、不可靠網路中檢驗既有 FL 框架,為 Async-HFL 的演算法及管理設計提供依據。

近期研究已注意客戶端選擇須考量延遲 [11, 33, 37],但只考量資料中心的延遲分布,這與真實無線網路差異很大。實際量測顯示,無線網路具有長尾延遲分布,且高度不可預測 [46]。我們以 ns3-fl [19] 實作 FL 框架,從 NYCMesh [2] 已安裝節點的位置擷取三層拓樸,見圖 3 左。假設邊緣裝置透過 Wi-Fi 連接閘道,閘道透過 Ethernet 連接伺服器。各節點經度、緯度與高度輸入 ns-3 的 HybridBuildingsPropagationLossModel,以取得平均點對點延遲。為納入網路不確定性,每輪本地訓練在平均延遲上疊加對數常態延遲。假設所有裝置都被選中,一輪內各裝置延遲分布見圖 3 左起第二圖。模擬延遲仿照 [46] 的量測結果。

使用人體活動辨識資料集 [7],把每位受試者的資料配置到單一裝置,形成自然 non-iid 資料。所有實驗中閘道頻寬上限設為 20 KB/s。

圖 3:左:NYCMesh 拓樸;左起第二:所有邊緣裝置一輪延遲的分布;右:NYCMesh 動機研究中,以實際經過時間衡量的收斂表現。原圖見 第 5 頁。

在此設定中,比較:使用先進延遲感知客戶端選擇 Oort [33] 的 FedAvg;雲端非同步、閘道同步聚合的 RFL-HA [50];雲端同步、閘道依 [44] 半非同步聚合的 Semi-async;閘道僅轉送資料,把兩層非同步演算法 [16, 54] 直接延伸成三層的 Two-tier Async;以及本文提出、具有中介閘道聚合的三層 Async-HFL。

圖 3 右顯示實際時間收斂;表 3 顯示相對於 Async-HFL 的總傳輸資料量。即使使用先進客戶端選擇,FedAvg 與 RFL-HA 收斂仍顯著變慢。Semi-async 的等待期間測試 50、100、150 秒,採最佳結果。它雖可加速收斂,但達到相同 95% 準確率仍比完全非同步方法多花 0.39 倍時間。

表 3:動機研究中,達到 95% 測試準確率前的總通訊資料量,相對於 Async-HFL 的比例。

Sync-OortRFL-HASemi-asyncTwo-tier Async
0.79 倍1.42 倍1.66 倍1.30 倍

相較 Two-tier Async,三層 Async-HFL 收斂更穩定、稍快,少掉以 Async-HFL 為基準的 0.3 倍通訊負載,相當於 426 MB 或 1,498 個 MLP 模型。此增益來自中介閘道聚合:多一層「平均」不只平滑曲線,也避免不必要的來回傳輸。

除了演算法,框架管理也是關鍵。我們測試隨機選擇(Async-Random)、優先短延遲(Async-ST)的閘道裝置選擇,以及完整管理的 Async-HFL。圖 3 右顯示,精心設計模組後,Async-HFL 收斂比隨機選擇快 1.24 倍。然而,忽略資料異質性的 Async-ST 等不良選擇,可能使達到 95% 準確率所需時間增至 2.13 倍,最壞情況甚至不收斂。

5. Async-HFL 設計

5.1 概觀

為系統性處理 C1–C4,提出端到端框架 Async-HFL,特別針對階層式、不可靠 IoT 網路,提供有效率且穩健的 FL。主要差異是:以梯度多樣性的學習效用量化 non-iid 資料分布;加入策略性管理元件,即雲端裝置閘道關聯(圖 4 步驟 ①)與閘道裝置選擇(③),共同加速異質資料及系統下的實際收斂。

圖 4 逐步呈現一輪雲端聚合。為簡化,只展示一個階層分支。暖機初始化後,雲端決定底層網路拓樸,即裝置閘道關聯(①),然後向所有閘道分發最新全域模型(②)。閘道把模型分發到選定裝置(③–④)。裝置訓練 EE 個 epoch,回傳更新模型(⑤–⑥)。閘道立即非同步聚合新模型(⑦)。經過 ZZ 次閘道更新後,當前閘道模型上傳雲端,進行全域非同步聚合(⑧–⑨)。

圖 4:Async-HFL 在一個階層網路分支內的逐步流程。原圖見 第 6 頁。

本節後續詳述設計:5.2 提出非同步階層演算法及收斂證明;5.3 定義量化梯度多樣性的學習效用;5.4 說明裝置選擇與裝置閘道關聯的具體設計。

5.2 非同步階層式 FL 演算法

除雲端非同步聚合外,Async-HFL 使用中介閘道層,在閘道進行 ZZ 個 epoch 的陳舊度感知非同步聚合。相較於閘道直接轉送非同步模型更新到雲端,中介聚合降低通訊負擔,且只加入少量假設便維持收斂保證。圖 4 的④–⑦對應兩層非同步 FL,本文階層演算法則涵蓋②、④–⑨,跨越 IoT 三層。

演算法 1 呈現雲端、閘道、裝置的具體實作。雲端及各閘道執行 Cloud 或 Gateway 程序,完成初始化,並非同步觸發 Updater 執行緒進行聚合,直到達到預定 epoch 數。各感測裝置執行 Sensor 程序,以 SGD 本地求解正則化最佳化問題。過去研究按計算資源調整本地 epoch [38],以模型品質交換較快回傳;但因 Async-HFL 採非同步聚合,且網路延遲更長、更不可預期,這類增益可能很小。因此裝置及閘道分別使用固定 EE、ZZ。

演算法 1:非同步階層式 FL。 輸入:C,G,N,H,Z,E,α,β,s(⋅),gi(⋅),ω0\mathcal C,\mathcal G,\mathcal N,H,Z,E,\alpha,\beta,s(\cdot),g_i(\cdot),\omega_0。

程序 Cloud()
  將 (ω₀, 0) 傳給全部閘道 j∈G
  非同步平行執行 CloudUpdater()

執行緒 CloudUpdater()
  對雲端 epoch h∈[H]:
    若收到閘道 j 的 (ω_new^j, τ):
      ω_h ← (1−α_h)ω_(h−1) + α × s(h−τ)ω_new^j
      將 (ω_h, h) 傳給全部閘道

程序 Gateway(j)
  若 Cloud() 觸發:
    接收全域模型與時間戳 (ω_h, h)
    更新 τ←h,ω_(τ,0)^j←ω_h
    將 (ω_(τ,z)^j, z) 傳給連接 j 的選定感測器
    非同步平行執行 GatewayUpdater(j)

執行緒 GatewayUpdater(j)
  對閘道 epoch z∈[Z]:
    接收感測節點 i 的 (ω_new^i, ζ)
    ω_(τ,z)^j ← (1−β_z^j)ω_(τ,z−1)^j + β_z^j ω_new^i
    將 (ω_(τ,z)^j, z) 傳給連接 j 的全部感測器 i
  將 (ω_(τ,Z)^j, τ) 上傳雲端

程序 Sensor(i)
  若 Gateway(j) 觸發:
    接收 (ω_(τ,z)^j, z)
    更新 ζ←z,ω_(τ,ζ,0)^i←ω_(τ,z)^j
    對裝置 epoch e∈[E]:
      ω_(τ,ζ,e)^i ← ω_(τ,ζ,e−1)^i
                       − γ∇g_i(ω_(τ,ζ,e−1)^i; ω_(τ,z)^j)
    將 (ω_(τ,ζ,E)^i, ζ) 上傳閘道 j

收斂分析:本節令 s(⋅)=1s(\cdot)=1。損失函數需满足 LL-smooth、μ\mu-weakly convex 及梯度有界等正則條件。μ\mu 弱凸允許處理非凸(μ>0\mu>0)、凸(μ=0\mu=0)、強凸(μ<0\mu<0)損失。下列是額外假設與收斂結果。完整證明在補充材料,亦可見 arXiv:2301.06646。

假設 1,梯度有界:雲端 LNL_N 及每個裝置 gig_i 的梯度滿足:

∥∇LN(ω)∥2≤V1,∀ω∈Rd,\|\nabla L_N(\omega)\|^2\le V_1,\quad\forall\omega\in\mathbb R^d, ∥∇gi(ω;ω′)∥2≤V2,∀ω,ω′∈Rd, i∈N.\|\nabla g_i(\omega;\omega')\|^2\le V_2,\quad\forall\omega,\omega'\in\mathbb R^d,\ i\in\mathcal N.

假設 2,延遲有界:雲端模型延遲 h−τh-\tau 及閘道模型延遲 z−ζz-\zeta 滿足:

h-\tau\le K_c,\qquad z-\zeta\le K_g.\tag{6}

假設 3,正則化 ρ\rho 足夠大:ρ\rho 夠大,使某固定 c>0c>0、所有 τ,ζ>0\tau,\zeta>0、h≥1h\ge1、i∈[N]i\in[N],有:

-(1+2\rho+c)V_1+(\rho^2-\rho/2)\mathbb E\|\omega^i_{\tau,\zeta,h-1}-\omega^i_{\tau,\zeta}\|^2\ge0.\tag{7}

定理 5.1:對 LL-smooth、μ\mu 弱凸損失 ℓ\ell,在假設 1–3 下,取 γ≤L−1\gamma\le L^{-1}、α≤Kc−3/2\alpha\le K_c^{-3/2}、β≤Kg−3/2\beta\le K_g^{-3/2}。演算法 1 執行 H,Z,EH,Z,E 個雲端、閘道、裝置 epoch 後:

\le\frac{\mathbb E[L_N(\omega_0)-L_N(\omega_H)]}{\alpha\beta\gamma cHZE}+\Xi.\tag{8}$$ 此定理把 [54] 的兩層非同步 FL 證明延伸到裝置、閘道、雲端三層。增加閘道層只需閘道延遲有界假設,並在 $\Xi$ 加入由閘道造成的常數項,不犧牲收斂速率。在固定陳舊度 $s=1$、弱凸等假設下仍能確保收斂;更強假設可使理論及實驗結果更強。 ### 5.3 學習效用 為量化各裝置本地模型聚合到全域模型的學習貢獻,需要能考量資料異質性的指標。先進非同步客戶端選擇 [25, 26] 用本地損失值表示貢獻。Async-HFL 提出考量梯度多樣性的學習效用,比損失值含有更細緻的異質性資訊。 取裝置 $i$ 的最新梯度 $\nabla g_i$,以所有最新梯度平均估計全域梯度 $\nabla L_N=\frac1N\sum_i\nabla g_i$。每個裝置的效用由與全域梯度的親和性 $\eta_i$,及與其他裝置的梯度多樣性 $\nu_i$ 定義: $$u_i=\eta_i+\nu_i,\tag{9a}$$ $$\eta_i=\nabla g_i^\top\nabla L_N,\tag{9b}$$ $$\nu_i=-\frac1{N-1}\sum_{j\ne i}\nabla g_i^\top\nabla g_j.\tag{9c}$$ > 譯者註:式 (9c) 按「全部其他裝置」的文字敘述表示求和範圍;原 PDF 的求和上界印為 $N-1$,並同時標 $j\ne i$,記號存在歧義,見原文第 7 頁。 **圖 5**:結合梯度親和性及多樣性的學習效用視覺化。原圖見 [第 7 頁](../image/Async-HFL-全文繁體中文翻譯/page-07.png)。 圖 5 協助理解:親和性以裝置梯度與全域梯度內積衡量相似程度;多樣性項則累加裝置 $i$ 與其他裝置的成對不相似程度。結合兩者,$u_i$ 偏好資料分布接近全域,或高度多樣的裝置。此概念受線上 coreset 選擇 [57] 啟發:挑選有限樣本存入記憶體,同時盡量保留資料分布知識。此效用同時考量梯度範數和分布,因此包含比單純梯度範數/損失值更多的資訊。 ### 5.4 裝置選擇與裝置閘道關聯 定義量化資料異質性的效用後,設計閘道裝置選擇與雲端裝置閘道關聯,以改善實際收斂。兩者都考量多樣資料分布、異質延遲及非預期落後節點。圖 6 呈現設計與需收集、交換的資訊。閘道裝置選擇在上一次閘道聚合後執行,即時調整參與。關聯配置較不頻繁,每隔若干雲端 epoch 執行,管理較長期效能。 **圖 6**:分散式閘道裝置選擇及雲端裝置閘道關聯概觀。粗箭頭是 FL 必要通訊,細箭頭是額外通訊。原圖見 [第 8 頁](../image/Async-HFL-全文繁體中文翻譯/page-08.png)。 **閘道裝置選擇**:給定時間 $t$ 連接閘道 $j$ 的裝置($I_{t,ij}=1$),挑選子集合觸發非同步訓練。$d_i=1$ 表示裝置已被選擇、最新模型已送出,但更新模型尚未返回。返回後,模組記錄壓縮梯度 $\nabla g_i$,據以更新效用 $u_i$,並追蹤一輪延遲 $\tau_{ij}$ 的移動平均。即時選擇偏好效用高、延遲短的裝置,使它們能快速提供較大的收斂貢獻。將問題建為整數線性規劃: $$\max\sum_{I_{t,ij}=1}d_iu_i(1/\tau_{ij})^\kappa,\tag{10a}$$ $$d_iR_{ij}\le B_j,\quad\forall i:\ I_{t,ij}=1,\tag{10b}$$ $$d_i\in\{0,1\},\quad\forall i:\ I_{t,ij}=1.\tag{10c}$$ 目標結合效用與延遲,$\kappa$ 控制延遲貢獻。式 (10b) 為閘道頻寬上限;問題最多有 $O(|\mathcal N|)$ 個變數及線性限制。 > 譯者註:原文 (10b) 確實逐裝置列出 $d_iR_{ij}\le B_j$,沒有總和。依其「閘道總頻寬」說明,實作時應另外核對是否需 $\sum_i d_iR_{ij}\le B_j$;這裡保留印刷原式,不默默修正。 **雲端裝置閘道關聯**:給定可行鏈路 $J_t$,決定接下來若干雲端 epoch 使用的 $I_t$。$J_t$ 表示即時鏈路可用性,因此裝置/鏈路非預期故障會反映其中,求解器可即時考量。雲端從相應閘道取得梯度及延遲,再回傳決定的關聯矩陣。先前實驗顯示,閘道資料分布越接近全域,收斂越快 [18]。為「塑造」閘道分布並充分利用頻寬,建立多目標問題: $$\max u_{slack}-\phi R_{slack},\tag{11a}$$ $$\sum_{i=1}^N I_{t,ij}u_i\ge u_{slack},\quad\forall j\in\mathcal G,\tag{11b}$$ $$\sum_{i=1}^N I_{t,ij}R_{ij}/B_j\le R_{slack},\quad\forall j\in\mathcal G,\tag{11c}$$ $$I_{t,ij}\le J_{t,ij},\quad\forall i,j,\tag{11d}$$ $$\sum_{j=1}^G I_{t,ij}\le1,\quad\forall i,\tag{11e}$$ $$I_{t,ij}\in\{0,1\}.\tag{11f}$$ 目標平衡各閘道全部關聯裝置的學習效用及吞吐量。鬆弛變數拆開 max-min 運算,使問題仍為整數線性規劃。第一項 $u_{slack}$ 表示所有閘道的最小效用(11b);第二項 $R_{slack}$ 表示所有閘道最大的吞吐比例 $R_j/B_j$(11c)。目標是在考量效用和資料速率下,按閘道頻寬限制平衡配置裝置。$\phi$ 調節子目標重要性。11d 限制使用可行鏈路,11e 使裝置最多連一閘道。問題有 $O(|\mathcal G||\mathcal N|)$ 個變數及線性限制。 兩個整數線性規劃屬於 0–1 背包問題 [21] 的形式,可用最佳化求解器、貪婪啟發式至後設啟發式等多種演算法求解。本文用 Gurobi [22] 實作,顯示 200 節點網路中的計算額外成本,相對於節省的收斂時間可忽略。 **最小化通訊額外成本**:圖 6 細箭頭包含最新梯度、輪次延遲及拓樸 $J_t,I_t$。這些管理中繼資訊中,梯度是主要額外成本來源。為降低成本,先在暖機收集所有裝置梯度,執行 PCA,再把參數分發到裝置。實際訓練只與閘道、雲端交換梯度主成分。第 6.7 節分析額外成本。 ## 6. 評估 ### 6.1 資料集與模型 為模擬資料分布異質性,使用四種典型 IoT 應用的 non-iid 資料集。下列整理各類資料、分割設定及模型。 **應用 1:影像分類**。選用 MNIST [17]、FashionMNIST [53]、CIFAR-10 [32]。前兩者採兩層卷積 CNN,CIFAR-10 採標準 ResNet-18 [24]。三者皆人工分割,每個裝置隨機配置兩個類別,本地樣本由相同分布動態更新。 **應用 2:下一字元預測**。採 Shakespeare [10],給定 80 個字元,預測下一字元。每個角色的台詞配置到一個裝置。模型為兩層、100 個隱藏單元的 LSTM 分類器,包含 8 維 embedding,為此應用的基礎模型。 **應用 3:人體活動辨識**。HAR [7] 資料來自 30 位志願者,每人資料配置至一裝置,採兩個全連接層的典型 MLP。 **應用 4:時間序列預測**。以 HPWREN [1] 歷史資料建構預測任務。HPWREN 為大型環境監測感測器網路,涵蓋 20,000 平方英里,每半小時收集溫度、濕度、風速等。每筆讀值有 11 個特徵,將過去 24 小時的 48 筆讀值合成一筆樣本。每個裝置持有一個測站資料,目標預測下一筆讀值。採 MSE 損失及單層、128 隱藏單元的 LSTM。 **表 4:聯邦資料集統計。模型大小為包含全部權重的封包大小。** | 資料集 | 裝置數 | 平均樣本/裝置 | 資料分割 | 模型 | 大小 | |---|---:|---:|---|---|---| | MNIST | 184 | 600 | 人工,每裝置 2 類 | CNN | 1.6 MB | | FashionMNIST | 184 | 600 | 人工,每裝置 2 類 | CNN | 1.7 MB | | CIFAR-10 | 50 | 1,000 | 人工,每裝置 2 類 | ResNet-18 | 43 MB | | Shakespeare | 143 | 2,892.5 | 自然,每裝置一角色 | LSTM | 208 KB | | HAR | 30 | 308.5 | 自然,每裝置一受試者 | MLP | 285 KB | | HPWREN | 26 | 6,377.4 | 自然,每裝置一測站 | LSTM | 292 KB | ### 6.2 系統實作 大型實體部署一次 FL 試驗可能耗時數天,因此主要使用模擬,仿真實際系統及網路異質性;另在較小實體部署實作、驗證 Async-HFL。 **大規模模擬**:以 ns3-fl [19] 實作離散事件模擬器。該先進模擬器用 PyTorch 做 FL 實驗,用 ns-3 [3] 做網路模擬。與多數只模擬通訊輪次的框架 [30, 39, 49, 54] 不同,ns3-fl 以實際量測模型模擬真實計算與通訊時間。若不考量系統異質性,輪次收斂優異的方法,實際時間可能很差。依第 4 節以 NYCMesh 配置拓樸,含 184 個邊緣裝置、6 個閘道及 1 個伺服器。模擬實作:[Async-HFL 程式碼](https://github.com/Orienfish/Async-HFL)。 **表 5:各資料集的重要參數。** | 資料集 | 目標準確率(%)/誤差 | 閘道參數(原表標為 Gateway)| $\gamma$ | $\rho$ | |---|---:|---:|---:|---:| | MNIST | 95 | 1M | 0.01 | 0.1 | | FashionMNIST | 75 | 1M | 0.01 | 0.1 | | CIFAR-10 | 50 | 20M | 0.001 | 1.0 | | Shakespeare | 35 | 20k | 0.01 | 0.2 | | HAR | 95 | 20K | 0.003 | 0.1 | | HPWREN | $1.5\times10^{-5}$(預測誤差)| 20K | 0.001 | 0.1 | > 譯者註:正文將此欄用於頻寬設定,但原表未在欄名完整標出單位,故數值保留原樣;第 4 節 HAR 明示 20 KB/s。 **實體部署**:以 FedML [13] 在 Raspberry Pi(RPi)4B 及 400 實作。部署含 7 台 RPi 4B、3 台 RPi 400,分布於 7 個不同住家,均連接家用 Wi-Fi 路由器。一些置於後院較遠處,另一些在靠近路由器的臥室,確保網路條件差異。即時 Wi-Fi 流量也會影響,例如住戶同時串流電影,延遲可能增加。此設定模擬應用與其他流量共享頻寬、網路延遲高度多樣的情境。實驗中也觀察到 RPi 一開始無法連線,或較少見地在一次試驗中途非預期暫停。每台 RPi 建立兩個虛擬客戶端,合計 20 個。 另從 CPU 叢集配置 20 個客戶端,各請求 1、2 或 4 個 CPU 核心及 4 GB RAM。RPi 網路條件差異很大;CPU 叢集的網路穩定,但計算延遲隨所請求資源變化。因無法存取家用 Wi-Fi 路由器,閘道及雲端部署於 Ethernet 桌機:Intel Core i7-8700@3.2 GHz、16 GB RAM、NVIDIA GeForce GTX 1060 6 GB GPU。實體部署實作:[FedML 程式碼](https://github.com/Orienfish/FedML)。 ### 6.3 實驗設定 **基準方法**:Async-HFL 主要設計圍繞裝置選擇及關聯,因此比較同步、混合、半非同步、非同步 FL 中的先進客戶端選擇。sync 前綴表示閘道及雲端皆同步,async 表示兩者皆非同步。RFL-HA 與 Semi-async 不遵循此命名規則。 - **Sync-Random/TiFL/Oort/DivFL**:裝置閘道關聯皆隨機,裝置選擇分別採隨機、TiFL [11]、Oort [33]、DivFL [8]。TiFL 將延遲近似裝置分一層,在一層內貪婪選高損失裝置,直到吞吐上限。Oort 用多臂賭徒演算法平衡損失與延遲。DivFL 用貪婪方法最大化考量梯度多樣性的次模組函數。 - **RFL-HA [50]**:閘道同步、雲端非同步;閘道隨機選裝置,重新分群啟發式調整裝置閘道關聯。 - **Semi-async**:閘道依 [44] 半非同步、雲端同步;裝置選擇與關聯隨機。測試 50、100、150 秒半同步期間,取最佳結果。 - **Async-Random/HL**:非同步架構下,關聯隨機,選擇分別隨機或高損失優先。優先高損失/大梯度範數是先進非同步 FL 方法 [25, 26]。未比較 [62],因其演算法依賴完全不同的指標。 **評估指標**:模擬比較達到預定準確率或測試損失所需的實際時間,設定見表 5。目標接近 FedAvg 可達的最佳值;另比較總通訊資料量衡量通訊效率。實體部署以相同實際經過時間的準確率或測試損失量化收斂;也研究執行時間,作為真實平台能源消耗的指示。 ### 6.4 大規模模擬結果 **收斂結果**:先以 NYCMesh 拓樸報告全部資料集的模擬收斂。裝置 $E=5$;非同步閘道 $Z=20$,半非同步及同步閘道 $Z=5$。各方法測試三次隨機試驗,表 6 為平均收斂時間及相應加速比。 **表 6:大規模模擬各資料集的收斂時間加速比。數值為基準耗時/Async-HFL 耗時,小於 1 表示 Async-HFL 較慢。** | 資料集 | Async-HL | Async-Random | Semi-async | RFL-HA | Sync-Oort | Sync-TiFL | Sync-DivFL | Sync-Random | |---|---:|---:|---:|---:|---:|---:|---:|---:| | MNIST | 1.11 | 1.27 | 6.2 | 32.5 | 40.0 | 27.13 | 63.4 | 67.3 | | FashionMNIST | 1.08 | 1.49 | 8.3 | 36.7 | 20.5 | 32.8 | 73.4 | 96.8 | | CIFAR-10 | 1.09 | 1.40 | 2.3 | 12.3 | 44.3 | 59.0 | 62.0 | 61.7 | | Shakespeare | 1.19 | 1.79 | 0.59 | 0.71 | 0.31 | 2.39 | 5.87 | 5.46 | | HAR | 1.31 | 1.22 | 2.7 | 7.4 | 10.3 | 21.6 | 22.5 | 24.1 | | HPWREN | 1.11 | 1.48 | 2.4 | 12.8 | 19.5 | 26.5 | 27.7 | 31.4 | > 原文粗體標各資料集表現最佳的基準:前三者、HPWREN 為 Async-HL,HAR 為 Async-Random,Shakespeare 為 Sync-Oort。 三種影像分類資料集,Async-HFL 相對最佳基準至少加速 1.11、1.08、1.09 倍。HAR 及 HPWREN 裝置較少(表 4),相對最佳基準分別有 0.22 與 0.11 倍額外加速。因此平衡學習效用與系統特性的設計,在人工及自然資料分割都有效。 唯一例外是 Shakespeare,Sync-Oort、RFL-HA、Semi-async 更快達目標;但相對先進非同步 FL,Async-HFL 仍加速 1.19 倍。我們推測所有非同步方法在 Shakespeare 較慢,來自兩層非同步演算法本質的收斂困難。非同步方法約 100 個雲端 epoch 後才首次看到測試準確率提升,同步基準第一個 epoch 就提升。未來將改善 Async-HFL 在 Shakespeare 的演算法設計。 一般而言,即使 Oort、TiFL 等延遲感知選擇,同步方法仍因漫長等待而較慢。RFL-HA、Semi-async 為混合聚合,因此除 Shakespeare 外,比完全非同步慢、比同步快。Async-HFL 相對先進非同步方法,在各資料集皆加速 1.08–1.31 倍。 **圖 7**:左:大規模模擬各資料集總通訊量相對 Async-HFL 的比例;右:不同裝置選擇/關聯組合的收斂時間相對 Async-HFL 比例。原圖見 [第 10 頁](../image/Async-HFL-全文繁體中文翻譯/page-10.png)。 圖 7 左比較非同步類方法的總通訊。只比較非同步類,是因同步方法以漫長等待換取較少雲端 epoch 與通訊;雖通訊較少,表 6 的減速通常無法接受。相對 Async-HL、Async-Random,Async-HFL 在各資料集分別節省 2.6%–21.6%、14.5%–66.8%。Shakespeare 的 Semi-async、RFL-HA 因收斂快,傳輸量顯著較低。CIFAR-10 上 RFL-HA 通訊量僅 Async-HFL 的 0.65 倍,但原文敘述達同準確率多花 11.3 倍時間。MNIST、FashionMNIST、HPWREN、HAR 上,混合方法未同時帶來較快收斂或通訊節省。 > 譯者註:CIFAR-10 原文表 6 記 RFL-HA 加速比 12.3;正文寫「11.3x longer」。兩者可能分別表總耗時倍數與額外耗時,原始措辭不夠明確,此處保留並列。 **效能拆解**:為評估兩模組貢獻,圖 7 右比較純隨機、只選裝置、只做關聯、完整 Async-HFL。目標及頻寬同表 5。各模組在不同資料集貢獻不同。MNIST、CIFAR-10、HAR 主要受惠於平衡拓樸的關聯,單獨加速 1.71、1.23、1.17 倍。FashionMNIST 單模組效果小,合用加速 1.23 倍。Shakespeare 主要受惠於 coreset 裝置選擇,單獨加速 1.12 倍。HPWREN 單用選擇或關聯加速 1.45、1.17 倍,合用加速 1.64 倍。因此兩模組都需要,才能處理不同的資料及系統異質性。 ### 6.5 實體部署結果 **收斂結果**:在 MNIST、FashionMNIST、HAR、HPWREN 驗證。圖 8 彙整實際時間下準確率/測試損失。除非落後節點使系統暫停,Async-HFL 與 Async-HL 執行 30 個雲端輪次,Sync-Random 執行 3 輪。Async-HL 為先進非同步基準,模擬通常第二佳。 在近似時間(分別 7.6K、3.4K、2K 秒),Async-HFL 在 MNIST、FashionMNIST、HAR 達 70%、56%、75%,Async-HL 只達 62%、36%、73%。同步基準受落後節點影響,只取得有限軌跡;設定同步聚合逾時後,取得圖 8 中收斂很慢的曲線。HPWREN 對全部方法計算負擔高,許多裝置因長時間無通訊而退出。Async-HFL 約 2K 秒內朝收斂進展,Async-HL 與 Sync-Random 約 3K、7.7K 秒才達相近測試損失。 小型實體部署雖易受不確定性影響,四資料集仍一致觀察到 Async-HFL 優於基準。結果展示面對延遲異質性與落後節點的穩健性,因兩模組動態引導:閘道選擇即時調整參與,雲端關聯以 $J_t$ 為輸入考量裝置退出。 **圖 8**:實體部署,以實際經過時間呈現收斂結果。原圖見 [第 11 頁](../image/Async-HFL-全文繁體中文翻譯/page-11.png)。 **輪次延遲**:圖 9 量測實際設定,顯示部署具有挑戰性。一輪包括下載模型、本地訓練、回傳模型。量測支持 IoT 網路存在異質系統與網路特性。左右圖分別為 10 個 RPi、CPU 叢集代表客戶端的時間拆解,缺欄表示裝置失敗。部署涵蓋兩種典型但拆解差異很大的情境:RPi 的主要異質性來自網路,因放置位置及路由器距離不同;CPU 叢集主要來自計算,因核心數不同。兩者均造成高度不同且不穩定的一輪延遲,是 Async-HFL 的處理目標。 **圖 9**:實體部署執行 MNIST 的輪次延遲。左:RPi 時間拆解;右:CPU 時間拆解。原圖見第 11 頁。 ### 6.6 敏感度分析 **頻寬限制**:圖 10 左呈現改變全部閘道頻寬後,MNIST 達 95% 的時間。頻寬越受限,相對 Async-Random 的加速越明顯(0.5 MB/s 下 3.47 倍,2 MB/s 下原文記 0.56 倍),因資源有限時智慧選子集合效益更大。0.5 MB/s 時相對 Sync-Random 的加速差距縮小;只能選少數裝置時,FedAvg(Sync-Random)藉多裝置平均維持穩定收斂。 > 譯者註:原文把 0.56x 也稱為 speedup。若照表 6「基準時間/本方法時間」定義,它小於 1,代表較慢;不要解讀為任何頻寬下皆更快。 **目標準確率**:圖 10 右固定其他設定,改變 MNIST 目標。達 85%、90%、95% 時,相對 Async-Random 加速 1.99、1.50、1.34 倍;相對 Sync-Random 為 106、78、51 倍。結果顯示早期收斂快,可歸因於管理設計優先選多樣且快速裝置。 **圖 10**:MNIST 在不同閘道頻寬(左)、雲端目標準確率(右)下,收斂時間相對 Async-HFL 的比例。原圖見 [第 12 頁](../image/Async-HFL-全文繁體中文翻譯/page-12.png)。 **超參數 $\kappa,\phi$**:兩者分別在 (10a)、(11a) 平衡資料異質性/學習效用與系統異質性/延遲,故測試對最終收斂時間的影響。HAR 設定同表 5。圖 11 左用 $\kappa=0.2,0.5,1.0$,較大值提高閘道選擇中延遲權重,收斂較快。中圖用 $\phi=0,0.1,0.2,0.3$;0 只考量資料,較大值提高雲端關聯中的頻寬限制貢獻。合適值(本例 0.1)聯合資料、系統因素,收斂最佳。 **圖 11**:HAR 的敏感度實驗:左為 $\kappa$,中為 $\phi$,右為 PCA 維度。原圖見第 12 頁。 ### 6.7 額外成本分析 如圖 6,主要通訊額外成本為梯度交換。圖 11 右呈現 HAR 不同 PCA 維度對收斂的影響。壓縮為 30 維,通訊額外成本低於 0.5%,相對完整梯度,收斂時間增加低於 6%。因此 PCA 有效降低通訊成本並保留收斂速度。 計算方面,MNIST、FashionMNIST、CIFAR-10、Shakespeare 每次裝置選擇耗時 1.6、1.4、4.3、0.1 秒;雲端關聯每次耗時 1.1、0.9、8.7、0.3 秒。相對實體部署平均 120.26 秒一輪延遲,這些額外計算時間可忽略。 ## 7. 結論 本文提出首個端到端非同步階層聯邦學習框架 Async-HFL,聯合考量 IoT 網路的資料、系統異質性、落後節點及可擴展性。閘道、雲端皆非同步聚合,使異質延遲下收斂更快,且對落後節點穩健。利用量化梯度多樣性的學習效用,設計裝置選擇及裝置閘道關聯,平衡效用、延遲及非預期落後節點,共同最佳化實際收斂。 以 ns-3、NYCMesh 進行全面模擬,在各種網路特性下評估。相對先進非同步 FL,實際時間收斂加速 1.08–1.31 倍,通訊節省 2.6%–21.6%。實體部署證明面對非預期落後節點仍可穩健收斂。 ## 致謝 本研究始於 Xiaofan Yu 2021 年夏季在 Arm Research 的實習。部分經費由美國 National Science Foundation 補助,計畫編號 #2112665(TILOS AI Research Institute)、#2003279、#1911095、#1826967、#2100237、#2112167。 ## 參考文獻 以下保留原始作者、年份、出版資訊及編號,題名附繁體中文翻譯;期刊、會議及方法名稱保留英文,以便查找。 [1] 2022. *High Performance Wireless Research & Education Network (HPWREN)*(高效能無線研究與教育網路)。http://hpwren.ucsd.edu/,線上。 [2] 2022. *New York City (NYC) Mesh*(紐約市 Mesh 網路)。https://www.nycmesh.net/map/,線上。 [3] 2022. *ns-3: a discrete-event network simulator for internet systems*(ns-3:網際網路系統的離散事件網路模擬器)。https://www.nsnam.org/,線上。 [4] Mehdi Salehi Heydar Abad, Emre Ozfatura, Deniz Gunduz, and Ozgur Ercetin. 2020. *Hierarchical federated learning across heterogeneous cellular networks*(跨異質行動網路的階層式聯邦學習)。ICASSP, IEEE, 8866–8870。 [5] Alaa Awad Abdellatif, Naram Mhaisen, Amr Mohamed, Aiman Erbad, Mohsen Guizani, Zaher Dawy, and Wassim Nasreddine. 2022. *Communication-efficient hierarchical federated learning for IoT heterogeneous systems with imbalanced data*(具不平衡資料之 IoT 異質系統的通訊高效率階層聯邦學習)。Future Generation Computer Systems 128, 406–419。 [6] Irfan Ahmad and Karunakar Pothuganti. 2020. *Design & implementation of real time autonomous car by using image processing & IoT*(利用影像處理與 IoT 設計及實作即時自駕車)。ICSSIT, IEEE, 107–113。 [7] Davide Anguita, Alessandro Ghio, Luca Oneto, Xavier Parra Perez, and Jorge Luis Reyes Ortiz. 2013. *A public domain dataset for human activity recognition using smartphones*(以智慧型手機進行人體活動辨識的公開資料集)。ESANN, 437–442。 [8] Ravikumar Balakrishnan, Tian Li, Tianyi Zhou, Nageen Himayat, Virginia Smith, and Jeff Bilmes. 2021. *Diverse Client Selection for Federated Learning via Submodular Maximization*(透過次模組最大化選擇多樣的聯邦學習客戶端)。ICLR。 [9] Sándor Beniczky, Philippa Karoly, Ewan Nurse, Philippe Ryvlin, and Mark Cook. 2021. *Machine learning and wearable devices of the future*(未來的機器學習與穿戴式裝置)。Epilepsia 62, S116–S124。 [10] Sebastian Caldas, Sai Meher Karthik Duddu, Peter Wu, Tian Li, Jakub Konečný, H Brendan McMahan, Virginia Smith, and Ameet Talwalkar. 2018. *Leaf: A benchmark for federated settings*(LEAF:聯邦情境的基準)。arXiv:1812.01097。 [11] Zheng Chai, Ahsan Ali, Syed Zawad, Stacey Truex, Ali Anwar, Nathalie Baracaldo, Yi Zhou, Heiko Ludwig, Feng Yan, and Yue Cheng. 2020. *Tifl: A tier-based federated learning system*(TiFL:分層聯邦學習系統)。HPDC, 125–136。 [12] Zheng Chai, Yujing Chen, Liang Zhao, Yue Cheng, and Huzefa Rangwala. 2020. *Fedat: A communication-efficient federated learning method with asynchronous tiers under non-iid data*(FedAT:non-iid 資料下具有非同步分層的通訊高效率聯邦學習方法)。arXiv:2010.05958。 [13] Chaoyang He et al. 2020. *Fedml: A research library and benchmark for federated machine learning*(FedML:聯邦機器學習研究函式庫與基準)。arXiv:2007.13518。 [14] Mingzhe Chen, H Vincent Poor, Walid Saad, and Shuguang Cui. 2020. *Convergence time minimization of federated learning over wireless networks*(無線網路聯邦學習的收斂時間最小化)。ICC, IEEE, 1–6。 [15] Shuai Chen, Xiumin Wang, Pan Zhou, Weiwei Wu, Weiwei Lin, and Zhenyu Wang. 2022. *Heterogeneous Semi-Asynchronous Federated Learning in Internet of Things: A Multi-Armed Bandit Approach*(IoT 的異質半非同步聯邦學習:多臂賭徒方法)。IEEE Transactions on Emerging Topics in Computational Intelligence 6(5), 1113–1124。 [16] Zheyi Chen, Weixian Liao, Kun Hua, Chao Lu, and Wei Yu. 2021. *Towards asynchronous federated learning for heterogeneous edge-powered internet of things*(邁向異質邊緣運算 IoT 的非同步聯邦學習)。Digital Communications and Networks 7(3), 317–326。 [17] Li Deng. 2012. *The mnist database of handwritten digit images for machine learning research*(供機器學習研究使用的 MNIST 手寫數字影像資料庫)。IEEE Signal Processing Magazine。 [18] Yongheng Deng, Feng Lyu, Ju Ren, Yongmin Zhang, Yuezhi Zhou, Yaoxue Zhang, and Yuanyuan Yang. 2021. *SHARE: Shaping Data Distribution at Edge for Communication-Efficient Hierarchical Federated Learning*(SHARE:在邊緣塑造資料分布以實現通訊高效率階層聯邦學習)。ICDCS, IEEE, 24–34。 [19] Emily Ekaireb, Xiaofan Yu, Kazim Ergun, Quanling Zhao, Kai Lee, Muhammad Huzaifa, and Tajana Rosing. 2022. *ns3-fl: Simulating Federated Learning with ns-3*(ns3-fl:使用 ns-3 模擬聯邦學習)。WNS-3, 97–104。 [20] Chenyuan Feng, Howard H Yang, Deshun Hu, Zhiwei Zhao, Tony QS Quek, and Geyong Min. 2022. *Mobility-aware cluster federated learning in hierarchical wireless networks*(階層無線網路中感知移動性的叢集聯邦學習)。IEEE Transactions on Wireless Communications 21(10), 8441–8458。 [21] Arnaud Fréville. 2004. *The multidimensional 0–1 knapsack problem: An overview*(多維 0–1 背包問題概觀)。European Journal of Operational Research 155。 [22] Gurobi Optimization, LLC. 2022. *Gurobi Optimizer Reference Manual*(Gurobi 最佳化工具參考手冊)。https://www.gurobi.com。 [23] Jiangshan Hao, Yanchao Zhao, and Jiale Zhang. 2020. *Time efficient federated learning with semi-asynchronous communication*(使用半非同步通訊的時間高效率聯邦學習)。ICPADS, IEEE。 [24] Kaiming He, Xiangyu Zhang, Shaoqing Ren, and Jian Sun. 2016. *Deep residual learning for image recognition*(影像辨識的深度殘差學習)。CVPR, 770–778。 [25] Chung-Hsuan Hu, Zheng Chen, and Erik G Larsson. 2021. *Device scheduling and update aggregation policies for asynchronous federated learning*(非同步聯邦學習的裝置排程與更新聚合策略)。SPAWC, IEEE, 281–285。 [26] Chung-Hsuan Hu, Zheng Chen, and Erik G Larsson. 2023. *Scheduling and Aggregation Design for Asynchronous Federated Learning over Wireless Networks*(無線網路非同步聯邦學習的排程與聚合設計)。IEEE Journal on Selected Areas in Communications。 [27] Dzmitry Huba, John Nguyen, Kshitiz Malik, Ruiyu Zhu, Mike Rabbat, Ashkan Yousefpour, Carole-Jean Wu, Hongyuan Zhan, Pavel Ustinov, Harish Srinivas, et al. 2022. *Papaya: Practical, private, and scalable federated learning*(Papaya:實用、具隱私且可擴展的聯邦學習)。Proceedings of Machine Learning and Systems 4, 814–832。 [28] Ahmed Imteaj and M Hadi Amini. 2020. *Fedar: Activity and resource-aware federated learning model for distributed mobile robots*(FedAR:分散式行動機器人的活動與資源感知聯邦學習模型)。ICMLA, IEEE。 [29] Nabaa Ali Jasim, Haider TH, and Salim AL Rikabi. 2021. *Design and implementation of smart city applications based on the internet of things*(基於 IoT 的智慧城市應用設計與實作)。iJIM 15(13)。 [30] Sai Praneeth Karimireddy, Satyen Kale, Mehryar Mohri, Sashank Reddi, Sebastian Stich, and Ananda Theertha Suresh. 2020. *Scaffold: Stochastic controlled averaging for federated learning*(SCAFFOLD:聯邦學習的隨機控制平均)。ICML, PMLR, 5132–5143。 [31] Latif U Khan, Shashi Raj Pandey, Nguyen H Tran, Walid Saad, Zhu Han, Minh NH Nguyen, and Choong Seon Hong. 2020. *Federated learning for edge networks: Resource optimization and incentive mechanism*(邊緣網路的聯邦學習:資源最佳化與激勵機制)。IEEE Communications Magazine 58(10), 88–93。 [32] Alex Krizhevsky, Geoffrey Hinton, et al. 2009. *Learning multiple layers of features from tiny images*(從小型影像學習多層特徵)。 [33] Fan Lai, Xiangfeng Zhu, Harsha V Madhyastha, and Mosharaf Chowdhury. 2021. *Oort: Efficient federated learning via guided participant selection*(Oort:透過引導參與者選擇的高效率聯邦學習)。OSDI, 19–35。 [34] Hyun-Suk Lee and Jang-Won Lee. 2021. *Adaptive transmission scheduling in wireless networks for asynchronous federated learning*(非同步聯邦學習在無線網路中的適應性傳輸排程)。IEEE Journal on Selected Areas in Communications 39(12), 3673–3687。 [35] Ang Li, Jingwei Sun, Pengcheng Li, Yu Pu, Hai Li, and Yiran Chen. 2021. *Hermes: an efficient federated learning framework for heterogeneous mobile clients*(Hermes:異質行動客戶端的高效率聯邦學習框架)。MobiCom, 420–437。 [36] Ang Li, Jingwei Sun, Xiao Zeng, Mi Zhang, Hai Li, and Yiran Chen. 2021. *Fedmask: Joint computation and communication-efficient personalized federated learning via heterogeneous masking*(FedMask:透過異質遮罩同時提高計算與通訊效率的個人化聯邦學習)。SenSys, 42–55。 [37] Chenning Li, Xiao Zeng, Mi Zhang, and Zhichao Cao. 2022. *PyramidFL: A fine-grained client selection framework for efficient federated learning*(PyramidFL:高效率聯邦學習的細粒度客戶端選擇框架)。MobiCom, 158–171。 [38] Tian Li, Anit Kumar Sahu, Manzil Zaheer, Maziar Sanjabi, Ameet Talwalkar, and Virginia Smith. 2020. *Federated optimization in heterogeneous networks*(異質網路中的聯邦最佳化)。Proceedings of Machine Learning and Systems 2, 429–450。 [39] Tian Li, Maziar Sanjabi, Ahmad Beirami, and Virginia Smith. 2019. *Fair resource allocation in federated learning*(聯邦學習的公平資源配置)。arXiv:1905.10497。 [40] Lumin Liu, Jun Zhang, SH Song, and Khaled B Letaief. 2020. *Client-edge-cloud hierarchical federated learning*(客戶端、邊緣、雲端階層聯邦學習)。ICC, IEEE, 1–6。 [41] Siqi Luo, Xu Chen, Qiong Wu, Zhi Zhou, and Shuai Yu. 2020. *Hfel: Joint edge association and resource allocation for cost-efficient hierarchical federated edge learning*(HFEL:聯合邊緣關聯與資源配置的成本高效率階層聯邦邊緣學習)。IEEE Transactions on Wireless Communications 19(10), 6535–6548。 [42] Brendan McMahan, Eider Moore, Daniel Ramage, Seth Hampson, and Blaise Aguera y Arcas. 2017. *Communication-efficient learning of deep networks from decentralized data*(由分散資料進行通訊高效率深度網路學習)。AISTATS, PMLR, 1273–1282。 [43] Aritra Mitra, Rayana Jaafar, George J Pappas, and Hamed Hassani. 2021. *Linear convergence in federated learning: Tackling client heterogeneity and sparse gradients*(聯邦學習的線性收斂:處理客戶端異質性與稀疏梯度)。Advances in Neural Information Processing Systems 34, 14606–14619。 [44] John Nguyen, Kshitiz Malik, Hongyuan Zhan, Ashkan Yousefpour, Mike Rabbat, Mani Malek, and Dzmitry Huba. 2022. *Federated learning with buffered asynchronous aggregation*(使用緩衝非同步聚合的聯邦學習)。AISTATS, PMLR, 3581–3607。 [45] Monica Ribero and Haris Vikalo. 2020. *Communication-efficient federated learning via optimal client sampling*(透過最佳客戶端抽樣的通訊高效率聯邦學習)。arXiv:2007.15197。 [46] Kaixin Sui, Mengyu Zhou, Dapeng Liu, Minghua Ma, Dan Pei, Youjian Zhao, Zimu Li, and Thomas Moscibroda. 2016. *Characterizing and improving wifi latency in large-scale operational networks*(刻畫及改善大型運作網路中的 Wi-Fi 延遲)。MobiSys, 347–360。 [47] Alysa Ziying Tan, Han Yu, Lizhen Cui, and Qiang Yang. 2022. *Towards personalized federated learning*(邁向個人化聯邦學習)。IEEE Transactions on Neural Networks and Learning Systems。 [48] Hao Wang, Zakhary Kaplan, Di Niu, and Baochun Li. 2020. *Optimizing federated learning on non-iid data with reinforcement learning*(以強化學習最佳化 non-iid 資料下的聯邦學習)。INFOCOM, IEEE。 [49] Jianyu Wang, Qinghua Liu, Hao Liang, Gauri Joshi, and H Vincent Poor. 2020. *Tackling the objective inconsistency problem in heterogeneous federated optimization*(處理異質聯邦最佳化中的目標不一致問題)。Advances in Neural Information Processing Systems 33, 7611–7623。 [50] Zhiyuan Wang, Hongli Xu, Jianchun Liu, He Huang, Chunming Qiao, and Yangming Zhao. 2021. *Resource-efficient federated learning with hierarchical aggregation in edge computing*(邊緣運算中使用階層聚合的資源高效率聯邦學習)。INFOCOM, IEEE, 1–10。 [51] Zhongyu Wang, Zhaoyang Zhang, Yuqing Tian, Qianqian Yang, Hangguan Shan, Wei Wang, and Tony QS Quek. 2022. *Asynchronous federated learning over wireless communication networks*(無線通訊網路的非同步聯邦學習)。IEEE Transactions on Wireless Communications 21(9), 6961–6978。 [52] Wentai Wu, Ligang He, Weiwei Lin, Rui Mao, Carsten Maple, and Stephen Jarvis. 2020. *Safa: a semi-asynchronous protocol for fast federated learning with low overhead*(SAFA:低額外成本快速聯邦學習的半非同步協定)。IEEE Transactions on Computers 70(5), 655–668。 [53] Han Xiao, Kashif Rasul, and Roland Vollgraf. 2017. *Fashion-mnist: a novel image dataset for benchmarking machine learning algorithms*(Fashion-MNIST:用於機器學習演算法基準測試的新影像資料集)。 [54] Cong Xie, Sanmi Koyejo, and Indranil Gupta. 2019. *Asynchronous federated optimization*(非同步聯邦最佳化)。arXiv:1903.03934。 [55] Bo Xu, Wenchao Xia, Jun Zhang, Tony QS Quek, and Hongbo Zhu. 2021. *Online client scheduling for fast federated learning*(快速聯邦學習的線上客戶端排程)。IEEE Wireless Communications Letters 10(7), 1434–1438。 [56] Bo Xu, Wenchao Xia, Jun Zhang, Xinghua Sun, and Hongbo Zhu. 2021. *Dynamic client association for energy-aware hierarchical federated learning*(能源感知階層聯邦學習的動態客戶端關聯)。WCNC, IEEE, 1–6。 [57] Jaehong Yoon, Divyam Madaan, Eunho Yang, and Sung Ju Hwang. 2021. *Online coreset selection for rehearsal-based continual learning*(以重演為基礎之持續學習的線上 coreset 選擇)。arXiv:2106.01085。 [58] Linlin You, Sheng Liu, Yi Chang, and Chau Yuen. 2022. *A triple-step asynchronous federated learning mechanism for client activation, interaction optimization, and aggregation enhancement*(用於客戶端啟動、互動最佳化及聚合強化的三步驟非同步聯邦學習機制)。IEEE Internet of Things Journal。 [59] Yu Zhang, Morning Duan, Duo Liu, Li Li, Ao Ren, Xianzhang Chen, Yujuan Tan, and Chengliang Wang. 2021. *CSAFL: A clustered semi-asynchronous federated learning framework*(CSAFL:叢集式半非同步聯邦學習框架)。IJCNN, IEEE, 1–10。 [60] Zhengyi Zhong, Weidong Bao, Ji Wang, Xiaomin Zhu, and Xiongtao Zhang. 2022. *FLEE: A hierarchical federated learning framework for distributed deep neural network over cloud, edge and end device*(FLEE:跨雲端、邊緣與終端裝置之分散深度神經網路的階層聯邦學習框架)。ACM TIST。 [61] Chendi Zhou, Hao Tian, Hong Zhang, Jin Zhang, Mianxiong Dong, and Juncheng Jia. 2021. *TEA-fed: time-efficient asynchronous federated learning for edge computing*(TEA-fed:邊緣運算的時間高效率非同步聯邦學習)。ACM CF, 30–37。 [62] Hongbin Zhu, Yong Zhou, Hua Qian, Yuanming Shi, Xu Chen, and Yang Yang. 2022. *Online client selection for asynchronous federated learning with fairness consideration*(考量公平性的非同步聯邦學習線上客戶端選擇)。IEEE Transactions on Wireless Communications。 ## 附錄 A:定理 5.1 的證明 > 本附錄翻譯全部引理及證明說明。為完整保留原文的長串公式與印刷記號,每個部分附上對應的原始圖頁;圖頁中的數學推導屬於本譯文保留的公式內容。下列另外轉錄各引理的主要結論,方便 Obsidian 閱讀。原文疑似錯字不作無聲修正,文末另列核對注意事項。 正文在三項假設下建立 Async-HFL 收斂性。分析把 [54] 的單層非同步證明延伸成兩層非同步。證明涉及以梯度範數 $\nabla L_N$ 界定不同步驟間雲端損失 $L_N$ 的偏移。我們從最底層裝置開始,建立裝置步驟 $e$ 的進展;利用 $E$ 步後的進展,界定下一層閘道的進展。再往上到雲端則簡單得多,因雲端與閘道更新、假設相似。下一小節從任一裝置的進展開始,說明完整證明。 ### A.1 裝置步驟 各裝置只執行 SGD,因此可用假設 1、3 界定每次梯度更新的進展。 **引理 A.1,裝置單步**:在假設 1、3 下,一步裝置更新後,按 $L_N$ 衡量的進展為:

\mathbb E L_N(\omega^i_{\tau,\zeta,e}) \le\mathbb E L_N(\omega^i_{\tau,\zeta,e-1}) -\gamma c\mathbb E|\nabla L_N(\omega^i_{\tau,\zeta,e-1})|^2 +\gamma^2O(\rho E^2V_2),\tag{12}

對所有 $\tau,\zeta\ge0$、$e\ge1$、$i\in\mathcal N$ 成立。 **證明**:損失為 $L$-smooth,因此 $L_N(\omega)$、$g_{N,\omega'}(\omega)$ 分別為 $L$-smooth、$(2L+2\rho)$-smooth。為簡化,下列推導省略下標 $\tau$ 及上標 $i$。由於 $g_{N,\omega'}(\omega)\ge L_N(\omega)$,對任何 $\omega,\omega'$ 成立,因此先以上界 $g_N$ 取代 $L_N$,再套用平滑性,展開正則化項,最後用梯度界限;完整連續不等式保留在下方第 14 頁左欄。 第二步使用 $g_{N,\omega_\zeta}$ 平滑性;第三步展開 $g_{N,\omega_\zeta}$;第四步以假設 1 界定 $\mathbb E\|\nabla g_{\omega_\zeta}(\omega_{\zeta,e-1})\|^2$ 及 $\mathbb E\|\omega_\zeta-\omega_{\zeta,e-1}\|^2$。注意後者可寫成過去梯度更新總和的平方範數,依次有:

\mathbb E|\omega_\zeta-\omega_{\zeta,e-1}|^2 =\mathbb E\left|\sum_{r=1}^{e-1}\gamma\nabla g_{\omega_\zeta}(\omega_{\zeta,r})\right|^2 \le\gamma^2(e-1)\sum_{r=1}^{e-1}\mathbb E|\nabla g_{\omega_\zeta}(\omega_{\zeta,r})|^2 \le\gamma^2(e-1)^2V_2\le\gamma^2E^2V_2.

接著利用假設 3 界定剩餘項,並引入 $\nabla L_N$。對某 $c>0$,考量 $\mathbb E\langle\nabla g_{N,\omega_\zeta},\nabla g_{\omega_\zeta}\rangle-c\mathbb E\|\nabla L_N\|^2$;展開後,透過內積下界及梯度界,得到:

\ge -V_1(\rho+\tfrac12+c)-V_2(\rho+\tfrac12) +(\rho^2-\tfrac\rho2)\mathbb E|\omega_{\zeta,e-1}-\omega_\zeta|^2.

完整展開在第 14 頁右欄。前兩步展開 $g_{N,\omega}$ 與 $g_\omega$ 定義;第二、第三步用 $\langle a,b\rangle\ge-\|a\|^2/2-\|b\|^2/2$;第四步用假設 1。最後依假設 3,存在 $c>0$ 使最後不等式始終大於等於 0。因此原文接著寫: $$\mathbb E\langle\nabla g_{N,\omega_\zeta}(\omega_{\zeta,e-1}),\nabla g_{\omega_\zeta}(\omega_{\zeta,e-1})\rangle \le c\mathbb E\|\nabla L_N(\omega_{\zeta,e-1})\|^2.$$ 利用此式即證明引理,獲得 (12) 的進展界。□ > 譯者註:以上最後的 $\le$ 是原文印刷方向;它與前句「內積減去 $c$ 倍範數不小於 0」不一致,按前句應導出 $\ge$。這是原文疑點,不能把它當成已獨立驗證的正確證明。 將單步引理加總 $E$ 步,即可界定一裝置全部本地更新。 **引理 A.2,裝置全部步驟**:原文標為在假設 2、3 下,裝置 $E$ 步後的目標進展:

\mathbb E L_N(\omega^i_{\tau,\zeta,E}) \le\mathbb E L_N(\omega^i_{\tau,\zeta}) -\gamma c\sum_{e=1}^E\mathbb E|\nabla L_N(\omega^i_{\tau,\zeta,e-1})|^2 +\gamma^2O(\rho E^3V_2),\tag{13}

對所有 $\tau,\zeta\ge0$、$i\in\mathcal N$ 成立。 **A.1 全部原式與連續推導**: ![](../image/Async-HFL-全文繁體中文翻譯/page-14.png) ### A.2 閘道步驟 往上到閘道層,更新只是當前閘道模型與收到的裝置訓練模型之凸組合。這是證明第一次需處理非同步之處。先用假設 2,在下列中介引理明確導出延遲遞迴。 **引理 A.3,閘道延遲遞迴**:令 $\omega^j_{\tau,\zeta,E}$ 為閘道 $j$ 在步驟 $z$ 收到的更新,定義 $\mathcal R^j_{\tau,z}=\mathbb E\|\omega^j_{\tau,\zeta}-\omega_{\tau,z-1}\|^2$。假設 1、2 下: $$\mathcal R^j_{\tau,z}\le K_g\beta^2\left(\gamma^2O(K_gE^2V_2)+4\sum_{r=1}^{K_g-1}\mathcal R^j_{\tau,z-r}\right),\tag{14}$$ 對所有 $\tau\ge0$、$z\ge1$、$j\in\mathcal G$ 成立。 **證明**:省略 $j$、$\tau$,只考量指定機器。由假設 2,$z-1-\zeta\le K_g$,所以 $\zeta$ 與 $z-1$ 間最多 $K_g$ 次閘道更新。利用此事實,把 $\mathcal R_z$ 改寫成連續模型差之總和的平方範數,再界定為各差平方範數之和。用三角不等式拆開範數項。若 $\omega_{\zeta_{z-r},E}$ 是閘道在 $z-r$ 步收到的更新,可得式 (15)。 接著考量 (15) 的單一求和項,把裝置自身 $E$ 步的改變與閘道陳舊造成的改變拆開,各自界定。利用裝置 $E$ 步可表示成 $E$ 個梯度更新總和,且依假設 1 每個梯度範數受 $O(V_2)$ 約束,配合三角不等式。代回 (15),得到 (14);最後使用假設 2 的延遲上界 $K_g$。□ 有了上述延遲遞迴,以及 A.2 的裝置進展,可用單次閘道更新取得閘道單步進展。 **引理 A.4,閘道單步**:假設 1、2、3 下: $$\mathbb E[L_N(\omega^j_{\tau,z})-L_N(\omega^j_{\tau,z-1})] \le-\gamma\beta c\sum_{e=0}^{E-1}\mathbb E\|\nabla L_N(\omega^{i_{j,z}}_{\tau,\zeta,e})\|^2+\Delta_z,\tag{16}$$ 對所有 $\tau\ge0$、$z\ge1$、$j\in\mathcal G$。$i_{j,z}$ 是第 $z$ 次閘道更新使用的裝置,且: $$\Delta_z=\beta\gamma^2O(\rho E^3V_2)+\beta O(\sqrt{V_1})\sqrt{\mathcal R^j_{\tau,z}}+\beta O(L+\rho)\mathcal R^j_{\tau,z}.$$ **證明**:再次省略 $\tau,j$。考量閘道收到 $\omega_{\zeta,E}$ 後,一次更新的函數差。式 (17)–(20) 先用 $g_N(\omega_1;\omega_2)\ge L_N(\omega_1)$,再用 $g_N$ 凸性,然後展開其組成。式 (20) 第一項是裝置 $E$ 步的函數差,可用 A.2;第二項可仿 A.3,使用三角不等式與假設 1 梯度界。模型差的平方範數得: $$\mathbb E\|\omega_{\zeta,E}-\omega_{z-1}\|^2\le2\gamma^2O(E^2V_2)+2\mathcal R_z.$$ 再用 $L_N$ 的 $L$-smoothness,另處理式 (20) 的函數差項;其上界為: $$O(\sqrt{V_1})\sqrt{\mathcal R_z}+\frac L2\mathcal R_z.$$ 使用 Cauchy–Schwarz 不等式 $\langle a,b\rangle\le\|a\|\|b\|$,假設 1 的梯度界,以及 Jensen 不等式 $\mathbb E\|a\|\le\sqrt{\mathbb E\|a\|^2}$。把這些值代入 (20) 完成證明。□ 利用 A.4、A.3 可界定閘道 $Z$ 步進展。為此,須界定 A.3 遞迴項總和,下個引理提供此界。 **第 15 頁:A.3–A.4 的全部原式,含 (14)–(20)。** ![](../image/Async-HFL-全文繁體中文翻譯/page-15.png) **引理 A.5,閘道遞迴總和**:假設 1、2 下,$\beta\le O(K_g^{-3/2})$ 時: $$\sum_{z=1}^Z\mathcal R_z\le\gamma^2\beta^2O(K_g^2E^2ZV_2),\tag{21}$$ $$\sum_{z=1}^Z\sqrt{\mathcal R_z}\le\gamma\beta O(K_gEZ\sqrt{V_2}).\tag{22}$$ **證明**:先將 A.3 從 $z=1$ 加總至 $Z$,得 (23)。把右側含 $\sum_z\mathcal R_z$ 的項移至左側,得 (24)。加總遞迴時需控制右側總和係數;若小於 1 即可。$\beta=O(K_g^{-3/2})$ 時,$\beta^2K_g^2=O(K_g^{-1})$,可使其為小於 1 的常數。 對平方根之和,先依原文所稱 Jensen 不等式,使用 $\sqrt{\sum_i a_i}\le\sum_i\sqrt{a_i}$、$a_i\ge0$ 於 A.3,得 (25),再加總得到 (26)–(27)。再次控制右側總和係數;$\beta=O(K_g^{-3/2})$ 時,該係數為常數。□ **引理 A.6,閘道全部步驟**:由 A.4、A.5,取 $\beta=O(K_g^{-3/2})$: $$\mathbb E[L_N(\omega^j_{\tau,Z})-L_N(\omega^j_\tau)] \le-\gamma\beta c\sum_{z=1}^Z\sum_{e=0}^{E-1}\mathbb E\|\nabla L_N(\omega^{i_{j,z}}_{\tau,\zeta,e})\|^2+\Delta_\tau,$$ 對所有 $\tau\ge0,j\in\mathcal G$,其中: $$\begin{aligned} \Delta_\tau={}&\beta\gamma^2O(\rho E^3ZV_2) +\gamma\beta^2O(K_gEZ\sqrt{V_1V_2})\\ &+\gamma^2\beta^3O((L+\rho)K_g^2E^2ZV_2).\tag{28} \end{aligned}$$ **證明**:省略 $\tau,j$。直接加總 A.4,取得式 (29);計算 $\sum_{z=1}^Z\Delta_z$ 時,利用 A.5 的两個總和上界,即得 (28)。□ ### A.3 雲端步驟 本節界定雲端單步進展,先界定證明會用到的量。 **引理 A.7,閘道迭代差異**:假設 1、2 下: $$\mathbb E\|\omega_{\tau,Z}-\omega_\tau\|^2 \le\gamma^2\beta^2\left(O(Z^2E^2V_2)+\beta^2O(K_g^2E^2Z^2V_2)\right).\tag{30}$$ **證明**:此項對應閘道對雲端步驟 $\tau$ 收到的模型,執行 $Z$ 步後的迭代差。可拆成閘道單步差的 telescoping sum(相消和),再用三角不等式,得 (31)–(33)。假設 $\omega_{\tau,\zeta_r,E}$ 是閘道第 $r$ 步收到的迭代值。用類似 A.6 的論證,建立 (33) 每個求和項的遞迴: $$\mathbb E\|\omega_{\tau,\zeta_r,E}-\omega_{\tau,r-1}\|^2 \le2\gamma^2O(E^2V_2)+2\mathcal R_r.$$ 代入 (33),再用 A.5 界定 $\sum_z\mathcal R_z$,即得結果。□ **引理 A.8,雲端延遲遞迴**:假設 1、3、2 下,定義 $\mathcal Q_h=\mathbb E\|\omega_\tau-\omega_{h-1}\|^2$,$\omega_{\tau,Z}$ 是雲端步驟 $h$ 收到的迭代值,則: $$\begin{aligned} \mathcal Q_h\le{}&\gamma^2\alpha^2\beta^2\left(O(K_c^2Z^2E^2V_2)+\beta^2O(K_c^2K_g^2E^2Z^2V_2)\right)\\ &+2\alpha^2K_c\sum_{r=1}^{K_c-1}\mathcal Q_{h-r}.\tag{34} \end{aligned}$$ **第 16 頁:A.5–A.8 全部原式,含 (21)–(34)。** ![](../image/Async-HFL-全文繁體中文翻譯/page-16.png) **A.8 證明**:由假設 2,$h-1-\tau\le K_c$,所以 $\omega_\tau$ 與 $\omega_{h-1}$ 間最多 $K_c$ 個雲端步驟。把此差寫成連續雲端步的相消和,再用三角不等式,得 (35)–(36)。分別分析 (36) 求和每項;令 $\omega_{\tau_{h-r},Z}$ 為雲端 $h-r$ 步收到的模型。利用雲端更新式、範數凸性、三角不等式及 A.7,界定每次更新差。代入 (36),再用假設 2,得到 (37)–(38),即為所求。□ 有了這些引理,可分析雲端單步進展。 **引理 A.9,雲端單步**:假設 1、3、2 下,一次雲端更新後: $$\mathbb E[L_N(\omega_h)-L_N(\omega_{h-1})] \le-\gamma\alpha\beta c\sum_{z=0}^{Z-1}\sum_{e=0}^{E-1}\mathbb E\|\nabla L_N(\omega_{\tau,\zeta,e})\|^2+\Xi_h,\quad h>0.\tag{39}$$ 其中: $$\begin{aligned} \Xi_h={}&\alpha\beta\gamma^2O(\rho E^3ZV_2) +\alpha\gamma\beta^2O(K_gEZ\sqrt{V_1V_2})\\ &+\alpha\gamma^2\beta^3O((L+\rho)K_g^2E^2ZV_2)\\ &+\alpha\rho\gamma^2\beta^2\left(O(Z^2E^2V_2)+\beta^2O(K_g^2E^2Z^2V_2)\right)\\ &+O(\alpha\sqrt{V_1})\sqrt{\mathcal Q_h}+\alpha O(\rho+L)\mathcal Q_h. \end{aligned}$$ **證明**:到式 (20) 為止,證明類似 A.4,所以直接列出對應的雲端單步式 (40)–(41): $$\begin{aligned} \mathbb E[L_N(\omega_h)-L_N(\omega_{h-1})]\le{}&\alpha\mathbb E[L_N(\omega_{\tau,Z})-L_N(\omega_\tau)]\\ &+\alpha\mathbb E[L_N(\omega_\tau)-L_N(\omega_{h-1})]\\ &+\frac{\alpha\rho}{2}\mathbb E\|\omega_{\tau,Z}-\omega_{h-1}\|^2.\tag{41} \end{aligned}$$ 仿 A.4 分析:原文稱 (41)「第二項」可用 A.6,因為是閘道 $Z$ 步的函數差;第三項用三角不等式及 A.7、A.8,其界為: $$\mathbb E\|\omega_{\tau,Z}-\omega_{h-1}\|^2 \le2\gamma^2\beta^2\left(O(Z^2E^2V_2)+\beta^2O(K_g^2E^2Z^2V_2)\right)+2\mathcal Q_h.$$ 原文稱「第一項」另用 $L_N$ 的平滑性處理,所展示的實際式子是 $L_N(\omega_\tau)-L_N(\omega_{h-1})$,界為 $O(\sqrt{V_1})\sqrt{\mathcal Q_h}+(L/2)\mathcal Q_h$。使用 Cauchy–Schwarz、假設 1 及 Jensen 不等式。代入 (41) 完成證明。□ > 譯者註:原文上述第一、第二項的文字指稱與式 (41) 順序對調;保留此疑點,讀者應以公式對照。 接著界定遞迴項 $\mathcal Q_h$、$\sqrt{\mathcal Q_h}$ 的總和。 **引理 A.10,雲端遞迴總和**:假設 1、2、3 下,$\alpha\le O(K_c^{-3/2})$ 時: $$\sum_{h=1}^H\mathcal Q_h \le\gamma^2\alpha^2\beta^2\left(O(HK_c^2Z^2E^2V_2)+\beta^2O(HK_c^2K_g^2E^2Z^2V_2)\right),\tag{42}$$ $$\sum_{h=1}^H\sqrt{\mathcal Q_h} \le\gamma\alpha\beta\left(O(HK_cZE\sqrt{V_2})+\beta O(HK_cK_gEZ\sqrt{V_2})\right).\tag{43}$$ **證明**:從 A.8 開始,將 $h=1$ 至 $H$ 加總。將右側 $2\alpha^2K_c^2\sum_h\mathcal Q_h$ 移項,得到乘以 $(1-2\alpha^2K_c^2)^{-1}$ 的界。加總遞迴需控制右側係數;係數小於 1 即可。$\alpha=O(K_c^{-3/2})$ 時,$\alpha^2K_c^2=O(K_c^{-1})$,可使其為小於 1 的常數。 對平方根總和,先依原文所稱 Jensen 不等式,用 $\sqrt{\sum_i a_i}\le\sum_i\sqrt{a_i}$、$a_i\ge0$ 於 A.8。再加總、移項,得到帶 $(1-\alpha K_c^{3/2})^{-1}$ 的界。再控制右側係數;$\alpha=O(K_c^{-3/2})$ 時該係數為常數。□ **第 17 頁:A.8–A.10 的原式與連續推導,含 (35)–(43);A.10 的證明續於第 18 頁左上。** ![](../image/Async-HFL-全文繁體中文翻譯/page-17.png) ### A.4 完成證明 雲端 $H$ 步後全部步驟的進展,就是定理 5.1。以下給出證明。 先將 A.9 從 $h=1$ 加總至 $H$,得到: $$\mathbb E[L_N(\omega_H)-L_N(\omega_0)] \le-\gamma\alpha\beta c\sum_{h=1}^H\sum_{z=0}^{Z-1}\sum_{e=0}^{E-1} \mathbb E\|\nabla L_N(\omega_{\tau,\zeta,e})\|^2+\sum_{h=1}^H\Xi_h.\tag{44}$$ 先利用 A.10 對 $\mathcal Q_h$ 及其平方根遞迴總和的界,計算 $\sum_h\Xi_h$。完整展開與代入結果保留於第 18 頁右上,包含裝置誤差、閘道延遲、雲端延遲各項。 接著考量: $$\begin{aligned} \min_{h=0}^{H-1}\mathbb E\|\nabla L_N(\omega_h)\|^2 &\le\frac1H\sum_{h=0}^{H-1}\mathbb E\|\nabla L_N(\omega_h)\|^2\\ &\le\frac1{HZ}\sum_{h=0}^{H-1}\sum_{z=0}^{Z-1}\mathbb E\|\nabla L_N(\omega_{\tau,z})\|^2\\ &\le\frac1{HZE}\sum_{h=0}^{H-1}\sum_{z=0}^{Z-1}\sum_{e=0}^{E-1}\mathbb E\|\nabla L_N(\omega_{\tau,\zeta,e})\|^2. \end{aligned}$$ 原文說明以上不等式使用 $\min_{i\in[n]}a_i\le\frac1n\sum_i a_i$ 及範數凸性。利用 (44) 得: $$\min_{h=0}^{H-1}\mathbb E\|\nabla L_N(\omega_h)\|^2 \le\frac{\mathbb E[L_N(\omega_0)-L_N(\omega_H)]}{\alpha\beta\gamma cHZE} +\frac{\sum_{h=1}^H\Xi_h}{\alpha\beta\gamma cHZE}.\tag{45}$$ 把常數項設為 $\Xi$。 **第 18 頁:A.10 證明續文、A.4 全部推導,含 (44)–(45)。** ![](../image/Async-HFL-全文繁體中文翻譯/page-18.png) ## 譯者核對註記(不是原文) 1. 這裡的 cloud/gateway epoch 主要計數聚合更新;device epoch 是本地訓練更新。在本文的實驗設定中,不能把三者都理解成整個資料集的同一種「完整訓練一遍」。 2. 原文圖頁、表格與公式有少量不一致:式 (9c) 求和上界、(10b) 未列總和、A.1 最後內積不等號方向、A.2 假設編號與 A.9 項次說明等。譯文標明已核對的疑點,不以翻譯代替數學審稿。 3. 第 15 頁 A.4 的平滑性展開,印刷中出現 $\omega_\zeta-\omega_\zeta$ 等疑似誤植;完整原式保留圖頁。讀者若要依論文重現證明,需核對作者程式或其他版本,不能直接抄錄這些行當作推導。 4. 1.08–1.31 倍是相對非同步基準的訓練收斂加速;沒有量測網路報修速度、預警提前量或根因定位正確率。 5. 參考文獻是原論文的文獻表;此翻譯沒有宣稱已閱讀或獨立驗證這 62 篇文獻。 相關分析:[[Async-HFL與主動式網路報修的關聯評估]]。