Saved in:
Bibliographic Details
Main Authors: Lai, Bingkun, He, Jiayi, Kang, Jiawen, Li, Gaolei, Xu, Minrui, zhang, Tao, Xie, Shengli
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2403.04430
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909131329241088
author Lai, Bingkun
He, Jiayi
Kang, Jiawen
Li, Gaolei
Xu, Minrui
zhang, Tao
Xie, Shengli
author_facet Lai, Bingkun
He, Jiayi
Kang, Jiawen
Li, Gaolei
Xu, Minrui
zhang, Tao
Xie, Shengli
contents Generative Artificial Intelligence (GAI) shows remarkable productivity and creativity in Mobile Edge Networks, such as the metaverse and the Industrial Internet of Things. Federated learning is a promising technique for effectively training GAI models in mobile edge networks due to its data distribution. However, there is a notable issue with communication consumption when training large GAI models like generative diffusion models in mobile edge networks. Additionally, the substantial energy consumption associated with training diffusion-based models, along with the limited resources of edge devices and complexities of network environments, pose challenges for improving the training efficiency of GAI models. To address this challenge, we propose an on-demand quantized energy-efficient federated diffusion approach for mobile edge networks. Specifically, we first design a dynamic quantized federated diffusion training scheme considering various demands from the edge devices. Then, we study an energy efficiency problem based on specific quantization requirements. Numerical results show that our proposed method significantly reduces system energy consumption and transmitted model size compared to both baseline federated diffusion and fixed quantized federated diffusion methods while effectively maintaining reasonable quality and diversity of generated data.
format Preprint
id arxiv_https___arxiv_org_abs_2403_04430
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle On-demand Quantization for Green Federated Generative Diffusion in Mobile Edge Networks
Lai, Bingkun
He, Jiayi
Kang, Jiawen
Li, Gaolei
Xu, Minrui
zhang, Tao
Xie, Shengli
Machine Learning
Distributed, Parallel, and Cluster Computing
Networking and Internet Architecture
Generative Artificial Intelligence (GAI) shows remarkable productivity and creativity in Mobile Edge Networks, such as the metaverse and the Industrial Internet of Things. Federated learning is a promising technique for effectively training GAI models in mobile edge networks due to its data distribution. However, there is a notable issue with communication consumption when training large GAI models like generative diffusion models in mobile edge networks. Additionally, the substantial energy consumption associated with training diffusion-based models, along with the limited resources of edge devices and complexities of network environments, pose challenges for improving the training efficiency of GAI models. To address this challenge, we propose an on-demand quantized energy-efficient federated diffusion approach for mobile edge networks. Specifically, we first design a dynamic quantized federated diffusion training scheme considering various demands from the edge devices. Then, we study an energy efficiency problem based on specific quantization requirements. Numerical results show that our proposed method significantly reduces system energy consumption and transmitted model size compared to both baseline federated diffusion and fixed quantized federated diffusion methods while effectively maintaining reasonable quality and diversity of generated data.
title On-demand Quantization for Green Federated Generative Diffusion in Mobile Edge Networks
topic Machine Learning
Distributed, Parallel, and Cluster Computing
Networking and Internet Architecture
url https://arxiv.org/abs/2403.04430