Resource-efficient federated learning with Non-IID data: An auction theoretic approach
2022 (English)In: IEEE Internet of Things Journal, ISSN 2327-4662, Vol. 9, no 24, p. 25506-25524Article in journal (Refereed) Published
Abstract [en]
Federated learning (FL) has gained significant importance for intelligent applications, following data produced on a massive scale by numerous distributed IoT devices. From an FL perspective, the key aspect is that this data is not identically and independently distributed (IID) across different data sources and locations. This distribution-skewness leads to significant quality degradation. Moreover, an intrinsic consequence of using such non-IID data in decentralized learning is increasing costs that would be mitigated if using IID data. As a remedy, we propose a resource-efficient method for training an FL-based application with non-IID data, effectively minimizing cost through an auction approach and mitigating quality degradation through data sharing. In an experimental evaluation, we investigate the FL performance using real-world non-IID data and use the resulting ground-truth outputs to develop functions for estimating the utility of non-IID data, computation resource costs, and data generation costs. These functions are used to optimize the costs of model training, ensuring resource efficiency. It is further demonstrated that using shared-IID data significantly increases the resource efficiency of FL with local non-IID data. This holds true even when the shared IID data size is less than 1% of the size of the local non-IID data. Moreover, this work demonstrates that the profitability of the stakeholders can be maximized using the proposed auction procedure. The integration of the auction procedure and a resource-efficient training strategy allows FL service providers to create practical trading strategies by minimizing the FL clients’ resources and payments in a machine learning marketplace.
Place, publisher, year, edition, pages
IEEE, 2022. Vol. 9, no 24, p. 25506-25524
Keywords [en]
auction theory, Collaborative work, Computational modeling, Costs, Data models, Federated learning, Internet of Things, non-IID data distribution, Resource efficiency, Stakeholders, Training
National Category
Computer Sciences
Identifiers
URN: urn:nbn:se:umu:diva-199222DOI: 10.1109/JIOT.2022.3197317ISI: 000895792600066Scopus ID: 2-s2.0-85136663219OAI: oai:DiVA.org:umu-199222DiVA, id: diva2:1693880
Funder
Knut and Alice Wallenberg Foundation2022-09-082022-09-082023-09-05Bibliographically approved