arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2610.22015cs.DC
收藏

面向异构去中心化学习的资源感知子网络分配

Resource-aware Sub-network Allocation for Heterogeneous Decentralized Learning

KTH Royal Institute of Technology(皇家理工学院)

Zhuojun Tian, Ming Xiao, Mikael Skoglund

AI总结:

针对同步去中心化学习受系统异质性限制的问题,提出RASA-DL框架,通过二分搜索分配实现资源感知子网络部署,在不同异质性下提升了精度-延迟权衡表现。

摘要:

计算、通信和内存方面的系统异质性会严重限制同步去中心化学习,每轮训练都受限于资源有限的节点。现有方法在联合确定节点特定模型容量与部署资源的匹配方面支持有限。本文提出RASA-DL,这是一种资源感知子网络分配框架,将分配给每个节点的最大子网络宽度视为部署变量。RASA-DL结合计算与内存成本的离线硬件分析,以及异构去中心化训练下学习效用的部署感知分析,利用这些分析结果和节点、链路级资源信息,将部署建模为受每个节点内存约束和所需聚合学习效用限制的瓶颈延迟最小化问题。利用延迟阈值的单调可行性,RASA-DL通过基于二分搜索的分配过程高效确定部署。在受控资源异质性下的实验显示出良好的精度-延迟权衡:在中等和严重异质性 regime 下,最小延迟操作点在相同建模轮次延迟下,相比最大全局可行均匀部署的精度分别提升2.80和2.43个百分点,在强非独立同分布(Non-IID)数据异质性下优势进一步放大。

英文摘要:

System heterogeneity in computation, communication, and memory can substantially limit synchronous decentralized learning, where each training round is constrained by resource-limited nodes. Existing approaches provide limited support for jointly determining node-specific model capacities from deployment resources. We propose RASA-DL, a resource-aware sub-network allocation framework that treats the maximum sub-network width assigned to each node as a deployment variable. RASA-DL combines offline hardware profiling of computation and memory costs with deployment-aware profiling of learning utility under heterogeneous decentralized training. Using these profiles together with node- and link-level resource information, we formulate deployment as a bottleneck-latency minimization problem subject to per-node memory constraints and a required aggregate learning utility. Exploiting the monotone feasibility of the latency threshold, RASA-DL efficiently identifies the deployment through a binary-search-based allocation procedure. Experiments under controlled resource heterogeneity demonstrate favorable accuracy--latency tradeoffs. Under the Moderate and Severe regimes, the minimum-latency operating point improves accuracy over the largest globally feasible uniform deployment by 2.80 and 2.43 percentage points, respectively, at the same modeled round latency. The advantage is further amplified under strong Non-IID data heterogeneity.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑