IDEAS home Printed from https://ideas.repec.org/a/gam/jmathe/v12y2024i6p920-d1360660.html
   My bibliography  Save this article

Federated Learning with Efficient Aggregation via Markov Decision Process in Edge Networks

Author

Listed:
  • Tongfei Liu

    (College of Computer Science and Technology, Zhejiang Normal University, Jinhua 341000, China)

  • Hui Wang

    (College of Computer Science and Technology, Zhejiang Normal University, Jinhua 341000, China
    These authors contributed equally to this work.)

  • Maode Ma

    (College of Engineering, Qatar University, Doha 974, Qatar
    These authors contributed equally to this work.)

Abstract

Federated Learning (FL), as an emerging paradigm in distributed machine learning, has received extensive research attention. However, few works consider the impact of device mobility on the learning efficiency of FL. In fact, it is detrimental to the training result if heterogeneous clients undergo migration or are in an offline state during the global aggregation process. To address this issue, the Optimal Global Aggregation strategy (OGAs) is proposed. The OGAs first models the interaction between clients and servers of the FL as a Markov Decision Process (MDP) model, which jointly considers device mobility and data heterogeneity to determine local participants that are conducive to global aggregation. To obtain the optimal client participation strategy, an improved σ -value iteration method is utilized to solve the MDP, ensuring that the number of participating clients is maintained within an optimal interval in each global round. Furthermore, the Principal Component Analysis (PCA) is used to reduce the dimensionality of the original features to deal with the complex state space in the MDP. The experimental results demonstrate that, compared with other existing aggregation strategies, the OGAs has the faster convergence speed and the higher training accuracy, which significantly improves the learning efficiency of the FL.

Suggested Citation

  • Tongfei Liu & Hui Wang & Maode Ma, 2024. "Federated Learning with Efficient Aggregation via Markov Decision Process in Edge Networks," Mathematics, MDPI, vol. 12(6), pages 1, March.
  • Handle: RePEc:gam:jmathe:v:12:y:2024:i:6:p:920-:d:1360660
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2227-7390/12/6/920/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/2227-7390/12/6/920/
    Download Restriction: no
    ---><---

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jmathe:v:12:y:2024:i:6:p:920-:d:1360660. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    We have no bibliographic references for this item. You can help adding them by using this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.