Logo image
D2D-Assisted Split Learning Approach for Model-Heterogeneous Federated Learning
會議論文集

D2D-Assisted Split Learning Approach for Model-Heterogeneous Federated Learning

P.-C. Chang, T.-C. Chiu 和 H.-T. Huang
Proceedings - IEEE Global Communications Conference, GLOBECOM, 頁碼.5653-5658
2025
Web of Science ID: WOS:001810917600395

摘要

D2D Federated Learning Model Heterogeneity Split Learning System Heterogeneity Artificial intelligence Learning systems D2D Edge intelligence Learning approach Model bias Model heterogeneity Shared model Split learning System heterogeneity Training time User privacy Federated learning
Federated Learning (FL) is a promising framework for edge intelligence, allowing clients to preserve user privacy by collaboratively training a shared model. However, in real-world scenarios, heterogeneous resource challenges, such as the straggler effect among devices, degrade training efficiency. Even worse, existing approaches, such as model heterogeneity that rely heavily on contributions from strong devices, tend to trigger model bias issues. In contrast, resource-constrained devices are only able to train the initial few layers, eventually leading to uneven parameter training. In this paper, we propose a HeteroSplit framework that enables all devices to iteratively train the full depths of the model, thereby addressing the FL training challenges mentioned above. We design a Progressive Least Contribution First training strategy based on D2D-assisted split learning to dynamically balance each device's contribution and prevent model bias from strong devices. Additionally, we introduce an early transmission mechanism to accelerate overall training time. The experiments show that HeteroSplit improves accuracy by 15% and reduces training time by 41% compared to other state-of-the-art baselines under various device distributions and Non-IID data settings. © 2025 IEEE.

檔案與連結 (1)

url
https://www.scopus.com/inward/record.uri?eid=2-s2.0-105036264788&doi=10.1109%2fGLOBECOM59602.2025.11432770&partnerID=40&md5=89a54bc8b548a55305fcd5a6577d1379檢視

相關連結

指標

1 檢視次數

詳細資料

Logo image