Reliable Vertical Federated Learning in 5G Core Network Architecture

TL;DR

This paper introduces a new algorithm for Vertical Federated Learning in 5G Core Networks that enhances model reliability by optimizing feature splits among clients under reliability constraints.

Contribution

It presents a novel method leveraging 5G CN data management capabilities to improve VFL performance amid client reliability issues.

Findings

Improved model accuracy compared to baseline methods

Effective handling of client reliability constraints

Enhanced data utilization in 5G core network environments

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.