Simultaneous Wireless Information and Power Transfer Assisted Federated Learning via Nonorthogonal Multiple Access

Yuan Wu, Yuxiao Song, Tianshun Wang, Minghui Dai, Tony Q.S. Quek

研究成果: Article同行評審

2 引文 斯高帕斯(Scopus)


Federated learning (FL) has been considered as a promising approach for enabling distributed learning without sacrificing edge-devices' (EDs') data privacy. However, training machine learning (ML) model distributively is challenging to the EDs with limited energy supply. In this work, we consider that the central parameter-server (which is co-located with the cellular base station, BS) exploits simultaneous wireless information and power transfer (SWIPT) to simultaneously send the aggregated model to all EDs and also transfer energy to them. With the collected energy from the BS, each ED firstly trains its local model, and then all EDs form a non-orthogonal multiple access (NOMA) cluster for sending their local models to the central parameter-server (i.e., the BS). To minimize the total energy consumption, we formulate a joint optimization of the BS's SWIPT duration, each ED's power-splitting ratio, the EDs' NOMA uploading duration as well as the EDs' and BS's processing-rates. To address the non-convexity of the joint optimization problem, we decompose it into two subproblems, both of which are efficiently solved. We then propose an enhanced block coordinate descent (EBCD) algorithm, which iteratively solves the two subproblems in sequence and exploits the idea of simulated annealing to avoid being trapped by some local optimum, to approach to the optimal solution of the original joint optimization problem. By using our EBCD-Algorithm, we further investigate how to properly select the EDs to participate in the FL process, with the objective of minimizing all participants' total energy consumption. A cross-entropy based algorithm, which exploits our EBCD-Algorithm as a subroutine, is proposed to determine the optimal ED-selection. Numerical results are provided to validate our proposed algorithms and demonstrate the performance advantage of the proposed SWIPT assisted FL via NOMA in comparison with some conventional FL schemes.

頁(從 - 到)1846-1861
期刊IEEE Transactions on Green Communications and Networking
出版狀態Published - 2022 9月 1

All Science Journal Classification (ASJC) codes

  • 可再生能源、永續發展與環境
  • 電腦網路與通信


深入研究「Simultaneous Wireless Information and Power Transfer Assisted Federated Learning via Nonorthogonal Multiple Access」主題。共同形成了獨特的指紋。