让低资源设备也能参与联邦学习,通过零阶优化提升数据多样性。
Warming Up for Zeroth-Order Federated Pre-Training with Low Resource Clients
- 设计零阶优化器ZOWarmUp,支持从随机初始化开始训练。
- 在低内存/通信设备上实现接近全量训练的性能,提升系统鲁棒性。
- 适合边缘计算中大量低资源设备参与的场景,降低数据偏差。
联邦学习可在不共享数据的前提下协同训练模型,但边缘设备的内存和通信限制常导致部分设备无法参与。本文针对此类低资源设备(低于关键阈值)无法进行模型更新的问题,提出一种新型联邦零阶优化器ZOWarmUp。受MeZO启发,该方法克服了零阶梯度估计方差大导致仅适用于微调的局限,首次实现从随机初始化起的零阶联邦训练。ZOWarmUp利用客户端能力差异与精细方差控制,使低资源设备可参与训练;无需上传完整梯度,仅需少量随机种子,上行通信成本可忽略。实验在多种数据集与模型架构上验证其有效性,显著提升高比例低资源系统中的数据覆盖率与多样性,改善整体训练效果。
原文摘要 · Abstract (English)
Federated learning enables collaborative model training across numerous edge devices without requiring participants to share data; however, memory and communication constraints on these edge devices may preclude their participation in training. We consider a setting in which a subset of edge devices are below a critical memory or communication threshold required to conduct model updates. Under typical federated optimization algorithms, these devices are excluded from training which renders their data inaccessible and increases system induced bias. We are inspired by MeZO, a zeroth-order method used for memory-efficient fine-tuning. The increased variance inherent to zeroth-order gradient approximations has relegated previous zeroth-order optimizers exclusively to the domain of fine tuning; a limitation we seek to correct. We devise a federated, memory-efficient zeroth-order optimizer, ZOWarmUp that permits zeroth-order training from a random initialization. ZOWarmUp leverages differing client capabilities and careful variance reduction techniques to facilitate participation of under-represented, low-resource clients in model training. Like other federated zeroth-order methods, ZOWarmUp eliminates the need for edge devices to transmit their full gradients to the server and instead relies on only a small set of random seeds, rendering the up-link communication cost negligible. We present experiments using various datasets and model architectures to show that ZOWarmUp is a robust algorithm that can can be applied under a wide variety of circumstances. For systems with a high proportion of edge devices that would otherwise be excluded from training, this algorithm provides access to a greater volume and diversity of data, thus improving training outcomes.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。