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

A Communication-Efficient Federated Text Classification Method Based on Parameter Pruning

Author

Listed:
  • Zheng Huo

    (Information Technology School, Hebei University of Economics and Business, Shijiazhuang 050061, China
    Hebei Cross-Border E-Commerce Technology Innovation Center, Shijiazhuang 050061, China)

  • Yilin Fan

    (Information Technology School, Hebei University of Economics and Business, Shijiazhuang 050061, China)

  • Yaxin Huang

    (Information Technology School, Hebei University of Economics and Business, Shijiazhuang 050061, China)

Abstract

Text classification is an important application of machine learning. This paper proposes a communication-efficient federated text classification method based on parameter pruning. In the federated learning architecture, the data distribution of different participants is not independent and identically distributed; a federated word embedding model FedW2V is proposed. Then the TextCNN model is extended to the federated architecture. To reduce the communication cost of the federated TextCNN model, a parameter pruning algorithm called FedInitPrune is proposed, which reduces the amount of communication data both in the uplink and downlink during the parameter transmission phase. The algorithms are tested on real-world datasets. The experimental results show that when the text classification model accuracy reduces by less than 2%, the amount of federated learning communication parameters can be reduced by 74.26%.

Suggested Citation

  • Zheng Huo & Yilin Fan & Yaxin Huang, 2023. "A Communication-Efficient Federated Text Classification Method Based on Parameter Pruning," Mathematics, MDPI, vol. 11(13), pages 1-13, June.
    • Handle: RePEc:gam:jmathe:v:11:y:2023:i:13:p:2804-:d:1176543
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2227-7390/11/13/2804/pdf
    Download Restriction: no
    File URL: https://www.mdpi.com/2227-7390/11/13/2804/
    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:11:y:2023:i:13:p:2804-:d:1176543. 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.