Personalized Over-the-Air Federated Learning with Personalized Reconfigurable Intelligent Surfaces

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Mao, Jiayu, Yener, Aylin
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910304865091584
author Mao, Jiayu
Yener, Aylin
author_facet Mao, Jiayu
Yener, Aylin
contents Over-the-air federated learning (OTA-FL) provides bandwidth-efficient learning by leveraging the inherent superposition property of wireless channels. Personalized federated learning balances performance for users with diverse datasets, addressing real-life data heterogeneity. We propose the first personalized OTA-FL scheme through multi-task learning, assisted by personal reconfigurable intelligent surfaces (RIS) for each user. We take a cross-layer approach that optimizes communication and computation resources for global and personalized tasks in time-varying channels with imperfect channel state information, using multi-task learning for non-i.i.d data. Our PROAR-PFed algorithm adaptively designs power, local iterations, and RIS configurations. We present convergence analysis for non-convex objectives and demonstrate that PROAR-PFed outperforms state-of-the-art on the Fashion-MNIST dataset.
format Preprint
id arxiv_https___arxiv_org_abs_2401_12149
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Personalized Over-the-Air Federated Learning with Personalized Reconfigurable Intelligent Surfaces
Mao, Jiayu
Yener, Aylin
Information Theory
Machine Learning
Over-the-air federated learning (OTA-FL) provides bandwidth-efficient learning by leveraging the inherent superposition property of wireless channels. Personalized federated learning balances performance for users with diverse datasets, addressing real-life data heterogeneity. We propose the first personalized OTA-FL scheme through multi-task learning, assisted by personal reconfigurable intelligent surfaces (RIS) for each user. We take a cross-layer approach that optimizes communication and computation resources for global and personalized tasks in time-varying channels with imperfect channel state information, using multi-task learning for non-i.i.d data. Our PROAR-PFed algorithm adaptively designs power, local iterations, and RIS configurations. We present convergence analysis for non-convex objectives and demonstrate that PROAR-PFed outperforms state-of-the-art on the Fashion-MNIST dataset.
title Personalized Over-the-Air Federated Learning with Personalized Reconfigurable Intelligent Surfaces
topic Information Theory
Machine Learning
url https://arxiv.org/abs/2401.12149