FedDeepFM: Ad CTR prediction based on Federated Factorization Machine

chenjia yu,shuhan qi,yang liu
DOI: https://doi.org/10.1109/DSC53577.2021.00034
2021-01-01
Abstract:Personal Advertising recommendation has become one of the main sources of income for Internet companies such as Google and Facebook. The prediction of ad click-through rate achieved through machine learning, deep learning and other technologies can effectively improve the accuracy of ad recommendation. However, the large amount of user data collected and used in this process is also constantly causing data security issues and user privacy leaks. In order to protect user privacy and fulfill the requirements of relevant laws and regulations, privacy-friendly machine learning methods such as federal learning have gradually attracted the attention of relevant scholars and practitioners. In this paper, we propose a framework of federated factorization machine, which is used to implement the function of advertising click-through rate prediction under the premise of protecting user privacy. First, we designed a user-level distributed factorization machine that only uses the gradient information of each client (instead of the original preference data) to update the model. Secondly, in order to solve the linear aggregation model loss caused by the heterogeneity of user data, we introduced the idea of cluster federation learning in the factorization machine, and designed the framework, FedDeepFM. Finally, we implemented the prototype of FedDeepFM and evaluated it with a real ad click data set. Experimental results show that the accuracy of our proposed framework is 8% higher than that of the traditional federated matrix factorization algorithm, and 2.5% higher than the federated learning accuracy of a single global model.
What problem does this paper attempt to address?