The widespread adoption of Federated Learning (FL), a privacy-preserving distributed learning methodology, has been impeded by the challenge of high communication overheads, typically arising from the transmission of large-scale models. Existing adaptive quantization methods, designed to mitigate these overheads, operate under the impractical assumption of uniform device participation in every training round. Additionally, these methods are limited in their adaptability due to the necessity of manual quantization level selection and often overlook biases inherent in local devices' data, thereby affecting the robustness of the global model. In response, this paper introduces AQUILA (adaptive quantization in device selection strategy), a novel adaptive framework devised to effectively handle these issues, enhancing the efficiency and robustness of FL. AQUILA integrates a sophisticated device selection method that prioritizes the quality and usefulness of device updates. Utilizing the exact global model stored by devices, it enables a more precise device selection criterion, reduces model deviation, and limits the need for hyperparameter adjustments. Furthermore, AQUILA presents an innovative quantization criterion, optimized to improve communication efficiency while assuring model convergence. Our experiments demonstrate that AQUILA significantly decreases communication costs compared to existing methods, while maintaining comparable model performance across diverse non-homogeneous FL settings, such as Non-IID data and heterogeneous model architectures.
翻译:联邦学习(Federated Learning, FL)作为一种保护隐私的分布式学习方法,其广泛应用因传输大规模模型时产生的高通信开销而受阻。现有旨在缓解此类开销的自适应量化方法,均基于每轮训练中设备统一参与的不可实际假设。此外,这些方法因需手动选择量化级别而适应性有限,且常忽略本地设备数据固有的偏差,从而影响全局模型的鲁棒性。为此,本文提出AQUILA(自适应量化的设备选择策略),一种新颖的自适应框架,旨在有效处理上述问题,提升联邦学习的效率与鲁棒性。AQUILA融合了优先考虑设备更新质量与实用性的精巧设备选择方法。通过利用设备存储的精确全局模型,它能够实现更精确的设备选择标准,减少模型偏差,并限制超参数调整的需求。此外,AQUILA提出了一种创新的量化标准,在确保模型收敛的同时优化通信效率。实验表明,与现有方法相比,AQUILA显著降低了通信成本,并在异构联邦学习设置(如非独立同分布数据与异构模型架构)中保持了相当的模型性能。