Build Yourself before Collaboration: Vertical Federated Learning with Limited Aligned Samples

Wei Shen, Mang Ye*, Wei Yu*, Pong C. Yuen

*Corresponding author for this work

Research output: Contribution to journalJournal articlepeer-review

Abstract

Vertical Federated Learning (VFL) has emerged as a crucial privacy-preserving learning paradigm that involves training models using distributed features from shared samples. However, the performance of VFL can be hindered when the number of shared or aligned samples is limited, a common issue in mobile environments where user data are diverse and unaligned across multiple devices. Existing approaches use feature generation and pseudo-label estimation for unaligned samples to address this issue, unavoidably introducing noise during the generation process. In this work, we propose Local Enhanced Effective Vertical Federated Learning (LEEF-VFL), which fully utilizes unaligned samples in the local learning before collaboration. Unlike previous methods that overlook private labels owned by each client, we leverage these private labels to learn from all local samples, constructing robust local models to serve as solid foundations for collaborative learning. Additionally, we reveal that the limited number of aligned samples introduces distribution bias from global data distribution. In this case, we propose to minimize the distribution discrepancies between the aligned samples and the global data distribution to enhance collaboration. Extensive experiments demonstrate the effectiveness of LEEF-VFL in addressing the challenges of limited aligned samples, making it suitable for VFL in mobile computing environments.
Original languageEnglish
Pages (from-to)6503-6516
Number of pages14
JournalIEEE Transactions on Mobile Computing
Volume24
Issue number7
Early online date20 Feb 2025
DOIs
Publication statusPublished - Jul 2025

User-Defined Keywords

  • Mobile applications
  • security
  • vertical federated learning

Fingerprint

Dive into the research topics of 'Build Yourself before Collaboration: Vertical Federated Learning with Limited Aligned Samples'. Together they form a unique fingerprint.

Cite this