Skip-VFL: Mitigating Extreme Class Imbalance in Feature-Heterogeneous Vertical Federated Learning for Intrusion Detection

Read full text
Authors: Alimov Abdulboriy Abdulkhay Ugli, Ji Sun Shin
Cited by: 0

Abstract

Distributed network environments increasingly rely on Vertical Federated Learning (VFL) to address the growing heterogeneity of device features, which conventional Federated Learning is ill-equipped to handle. While VFL facilitates collab- oration across diverse network entities in detecting anomalies, current models often remain impractical for long-tailed datasets, especially in network security domain, as they fail to account for the inherent class imbalances that bias models toward majority classes. Therefore, We propose Skip-VFL, a novel tiered framework for robust anomaly detection in heterogeneous networks. Skip- VFL organizes data into three categories: shared core fea- tures (Fcore), overlapping features (Foverlap), and unique features (Funique), bridging the gap between resource-constrained IoT devices and high-capacity servers. To address class imbalances, Skip-VFL introduces a two-stage training approach: Latent Space Alignment, which clusters feature representations, and Weighted Classification using Inverse Class Frequency (ICF) to increase sensitivity to rare anomalies. A key innovation is the Skip-Layer Mechanism, which preserves feature privacy by withholding tier-specific adapters (ϕk ) from the central server while enabling global model synchronization. Experiments on NSL-KDD, UNSW-NB15, and ToN-IoT datasets show that Skip-VFL outperforms baseline methods, achieving 88.4% recall on NSL-KDD in extreme 1% imbal- ance scenarios. Unlike peers, which struggles with instability, Skip-VFL maintains consistent performance, proving it to be a reliable, privacy-preserving solution for anomaly detection in decentralized networks.

Fields of Study & Subjects

Privacy-Preserving Technologies in Data Network Security and Intrusion Detection Internet Traffic Analysis and Secure E-voting

Publication details

Journal / source
Code Ocean
Publication date
1 January 2026
Language
EN