Reliable Vertical Federated Learning in 5G Core Network Architecture

2025/05/21 by Mohamad Mestoukirdi, Mestoukirdi, Mohamad, Mourad Khanfouci +1 · 1 citation
Computer Science · #Cooperative Communication and Network Coding #FOS: Computer and information sciences #FOS: Electrical engineering #Machine Learning (cs.LG) #Privacy-Preserving Technologies in Data #Stochastic Gradient Optimization Techniques #Systems and Control (eess.SY) #electronic engineering #information engineering

paper · pdf · doi:10.48550/arxiv.2505.15244

openalex publication_date 2025/05/21 · openalex created_date 2025/10/19 · openalex updated_date 2026/07/28

Abstract

This work proposes a new algorithm to mitigate model generalization loss in Vertical Federated Learning (VFL) operating under client reliability constraints within 5G Core Networks (CNs). Recently studied and endorsed by 3GPP, VFL enables collaborative and load-balanced model training and inference across the CN. However, the performance of VFL significantly degrades when the Network Data Analytics Functions (NWDAFs) - which serve as primary clients for VFL model training and inference - experience reliability issues stemming from resource constraints and operational overhead. Unlike edge environments, CN environments adopt fundamentally different data management strategies, characterized by more centralized data orchestration capabilities. This presents opportunities to implement better distributed solutions that take full advantage of the CN data handling flexibility. Leveraging this flexibility, we propose a method that optimizes the vertical feature split among clients while centrally defining their local models based on reliability metrics. Our empirical evaluation demonstrates the effectiveness of our proposed algorithm, showing improved performance over traditional baseline methods.

Citations

• TreeCSS: An Efficient Framework for Vertical Federated Learning
• Robust and IP-Protecting Vertical Federated Learning against Unexpected Quitting of Parties
• Vertical Federated Learning: Concepts, Advances, and Challenges
• Flower: A Friendly Federated Learning Research Framework
• Federated Learning over Wireless Fading Channels
• A Unified Approach to Interpreting Model Predictions

Cited by

• Dynamic Features Adaptation in Networking: Toward Flexible training and Explainable inference

Related

• On-the-fly Resource-Aware Model Aggregation for Federated Learning in Heterogeneous Edge
• Vertical Federated Learning over Cloud-RAN: Convergence Analysis and System Optimization
• Federated Learning: Strategies for Improving Communication Efficiency
• Robust Federated Learning in Unreliable Wireless Networks: A Client Selection Approach
• How global observation works in Federated Learning: Integrating vertical training into Horizontal Federated Learning