利用边缘设备闲置算力训练大模型,性能接近云端且抗设备波动能力强。
On Harnessing Idle Compute at the Edge for Foundation Model Training
- 基于GEMM的通信不对称性设计新架构,降低设备增多时的通信开销。
- 在相同设备数下,训练速度比现有方法快4至10倍,支持千级异构设备。
- 可快速恢复设备故障,适合分布式边缘训练场景。
当前基础模型训练高度集中,因需海量算力而主要由大型云服务商主导。利用边缘设备闲置算力进行模型训练可实现更普惠的替代方案。然而,现有方法难以匹配云端性能、扩展至更大模型、适配设备内存限制或控制通信开销,也未能有效应对设备异构性和动态变化。本文提出Cleave,其核心洞察是:每个GEMM运算的输入矩阵(通过下行链路传输)远大于部分输出块(通过上行链路返回),这与边缘网络中下行带宽为上行2–10倍的特点相吻合。基于此,采用以参数服务器为中心的架构,使每台设备的通信量随参与设备增加而减少,而非保持恒定。将训练分解为独立的子GEMM任务,统一调度抽象可同时处理内存约束、通信开销和设备故障容错。评估表明,Cleave在相同设备数下达到接近云端的GPU训练性能,相比现有先进边缘训练方法,单批次运行时间提升4–10倍;在更高规模下,可扩展至数千台异构设备——这是以往边缘训练系统无法实现的范围,并且设备故障恢复速度至少快100倍。
原文摘要 · Abstract (English)
The foundation-model ecosystem remains highly centralized because training requires immense compute resources and is therefore largely limited to large cloud operators. Edge-assisted foundation model training that harnesses spare compute on edge devices offers a more democratized alternative. However, existing edge-training approaches fall short: they struggle to match cloud-training performance, scale to larger models, fit within device memory limits, or keep communication overhead manageable. They also do not handle device heterogeneity and churn satisfactorily. We introduce Cleave, built on a structural insight: each GEMM has an asymmetric I/O pattern -- its input matrices, sent over downlink, are much larger than the partial output blocks returned over uplink -- matching edge networks where downlink bandwidth exceeds uplink by 2--10x. Exploiting this alignment with a parameter-server-centric architecture, Cleave makes per-device communication \emph{decrease} as more devices join, rather than stay constant as in conventional TP. Decomposing training into independent sub-GEMM tasks yields one scheduling abstraction that unifies memory constraints, communication overhead, and fault tolerance under device churn. Our evaluation shows that Cleave achieves cloud-comparable GPU training performance and outperforms state-of-the-art edge-training methods by 4--10x in per-batch runtime at the same device counts. Beyond this shared operating range, Cleave scales to thousands of heterogeneous devices -- a regime where prior edge-training systems cannot operate -- and achieves at least 100x faster recovery from device failures.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。