Cascade Vertical Federated Learning
Document Type
Conference Proceeding
Publication Date
8-26-2022
Department
Department of Electrical and Computer Engineering; Department of Applied Computing
Abstract
Vertical federated learning (VFL) enables collaborative machine learning on vertically partitioned data with privacy-preservation, attracting widespread attentions from academia and industry. Most existing VFL methods face two daunting challenges in real-world applications. First, most VFL methods assume at least one party holds the complete set of labels of all data samples. However, this assumption often violates the nature of many scenarios, where the parties only have partial labels. Second, the limitation of computational and communication resources in participated parties may cause the straggler problem and slow down training convergence. To address these challenges, we propose a novel VFL algorithm named Cascade Vertical Federated Learning (CVFL), in which partitioned labels can be fully utilized to train neural networks. To mitigate the straggler problem, we design a novel optimization objective to increase straggler's contribution to the trained models. We conduct comprehensive experiments and the results demonstrate the effectiveness and efficiency of CVFL.
Publication Title
Proceedings - IEEE International Conference on Multimedia and Expo
ISBN
9781665485630
Recommended Citation
Xia, W.,
Li, Y.,
Zhang, L.,
Wu, Z.,
&
Yuan, X.
(2022).
Cascade Vertical Federated Learning.
Proceedings - IEEE International Conference on Multimedia and Expo,
2022-July.
http://doi.org/10.1109/ICME52920.2022.9859921
Retrieved from: https://digitalcommons.mtu.edu/michigantech-p/16384