核心概念
ProgFed 是一種聯邦學習框架,透過漸進式訓練策略,在不犧牲模型效能的前提下,有效降低了計算和通訊成本。
摘要
ProgFed:透過漸進式訓練實現高效、通訊和計算效率的聯邦學習
本研究旨在解決聯邦學習中計算和通訊成本高的問題,特別是在邊緣設備資源有限的情況下。
研究提出了一種名為 ProgFed 的新型聯邦學習框架,該框架採用漸進式學習策略,逐步擴展網路模型,從而減少訓練過程中的資源需求。
ProgFed 的運作原理:
模型分割:將目標模型分割成多個重疊的子模型。
漸進式訓練:從最小的子模型開始訓練,逐步添加新的層級和參數,直到達到完整的模型結構。
局部監督:為每個子模型引入輕量級的局部監督頭,以提供更精確的訓練信號。