Treffer: Mixed-precision federated learning via multi-precision over-the-air aggregation

Title:
Mixed-precision federated learning via multi-precision over-the-air aggregation
Publisher Information:
IEEE
//ieeexplore.ieee.org/document/10978511
Publication Year:
2025
Collection:
Cranfield University: Collection of E-Research - CERES
Document Type:
Konferenz conference object
File Description:
application/pdf
Language:
English
DOI:
10.1109/wcnc61545.2025.10978511
Rights:
Attribution 4.0 International ; http://creativecommons.org/licenses/by/4.0/
Accession Number:
edsbas.F6C56772
Database:
BASE

Weitere Informationen

Over-the-Air Federated Learning (OTA-FL) is a privacy-preserving distributed learning mechanism, by aggregating updates in the electromagnetic channel rather than at the server. A critical research gap in existing OTA - FL research is the assumption of homogeneous client computational bit precision. While in real world application, clients with varying hardware resources may exploit approximate computing (AxC) to operate at different bit precisions optimized for energy and computational efficiency. Model updates with varying precisions among clients present a significant challenge for OTA - FL, as they are incompatible with the wireless modulation superposition process. Here, we propose an mixed-precision OTA-FL framework of clients with multiple bit precisions, demonstrating the following innovations: (i) the superior trade-off for both server and clients within the constraints of varying edge computing capabilities, energy efficiency, and learning accuracy requirements compared to homogeneous client bit precision, and (ii) a multi-precision gradient modulation scheme to ensure compatibility with OTA aggregation and eliminate the overheads of precision conversion. Through case study with real world data, we validate our modulation scheme that enables AxC based mixed-precision OTA-FL. In comparison to homogeneous standard precision of 32-bit and 16-bit, our framework presents more than 10% in 4-bit ultra low precision client performance and over 65% and 13% of energy savings respectively. This demonstrates the great potential of our mixed-precision OTA-FL approach in heterogeneous edge computing environments. ; Engineering and Physical Sciences Research Council (EPSRC) ; The work is supported by EPSRC CHEDDAR: Communications Hub for Empowering Distributed clouD computing Applications and Research (EP/X040518/1) (EP/Y037421/1). ; 2025 IEEE Wireless Communications and Networking Conference (WCNC)