Skip to main content
SpringerLink
Log in

Computation–Communication Resource Allocation for Federated Learning System with Intelligent Reflecting Surfaces

  • Research Article-Computer Engineering and Computer Science
  • Published:
Arabian Journal for Science and Engineering Aims and scope Submit manuscript

Abstract

Federal learning (FL) is a distributed learning algorithm, which enables user devices (UDs) to collaboratively train model through wireless networks. In order to enhance the performance of FL systems, an intelligent reflector surfaces (IRS) is deployed in the cell to assist the communication between UD and BS. Specifically, the IRS is composed of an array of passive reflective elements that is used for improving the communication link by varying the phase shift of the intended signal. In this paper, the computation and communication resource allocation of the FL system with IRS is designed to minimize the training latency, which is difficult to solve due to non-convex and the coupling of variables. In order to solve this problem, block coordinate descent (BCD) technique is first used to block the variables for alternately optimizing. Then, the user selection, the UDs’ CPU frequency and transmit power allocation are optimized by invoking the Majorize - Minimization(MM) algorithm, and IRS phase shift is obtained by using semidefinite relaxation (SDR) and Gaussian randomization. Numerical results show that the proposed algorithm effectively reduces the training time of the FL system.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Log in via an institution

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Institutional subscriptions

Fig. 1
Fig. 2
Fig. 3
Fig. 4

Similar content being viewed by others

References

  1. Gündüz, D.; Kurka, D.B.; Jankowski, M.; Amiri, M.M.; Ozfatura, E.; Sreekumar, S.: Communicate to learn at the edge. IEEE Commun. Mag. 58(12), 14–19 (2020)

    Article  Google Scholar 

  2. Nguyen, D.C.; Cheng, P.; Ding, M.; Lopez-Perez, D.; Pathirana, P.N.; Li, J.; Seneviratne, A.; Li, Y.; Poor, H.V.: Enabling ai in future wireless networks: a data life cycle perspective. IEEE Commun. Surveys Tutorials 23(1), 553–595 (2021)

    Article  Google Scholar 

  3. Chen, M.; Yang, Z.; Saad, W.; Yin, C.; Poor, H.V.; Cui, S.: A joint learning and communications framework for federated learning over wireless networks. IEEE Trans. Wireless Commun. 20(1), 269–283 (2021)

    Article  Google Scholar 

  4. Luo, S.; Chen, X.; Wu, Q.; Zhou, Z.; Yu, S.: Hfel: Joint edge association and resource allocation for cost-efficient hierarchical federated edge learning. IEEE Trans. Wireless Commun. 19(10), 6535–6548 (2020)

    Article  Google Scholar 

  5. Yang, Z.; Chen, M.; Saad, W.; Hong, C.S.; Shikh-Bahaei, M.: Energy efficient federated learning over wireless communication networks. IEEE Trans. Wireless Commun. 20(3), 1935–1949 (2021)

    Article  Google Scholar 

  6. Gong, S.; Lu, X.; Hoang, D.T.; Niyato, D.; Shu, L.; Kim, D.I.; Liang, Y.-C.: Toward smart wireless communications via intelligent reflecting surfaces: a contemporary survey. IEEE Commun. Surveys Tutorials 22(4), 2283–2314 (2020)

    Article  Google Scholar 

  7. Pan, C.; Ren, H.; Wang, K.; Xu, W.; Elkashlan, M.; Nallanathan, A.; Hanzo, L.: Multicell mimo communications relying on intelligent reflecting surfaces. IEEE Trans. Wireless Commun. 19(8), 5218–5233 (2020)

    Article  Google Scholar 

  8. Hong, S.; Pan, C.; Ren, H.; Wang, K.; Nallanathan, A.: Artificial-noise-aided secure mimo wireless communications via intelligent reflecting surface. IEEE Trans. Commun. 68(12), 7851–7866 (2020)

    Article  Google Scholar 

  9. Bai, T.; Pan, C.; Deng, Y.; Elkashlan, M.; Nallanathan, A.; Hanzo, L.: Latency minimization for intelligent reflecting surface aided mobile edge computing. IEEE J. Sel. Areas Commun. 38(11), 2666–2682 (2020)

    Article  Google Scholar 

  10. Chu, Z.; Xiao, P.; Shojafar, M.; Mi, D.; Mao, J.; Hao, W.: Intelligent reflecting surface assisted mobile edge computing for internet of things. IEEE Wireless Commun. Lett. 10(3), 619–623 (2021)

    Article  Google Scholar 

  11. Ren, J.; Yu, G.; Ding, G.: Accelerating dnn training in wireless federated edge learning systems. IEEE J. Sel. Areas Commun. 39(1), 219–232 (2021)

    Article  Google Scholar 

  12. Sun, Y.; Babu, P.; Palomar, D.P.: Majorization-minimization algorithms in signal processing, communications, and machine learning. IEEE Trans. Signal Process. 65(3), 794–816 (2017)

    Article  MathSciNet  Google Scholar 

  13. Grant, M., Boyd, S.: “CVX: Matlab software for disciplined convex programming, version 2.1,” Mar. (2014).

  14. Wu, Q.; Zhang, R.: Intelligent reflecting surface enhanced wireless network via joint active and passive beamforming. IEEE Trans. Wireless Commun. 18(11), 5394–5409 (2019)

    Article  Google Scholar 

  15. Wu, Q., Zhang, R.: “Intelligent reflecting surface enhanced wireless network: Joint active and passive beamforming design,” In 2018 IEEE Global Communications Conference (GLOBECOM), pp. 1–6, (2018).

  16. Xu, H.; Zhu, B.; Liu, J.; Zhou, A.: Robust beamforming design for secure multiuser miso interference channel. IEEE Commun. Lett. 21(4), 833–836 (2017)

    Article  Google Scholar 

  17. Christopoulos, D.; Chatzinotas, S.; Ottersten, B.: Weighted fair multicast multigroup beamforming under per-antenna power constraints. IEEE Trans. Signal Process. 62(19), 5132–5142 (2014)

    Article  MathSciNet  Google Scholar 

Download references

Author information

Authors and Affiliations

  1. Central China Normal University, Wuhan, China

    Li Zhao, Hongbo Xu, Jun Wang, Yun Chen, Xue Chen & Ze Wang

  2. Hubei Polytechnic University, Huangshi, China

    Jun Wang

Authors
  1. Li Zhao
    View author publications

    You can also search for this author in PubMed Google Scholar

  2. Hongbo Xu
    View author publications

    You can also search for this author in PubMed Google Scholar

  3. Jun Wang
    View author publications

    You can also search for this author in PubMed Google Scholar

  4. Yun Chen
    View author publications

    You can also search for this author in PubMed Google Scholar

  5. Xue Chen
    View author publications

    You can also search for this author in PubMed Google Scholar

  6. Ze Wang
    View author publications

    You can also search for this author in PubMed Google Scholar

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Zhao, L., Xu, H., Wang, J. et al. Computation–Communication Resource Allocation for Federated Learning System with Intelligent Reflecting Surfaces. Arab J Sci Eng 47, 10203–10209 (2022). https://doi.org/10.1007/s13369-021-06438-1

Download citation

  • Received:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s13369-021-06438-1

Keywords

Search

Navigation