Source-linked AI summary
Multi-Participant Multi-Class Vertical Federated Learning
Siwei Feng, Han Yu
TL;DR
Existing VFL methods generally support only two participants and focus on binary classification, while VFL must enable use of labels held by one participant without compromising privacy. MMVFL extends multi-view learning to multi-participant, multi-class VFL with privacy-preserving label sharing; experiments report performance comparable to supervised counterparts in most cases.
Problem
Method
MMVFL extends multi-view learning to multi-participant, multi-class VFL, using separate participant models and privacy-preserving label sharing.
Source: §1 · p. 1Results
MMVFL’s performance was comparable to supervised counterparts in most cases; on Handwritten, it exceeded supFL by 1.42% and supMVLFL by 2.31% averaged over 5 participants.
Source: §6.4 · p. 5Takeaways & Limitations
MMVFL can share label information among multiple VFL participants while achieving classification performance comparable to existing approaches.
Source: §7 · p. 5Each MMVFL iteration requires communication among participants, and training time depends on the slowest participant in each round.
Source: §5.2 · p. 4
Abstract
from arXiv · showhide
Federated learning (FL) is a privacy-preserving paradigm for training collective machine learning models with locally stored data from multiple participants. Vertical federated learning (VFL) deals with the case where participants sharing the same sample ID space but having different feature spaces, while label information is owned by one participant. Current studies of VFL only support two participants, and mostly focus on binaryclass logistic regression problems. In this paper, we propose the Multi-participant Multi-class Vertical Federated Learning (MMVFL) framework for multi-class VFL problems involving multiple parties. Extending the idea of multi-view learning (MVL), MMVFL enables label sharing from its owner to other VFL participants in a privacypreserving manner. To demonstrate the effectiveness of MMVFL, a feature selection scheme is incorporated into MMVFL to compare its performance against supervised feature selection and MVL-based approaches. Experiment results on real-world datasets show that MMVFL can effectively share label information among multiple VFL participants and match multi-class classification performance of existing approaches.
1 Introduction
Federated learning addresses privacy concerns when data are distributed across organizations, while VFL brings together participants with shared sample IDs but different features and labels held by one participant. MMVFL targets the limits of prior VFL, adding multi-participant, multi-class label sharing and feature-importance evaluation; experiments report performance comparable to existing approaches.
1 Introduction
- Federated learning suits settings where data are isolated across organizations and privacy is emphasized, unlike approaches that centralize data and learning.The introduction frames this as a motivation for distributed, privacy-preserving machine learning.Source: §1 · p. 1
- VFL assumes participants share sample IDs but have different feature spaces, with label information held by one participant.A central challenge is enabling privacy-preserving use of that participant’s labels for training.Source: §1 · p. 1
- Prior VFL approaches generally handle only two participants and binary classification, limiting their suitability for complex multi-participant tasks.The introduction identifies these restrictions as a barrier to complex VFL applications.Source: §1 · p. 1
- MMVFL extends multi-view learning to multi-class VFL with multiple participants, separate participant models, and privacy-preserving label sharing.Data and labels remain with their owners during training.Source: §1 · p. 1
- MMVFL’s feature-importance scheme quantifies each participant’s feature contributions and can reduce later incremental-training communication, computation, and storage costs by discarding redundant or harmful features.The feature discarding is described for initial training periods.Source: §1 · pp. 1–2
- MMVFL’s multi-class classification performance matches existing approaches while sharing label information among multiple VFL participants.The framework is presented as the first VFL framework for multi-class problems with multiple participants.Source: §1 · p. 2
2 Related Work
Prior VFL research includes privacy-preserving logistic regression and methods addressing entity-resolution errors, communication, or coordinator risks. Yet existing approaches remain limited to two participants and mostly binary classification, whereas MMVFL targets multi-class, multi-participant settings.
2 Related Work
- Hardy et al. [Hardy et al., 2017] proposed federated logistic regression with additively homomorphic encryption and analyzed entity-resolution mistakes; Nock et al. [Nock et al., 2018] extended the error analysis across losses.Yang et al. [Yang et al., 2019a] reduced communication rounds, while Yang et al. [Yang et al., 2019d] removed the third-party coordinator.Source: §2 · p. 2
- Wang et al. [Wang et al., 2019] assessed feature-group importance with Shapley Values, whose computation has exponential complexity.The method dynamically removes feature groups to measure their impact on model performance.Source: §2 · p. 2
- Existing VFL approaches generally support only two participants and focus on binary classification, limiting complex multi-participant applications.MMVFL is designed for multi-class, multi-participant VFL settings.Source: §2 · p. 2
3 Preliminaries
Multi-view learning jointly optimizes models for views of the same samples, but raw-data interaction across views conflicts with federated privacy requirements. Feature selection chooses useful features for a task and can reduce communication costs.
Multi-View Learning
- Multi-view learning models each view with a function and jointly optimizes the functions to improve generalization.Its shared sample IDs and heterogeneous features fit VFL, but existing methods require raw data interaction across views, conflicting with privacy preservation.Source: Multi-View Learning · p. 2
Feature Selection
- Feature selection chooses a useful subset of dataset features for a learning task and can save federated communication costs by compressing data according to feature importance.This common practice measures feature importance and discards less important features.Source: Feature Selection · p. 2
4 The Proposed MMVFL Framework
MMVFL addresses multi-class VFL with multiple participants while keeping raw data, labels, and local models on their owners’ machines. It combines participant-level feature selection with a shared representation that supports privacy-preserving label transfer.
4 The Proposed MMVFL Framework
- Only locally predicted labels cross the privacy barrier; raw data, labels, and local models remain on their owners’ machines.Source: §4 · p. 2
4.1 Notations and Problem Definition
- The setting assumes K participants with local datasets for an Nc-class task, participant 1 holding the labels, and sample IDs already matched.Source: §4.1 · p. 2
4.2 Sparse Learning-based Unsupervised Feature Selection
- For participants without labels, MMVFL learns a transformation and pseudo-label embedding; the row ℓ2-norms of the transformation matrix provide feature-importance scores.The linear transformation scheme is adopted as a proof of concept rather than an exhaustive choice of feature-selection methods.Source: §4.2 · pp. 2–3
4.3 Privacy-Preserving Label Sharing
- The VFL objective aligns participants’ local pseudo-label matrices with a shared matrix and anchors participant 1’s matrix to the true one-hot labels, enabling indirect label sharing.Source: §4.3 · p. 3
4.4 Optimization
- Large penalty terms relax Zk=Z and Z1=Y, while alternating updates address the ℓ2,1-regularized objective until convergence or the iteration limit.Source: §4.4 · p. 3
5 Analysis
The analysis establishes that MMVFL’s objective is non-increasing during optimization, characterizes per-round cost by the slowest participant, and describes which information is transmitted to preserve privacy.
5.1 Convergence
- MMVFL’s objective is consistently non-increasing during optimization, with each stated subproblem having a global minimum.The Z1, Zk, and Z subproblems are simple convex optimization problems when other parameters are fixed.Source: §5.1 · p. 4
5.2 Time Complexity
- Each federated iteration requires communication among all participants, and its time complexity depends on the participant with the largest feature dimension.The paper identifies the slowest participant in each round as the straggler.Source: §5.2 · p. 4
5.3 Privacy Preservation
- MMVFL transmits participants’ Zk values to the server while Xk and Y remain with their owners, supporting privacy-preserving label sharing.The paper states that the transformation matrices alone are insufficient to recover the original data, even if intercepted over multiple rounds.Source: §5.3 · p. 4
6 Experimental Evaluation
The evaluation compares MMVFL with supervised feature-selection baselines on Handwritten and Caltech7. MMVFL is comparable to those approaches, sometimes outperforming them, and in some cases performs similarly or better with fewer features.
6.1 Real-world Data
- Handwritten and Caltech7 provide five and six views, respectively; class counts are balanced across training and validation sets, and a six-feature Handwritten view was removed.The removed view had morphological features; the paper says its six features made feature selection insignificant.Source: §6.1 · p. 4, §6.2 · p. 4, Table 1 · p. 4
6.2 Comparison Baselines
- supFL selects features independently assuming all participants know labels; supMVLFL extends it to multi-view linear transformations, and neither preserves data privacy during information sharing.Both are described as distributed approaches capable of sharing information across participants.Source: §6.2 · p. 5
6.3 Experiment Settings
- Five-fold cross-validation uses four folds for training, tunes algorithms at each selected-feature percentage, and reports mean classification accuracy for each view and dataset.The selected-feature percentages range from 2% to 100%.Source: §6.3 · p. 5
6.4 Results and Discussion
- In some cases, MMVFL matches or exceeds approaches using all features while using fewer important features, reducing resource needs especially in incremental-learning settings.The stated resources include communication bandwidth, computing devices, and memory space.Source: §6.4 · p. 5
- Across five Handwritten participants, MMVFL beats supFL by 1.42% and supMVLFL by 2.31%; across six Caltech7 participants, it trails them by 1.21% and 0.88%, respectively.The reported differences are averaged over participants; Table 2 gives differences across selected-feature choices.Source: Table 2 · p. 5, §6.4 · p. 5
7 Conclusions and Future Work
MMVFL shares labels from their owner with other participants without data leakage and extends VFL to more complex scenarios. Feature-selection experiments report performance comparable to supervised counterparts, while future work targets classifiers, communication, stragglers, and cross-participant task relationships.
- Feature-selection experiments show MMVFL achieves performance comparable to its supervised counterparts.Source: §7 · p. 5
- MMVFL shares owner-held labels with other participants without data leakage and supports more complex scenarios than existing two-participant techniques.The authors describe it as the first attempt to transfer multi-view learning into VFL.Source: §7 · p. 5
- Future work targets more sophisticated classifiers, communication efficiency, straggler handling, and the effect of relationships across participants' tasks on overall model performance.Source: §7 · p. 5
- Figures 3 and 4 plot classification performance against the percentage of selected features on Handwritten and Caltech7, respectively.Source: Figure 3 · p. 6, Figure 4 · p. 6