More about HKUST
Secure Forward Aggregation for Vertical Federated Neural Networks
MPhil Thesis Defence Title: "Secure Forward Aggregation for Vertical Federated Neural Networks" By Mr. Shuowei CAI Abstract Vertical federated learning (VFL) is attracting much attention because it enables cross-silo data cooperation in a privacy-preserving manner. While most research works in VFL focus on linear and tree models, deep models (e.g., neural networks) are not well studied in VFL. In this thesis, we focus on SplitNN, a well-known neural network framework in VFL, and identify a trade-off between data security and model performance in SplitNN. Briefly, SplitNN trains the model by exchanging gradients and transformed data. On the one hand, SplitNN suffers from the loss of model performance since multiply parties jointly train the model using transformed data instead of raw data, and a large amount of low-level feature information is discarded. On the other hand, a naive solution of increasing the model performance through aggregating at lower layers in SplitNN (i.e., the data is less transformed and more low-level feature is preserved) makes raw data vulnerable to inference attacks. To mitigate the above trade-off, we propose a new neural network protocol in VFL called Security Forward Aggregation (SFA). It changes the way of aggregating the transformed data and adopts removable masks to protect the raw data. Experiment results show that networks with SFA achieve both data security and high model performance. Date: Thursday, 16 March 2023 Time: 4:00pm - 6:00pm Venue: Room 5501 lifts 25/26 Committee Members: Prof. Kai Chen (Supervisor) Dr. Yangqiu Song (Chairperson) Dr. Hao Liu (EMIA) **** ALL are Welcome ****