Security-enhanced machine learning framework based on PATE

Privacy aggregated teacher ensembles (PATE) is a general machine learning framework that provides privacy-preserving for training data. However, this framework faces security risks in the distributed learning environment. Firstly, the involvement of illicit nodes in communication may lead to aggregation result inaccuracies. Secondly, the semi-honest aggregator and teacher nodes could potentially result in privacy leaks of other teacher nodes. Thirdly, the aggregation results are influenced by each teacher, and there may be poisoning attacks during the aggregation process. Fourthly, malicious aggregator may tamper with the information sent to student nodes or attempt to access relevant information about student node training labels. To address the above issues, we propose a machine learning framework with stronger security and privacy in a distributed learning environment based on principal component analysis and secures multi-party computing. The framework is subjected to security analysis and experimental validation. The security analysis establishes the framework's robustness and privacy-preserving characteristics, while experimental validation demonstrates its practical viability.