Metadata Clustering-Driven Federated Learning for Multi-Domain Virtual Network Function Scaling

Abstract

In multi-domain networking, virtual network function (VNF) scaling using machine learning requires an accurate prediction model while addressing privacy constraints and non-identical and independently distributed (non-IID) data across domains. Current models have used conventional federated learning (FL) methods, such as federated averaging (FedAvg), yet they suffer from degraded performance due to heterogeneous traffic patterns in multi-domain networks. However, existing studies have not addressed the impact of non-IID characteristics on FL-based VNF scaling or developed an effective solution to mitigate it. This paper proposes a metadata-clustering-driven FL method that clusters domains with different traffic patterns and trains cluster-specific models. We extract statistical, spectral, and temporal features to represent traffic disturbance. We apply principal component analysis (PCA) followed by K-means clustering to group time series. We apply FedAvg within clusters to train cluster-specific prediction models. To evaluate the performance of the proposed method, we set up a testbench to synchronize three non-IID patterns. The numerical results demonstrate that the proposed clustered FL method consistently achieves a lower mean squared error (MSE) than the FedAvg baseline across all four evaluated non-IID settings. The proposed method yields an MSE of 0.7056 (a 23.1% reduction from FedAvg’s 0.9176) under label skew, 0.3615 (a 4.6% reduction from 0.3790) under label and feature skew, 0.6958 (a 28.1% reduction from 0.9682) under label and quantity skew, and 0.3748 (a 0.7% reduction from 0.3774) under the combined skew setting. These consistent reductions in MSE demonstrate that the proposed method effectively mitigates the performance degradation typically caused by non-IID effects.