Agent Tuning & Optimization 相关度: 6/10

QuantFL: Sustainable Federated Learning for Edge IoT via Pre-Trained Model Quantisation

Charuka Herath, Yogachandran Rahulamathavan, Varuna De Silva, Sangarapillai Lambotharan
arXiv: 2603.17507v1 发布: 2026-03-18 更新: 2026-03-18

AI 摘要

QuantFL通过预训练模型量化,降低边缘IoT联邦学习的通信能耗,实现可持续学习。

主要贡献

  • 提出QuantFL框架,利用预训练模型进行量化
  • 证明预训练模型能够集中更新统计信息,利于高效量化
  • 实验验证QuantFL在降低通信成本方面的有效性

方法论

利用预训练模型初始化,采用memory-efficient bucket量化方法,减少联邦学习中的通信比特数,降低能耗。

原文摘要

Federated Learning (FL) enables privacy-preserving intelligence on Internet of Things (IoT) devices but incurs a significant carbon footprint due to the high energy cost of frequent uplink transmission. While pre-trained models are increasingly available on edge devices, their potential to reduce the energy overhead of fine-tuning remains underexplored. In this work, we propose QuantFL, a sustainable FL framework that leverages pre-trained initialisation to enable aggressive, computationally lightweight quantisation. We demonstrate that pre-training naturally concentrates update statistics, allowing us to use memory-efficient bucket quantisation without the energy-intensive overhead of complex error-feedback mechanisms. On MNIST and CIFAR-100, QuantFL reduces total communication by 40\% ($\simeq40\%$ total-bit reduction with full-precision downlink; $\geq80\%$ on uplink or when downlink is quantised) while matching or exceeding uncompressed baselines under strict bandwidth budgets; BU attains 89.00\% (MNIST) and 66.89\% (CIFAR-100) test accuracy with orders of magnitude fewer bits. We also account for uplink and downlink costs and provide ablations on quantisation levels and initialisation. QuantFL delivers a practical, "green" recipe for scalable training on battery-constrained IoT networks.

标签

Federated Learning Quantization Edge Computing IoT Energy Efficiency

arXiv 分类

cs.LG cs.AI