LAUSR.org creates dashboard-style pages of related content for over 1.5 million academic articles. Sign Up to like articles & get recommendations!

Federated Learning for IoT Devices With Domain Generalization

Photo by hajjidirir from unsplash

Federated learning (FL) is a distributed machine learning (ML) technique that allows numerous Internet of Things (IoT) devices to jointly train an ML model using a centralized server for help.… Click to show full abstract

Federated learning (FL) is a distributed machine learning (ML) technique that allows numerous Internet of Things (IoT) devices to jointly train an ML model using a centralized server for help. Local data never leaves each IoT device in FL, so the local data of IoT devices are protected. In FL, distributed IoT devices usually collect their local data independently, so the data set of each IoT device may naturally form a distinct source domain. In real-world applications, the model trained over multisource domains may have poor generalization performance on unseen target domains. To address this issue, we propose federated adversarial domain generalization (FedADG) to equip FL with domain generalization capability. FedADG employs the federated adversarial learning approach to measure and align the distributions among different source domains via matching each distribution to a reference distribution. The reference distribution is adaptively generated (by accommodating all source domains) to minimize the domain shift distance during alignment. Therefore, the learned feature representation tends to be universal, and, thus, it has good generalization performance over the unseen target domains while protecting local data privacy. Intensive experiments on various data sets demonstrate that FedADG has comparable performance with the state-of-the-art.

Keywords: federated learning; iot devices; domain generalization; local data; generalization

Journal Title: IEEE Internet of Things Journal
Year Published: 2023

Link to full text (if available)


Share on Social Media:                               Sign Up to like & get
recommendations!

Related content

More Information              News              Social Media              Video              Recommended



                Click one of the above tabs to view related content.