Decentralized Edge Intelligence: A Dynamic Resource Allocation Framework for Hierarchical Federated Learning

• Published in: IEEE transactions on parallel and distributed systems Vol. 33; no. 3; pp. 536 - 550
• Main Authors: Lim, Wei Yang Bryan, Ng, Jer Shyuan, Xiong, Zehui, Jin, Jiangming, Zhang, Yang, Niyato, Dusit, Leung, Cyril, Miao, Chunyan
• Format: Journal Article
• Language: English
• Published: New York IEEE 01-03-2022; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: Artificial intelligence; auction; Clusters; Computational modeling; Data models; Deep learning; Edge computing; edge intelligence; evolutionary game; Federated learning; Game theory; Games; Machine learning; Magnetic heads; Performance evaluation; Resource allocation; Resource management; Servers; Stability analysis; Training
• ISSN: 1045-9219; 1558-2183
• DOI: 10.1109/TPDS.2021.3096076

To enable the large scale and efficient deployment of Artificial Intelligence (AI), the confluence of AI and Edge Computing has given rise to Edge Intelligence, which leverages on the computation and communication capabilities of end devices and edge servers to process data closer to where it is produced. One of the enabling technologies of Edge Intelligence is the privacy preserving machine learning paradigm known as Federated Learning (FL), which enables data owners to conduct model training without having to transmit their raw data to third-party servers. However, the FL network is envisioned to involve thousands of heterogeneous distributed devices. As a result, communication inefficiency remains a key bottleneck. To reduce node failures and device dropouts, the Hierarchical Federated Learning (HFL) framework has been proposed whereby cluster heads are designated to support the data owners through intermediate model aggregation. This decentralized learning approach reduces the reliance on a central controller, e.g., the model owner. However, the issues of resource allocation and incentive design are not well-studied in the HFL framework. In this article, we consider a two-level resource allocation and incentive mechanism design problem. In the lower level, the cluster heads offer rewards in exchange for the data owners' participation, and the data owners are free to choose which cluster to join. Specifically, we apply the evolutionary game theory to model the dynamics of the cluster selection process. In the upper level, each cluster head can choose to serve a model owner, whereas the model owners have to compete amongst each other for the services of the cluster heads. As such, we propose a deep learning based auction mechanism to derive the valuation of each cluster head's services. The performance evaluation shows the uniqueness and stability of our proposed evolutionary game, as well as the revenue maximizing properties of the deep learning based auction.