Privacy-preserving, Efficient, and Effective Machine Learning

Chuhan Wu,Fangzhao Wu,Tao Qi,Yongfeng Huang,Xing Xie
DOI: https://doi.org/10.21203/rs.3.rs-1682972/v1
2022-01-01
Abstract:Abstract Privacy protection is critical for responsible artificial intelligence. Federated learning is a privacy-aware machine learning paradigm, which is often combined with differential privacy to guarantee privacy protection. Unfortunately, existing methods cannot achieve a satisfactory tradeoff between privacy and utility when the models are large, and their computation and communication costs are also huge. Here, we present a differentially private, efficient, and effective machine learning method named FedPrompt to learn big models in a federated way via prompt tuning. It only learns, perturbs, and exchanges the small prompt models injected into the big models. FedPrompt is validated on five datasets. The results show FedPrompt can achieve 0.5%~34.7% better performance than standard federated learning under same privacy budgets, meanwhile saving 99% of communication cost, 75% of memory, and 64% of training time. FedPrompt offers a new direction to efficiently and effectively distributed machine learning with privacy guarantees.
What problem does this paper attempt to address?