Deep Latent Variable Model based Vertical Federated Learning with Flexible Alignment and Labeling Scenarios
Kihun Hong, Sejun Park, Ganguk Hwang
Abstract
Federated learning (FL) has attracted significant attention for enabling collaborative learning without exposing private data. Among the primary variants of FL, vertical federated learning (VFL) addresses feature-partitioned data held by multiple institutions, each holding complementary information for the same set of users. However, existing VFL methods often impose restrictive assumptions such as a small number of participating parties, fully aligned data, or only using labeled data. In this work, we reinterpret alignment gaps in VFL as missing data problems and propose a unified framework that accommodates both training and inference under arbitrary alignment and labeling scenarios, while supporting diverse missingness mechanisms. In the experiments on 168 configurations spanning four benchmark datasets, six training-time missingness patterns, and seven testing-time missingness patterns, our method outperforms all baselines in 160 cases with an average gap of 9.6 percentage points over the next-best competitors. To the best of our knowledge, this is the first VFL framework to jointly handle arbitrary data alignment, unlabeled data, and multi-party collaboration all at once.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 003573d0-5d02-4d6f-82d0-7ddb83b2933dBuilds on8
- not-MIWAE: Deep Generative Modelling with Missing not at Random DataNiels Bruun Ipsen, Pierre-Alexandre Mattei, Jes FrellsenICLR 2021 · 81 citations
- BlindFL: Vertical Federated Machine Learning without Peeking into Your DataFangcheng Fu, Huanran Xue, Yong Cheng, Yangyu Tao et al.SIGMOD 2022 · 53 citations
- How to deal with missing data in supervised deep learning?Niels Bruun Ipsen, Pierre-Alexandre Mattei, Jes FrellsenICLR 2022 · 39 citations
- LESS-VFL: Communication-Efficient Feature Selection for Vertical Federated LearningTimothy Castiglia, Yi Zhou, Shiqiang Wang, Swanand Kadhe et al.ICML 2023 · 33 citations
- Federated Transformer: Multi-Party Vertical Federated Learning on Practical Fuzzily Linked DataZhaomin Wu, Junyi Hou, Yiqun Diao, Bingsheng HeNeurIPS 2024 · 16 citations
Related papers
- FedMix: Boosting with Data Mixture for Vertical Federated LearningYihang Cheng, Lan Zhang, Junyang Wang, Xiaokai Chu et al.ICDE 2024 · 6 citations
- VFLAIR: A Research Library and Benchmark for Vertical Federated LearningTianyuan Zou, Zixuan Gu, Yu He, Hideaki Takahashi et al.ICLR 2024 · 15 citations
- Vertical Federated Learning with Missing Features During Training and InferencePedro Valdeira, Shiqiang Wang, Yuejie ChiICLR 2025
- ReFer: Retrieval-Enhanced Vertical Federated Recommendation for Full Set User BenefitWenjie Li, Zhongren Wang, Jinpeng Wang, Shutao Xia et al.SIGIR 2024 · 3 citations
- FEZE: Alignment-Flexible Zero-Shot Vertical Federated LearningWei Guo, Yiqi Tong, Yiyang Duan, Fuzhen Zhuang et al.KDD 2025 · 2 citations
