Over-the-air federated learning with joint privacy-accuracy optimization

Author Identifier (ORCID)

Wei Ni: https://orcid.org/0000-0002-4933-594X

Abstract

Federated learning (FL) contributes to data privacy by not disclosing raw data, but encounters challenges of privacy leakage from local gradient uploading. This paper introduces a novel over-the-air computation (AirComp)-based FL system that balances privacy and accuracy by leveraging the waveform superposition and channel propagation characteristics of AirComp. Specifically, we derive the privacy leakage metric to explicitly account for the effects of waveform aggregation and communication noise. We analyze the convergence upper bound to capture model update errors stemming from artificial and communication noise. We formulate a new joint privacy-accuracy optimization problem by incorporating privacy leakage in the model training objective, guiding the learning process towards enhanced privacy protection. We then employ convex optimization techniques to derive the optimal power scaling and artificial noise intensity. Simulations demonstrate up to 80% reduction in privacy leakage compared to baselines under stringent privacy constraints, while maintaining competitive learning performance. Our method exhibits enhanced robustness under low signal-to-noise ratios, achieving 40% lower privacy leakage under equivalent privacy budgets.

Document Type

Journal Article

Date of Publication

1-1-2025

Volume

20

Publication Title

IEEE Transactions on Information Forensics and Security

Publisher

IEEE

School

School of Engineering

Funders

National Natural Science Foundation of China (62271352, 42171404) / Fundamental Research Funds for the Central Universities (22120250094)

Comments

Feng, H., Wang, R., Liu, E., Ni, W., Niyato, D., & Jamalipour, A. (2025). Over-the-air federated learning with joint privacy-accuracy optimization. IEEE Transactions on Information Forensics and Security, 20, 11326–11341. https://doi.org/10.1109/TIFS.2025.3620108

Copyright

subscription content

First Page

11326

Last Page

11341

Share

 
COinS
 

Link to publisher version (DOI)

10.1109/TIFS.2025.3620108