Joint Communication Scheduling and Resource Allocation for Distributed Edge Learning: Seamless Integration in Next-Generation Wireless Networks

IEEE Transactions on Wireless Communications, vol. 25, pp. 8032–8050

Abstract

Distributed edge learning (DL) is considered a cornerstone of intelligence enablers, since it allows for collaborative training without the necessity for local clients to share raw data with other parties, thereby preserving privacy and security. Integrating DL into the 6G networks requires a coexistence design with existing services such as high-bandwidth (HB) traffic like eMBB. Current designs in the literature mainly focus on communication round-wise designs that assume a rigid resource allocation throughout each communication round (CR). However, rigid resource allocation within a CR is a highly inefficient and inaccurate representation of the system’s realistic behavior, especially when CR duration far exceeds the channel coherence time due to large model size or limited resources. This is due to the heterogeneous nature of the system, as clients inherently may need to access the network at different time instants. This work zooms into one arbitrary CR, and demonstrates the importance of considering a time-dependent design for sharing the resource pool with HB traffic. We first formulate a time-slot-wise optimization problem to minimize the consumed time by DL within the CR while constrained by a DL energy budget. Due to its intractability, a session-based optimization problem is formulated assuming a CR lasts less than a large-scale coherence time. Some scheduling properties of such multi-server joint communication scheduling and resource allocation framework have been established. An iterative algorithm has been designed to solve such non-convex and non-block-separable-constrained problems. Simulation results confirm the importance of the efficient and accurate integration design proposed in this work.

Authors 7

1. Paul Zheng Aachen Chair of Information Theory and Data Analytics

   RWTH Aachen University

   Affiliation as printed

   Chair of Information Theory and Data Analytics, RWTH Aachen University, Aachen, Germany

2. Navid Keshtiarast Aachen Mobile Communications and Computing Group

   RWTH Aachen University

   Affiliation as printed

   Mobile Communications and Computing Group, RWTH Aachen University, Aachen, Germany

3. Pradyumna Kumar Bishoyi Aachen Mobile Communications and Computing Group

   Indian Institute of Technology Jodhpur · RWTH Aachen University

   Affiliation as printed

   Department of Electrical Engineering, Indian Institute of Technology Jodhpur, Jodhpur, Rajasthan, India

   Mobile Communications and Computing Group, RWTH Aachen University, Aachen, Germany

4. Yao Lin Zhu Aachen Chair of Information Theory and Data Analytics

   Wuhan University · RWTH Aachen University

   Affiliation as printed

   School of Electronic Information, Wuhan University, Wuhan, China

   Chair of Information Theory and Data Analytics, RWTH Aachen University, Aachen, Germany

5. Yulin Hu

   Wuhan University

   Affiliation as printed

   School of Electronic Information, Wuhan University, Wuhan, China

   School of Electronic Information, Wuhan University, China

6. Marina Petrova Aachen Mobile Communications and Computing Group

   RWTH Aachen University

   Affiliation as printed

   Mobile Communications and Computing Group, RWTH Aachen University, Aachen, Germany

7. Anke Schmeink Aachen Chair of Information Theory and Data Analytics

   RWTH Aachen University

   Affiliation as printed

   Chair of Information Theory and Data Analytics, RWTH Aachen University, Aachen, Germany