LAPA-based Dynamic Privacy Optimization for Wireless Federated Learning in Heterogeneous Environments

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Sun, Pengcheng, Liu, Erwu, Ni, Wei, Wang, Rui, Geng, Yuanzhe, Lai, Lijuan, Jamalipour, Abbas
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913859539828736
author Sun, Pengcheng
Liu, Erwu
Ni, Wei
Wang, Rui
Geng, Yuanzhe
Lai, Lijuan
Jamalipour, Abbas
author_facet Sun, Pengcheng
Liu, Erwu
Ni, Wei
Wang, Rui
Geng, Yuanzhe
Lai, Lijuan
Jamalipour, Abbas
contents Federated Learning (FL) is a distributed machine learning paradigm based on protecting data privacy of devices, which however, can still be broken by gradient leakage attack via parameter inversion techniques. Differential privacy (DP) technology reduces the risk of private data leakage by adding artificial noise to the gradients, but detrimental to the FL utility at the same time, especially in the scenario where the data is Non-Independent Identically Distributed (Non-IID). Based on the impact of heterogeneous data on aggregation performance, this paper proposes a Lightweight Adaptive Privacy Allocation (LAPA) strategy, which assigns personalized privacy budgets to devices in each aggregation round without transmitting any additional information beyond gradients, ensuring both privacy protection and aggregation efficiency. Furthermore, the Deep Deterministic Policy Gradient (DDPG) algorithm is employed to optimize the transmission power, in order to determine the optimal timing at which the adaptively attenuated artificial noise aligns with the communication noise, enabling an effective balance between DP and system utility. Finally, a reliable aggregation strategy is designed by integrating communication quality and data distribution characteristics, which improves aggregation performance while preserving privacy. Experimental results demonstrate that the personalized noise allocation and dynamic optimization strategy based on LAPA proposed in this paper enhances convergence performance while satisfying the privacy requirements of FL.
format Preprint
id arxiv_https___arxiv_org_abs_2505_19823
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle LAPA-based Dynamic Privacy Optimization for Wireless Federated Learning in Heterogeneous Environments
Sun, Pengcheng
Liu, Erwu
Ni, Wei
Wang, Rui
Geng, Yuanzhe
Lai, Lijuan
Jamalipour, Abbas
Machine Learning
Artificial Intelligence
Federated Learning (FL) is a distributed machine learning paradigm based on protecting data privacy of devices, which however, can still be broken by gradient leakage attack via parameter inversion techniques. Differential privacy (DP) technology reduces the risk of private data leakage by adding artificial noise to the gradients, but detrimental to the FL utility at the same time, especially in the scenario where the data is Non-Independent Identically Distributed (Non-IID). Based on the impact of heterogeneous data on aggregation performance, this paper proposes a Lightweight Adaptive Privacy Allocation (LAPA) strategy, which assigns personalized privacy budgets to devices in each aggregation round without transmitting any additional information beyond gradients, ensuring both privacy protection and aggregation efficiency. Furthermore, the Deep Deterministic Policy Gradient (DDPG) algorithm is employed to optimize the transmission power, in order to determine the optimal timing at which the adaptively attenuated artificial noise aligns with the communication noise, enabling an effective balance between DP and system utility. Finally, a reliable aggregation strategy is designed by integrating communication quality and data distribution characteristics, which improves aggregation performance while preserving privacy. Experimental results demonstrate that the personalized noise allocation and dynamic optimization strategy based on LAPA proposed in this paper enhances convergence performance while satisfying the privacy requirements of FL.
title LAPA-based Dynamic Privacy Optimization for Wireless Federated Learning in Heterogeneous Environments
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2505.19823