Abstract
As a main part of automated machine learning, Neural Architecture Search (NAS) has been employed in exploring effective Click-Through Rate (CTR) prediction models of recommender systems in recent studies. However, these studies update the architecture parameters by different strategies, such as gradient-based methods or evolutionary algorithms, and may suffer from either interference problems or high time-consuming in the search stage. Besides, the blocks in the search space were usually homogeneous, which means they have the same candidate operations to choose. Such design will result in redundancy in the search space, because many structures are inherently invalid and just increase the complexity of searching. To address the above issues, we implement the three-level automatic design of CTR prediction models using NAS via Shapley value, named as AutoShape. For the search space, we divide it into three parts according to the characteristics of the CTR model. Each part comprises distinctive candidate operations and forms a cell as an individual processing level, which improves the stability of the searched models’ effect and reduces the computational complexity. For the search strategy, we leverage Shapley value, a metric derived from cooperative game theory, to estimate the contribution of the operations in each block and the connections between the blocks, which can find effective models and reduce the time cost. Furthermore, experiments on different benchmark datasets show that the structure obtained from the search performs better than several hand-crafted architectures and is more stable than another NAS-based algorithm.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Liu, B., Zhu, C., Li, G.: AutoFIS: Automatic feature interaction selection in factorization models for click-through rate prediction. In: Proceedings of the 26th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining, pp. 2636–2645 (2020)
Zheng, R., Qu, L., Cui, B., et al.: AutoML for deep recommender systems: a survey. ACM Trans. Inform. Syst. (2023)
Wan, X., Ru, B., Esperança, P. M., Li, Z.: On redundancy and diversity in cell-based neural architecture search. arXiv preprint arXiv:2203.08887 (2022)
Meng, Z., Zhang, J., Li, Y., et al.: A general method for automatic discovery of powerful interactions in click-through rate prediction. In: 44th International ACM SIGIR Conference on Research and Development in Information Retrieval, pp. 1298–1307 (2021)
Zhu, G., Cheng, F., Lian, D., et al.: NAS-CTR: efficient neural architecture search for click-through rate prediction. In Proceedings of the 45th International ACM SIGIR Conference on Research and Development in Information Retrieval, pp. 332–342 (2022)
Xiao, H., Wang, Z., Zhu, Z., et al.: Shapley-NAS: discovering operation contribution for neural architecture search. In: IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 11892–11901 (2022)
Rendle, S.: Factorization machines. In: 2010 IEEE International conference on data mining, pp. 995–1000 (2010)
Qu, Y., Cai, H., Ren, K.: Product-based neural networks for user response prediction. In: 16th international conference on data mining (ICDM), pp. 1149–1154 (2016)
Wang, R., Fu, B., Fu, G., et al.: Deep & cross network for ad click predictions. In: Proceedings of the ADKDD 2017, pp. 1–7 (2017)
Lian, J., Zhou, X., Zhang, F., et al.: xdeepfm: Combining explicit and implicit feature interactions for recommender systems. In: 24th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining, pp. 1754–1763 (2018)
He, X., Chua, T.S.: Neural factorization machines for sparse predictive analytics. In: 40th International ACM SIGIR conference on Research and Development in Information Retrieval, pp. 355–364 (2017)
Huang, T., Zhang, Z., Zhang, J.: FiBiNET: combining feature importance and bilinear feature interaction for click-through rate prediction. In:13th ACM Conference on Recommender Systems, pp. 169–177 (2019)
Song, Q., Cheng, D., Zhou, H.: Towards automated neural interaction discovery for click-through rate prediction. In: 26th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining, pp. 945–955 (2020)
Zhang, W., Du, T., Wang, J.: Deep learning over multi-field categorical data: a case study on user response prediction. In: 38th European Conference on IR Research, pp. 45–57. Springer International Publishing (2016). https://doi.org/10.1007/978-3-319-30671-1_4
Chen, B., Wang, Y., Liu, Z.: Enhancing explicit and implicit feature interactions via information sharing for parallel deep CTR models. In: 30th ACM international Conference on Information & Knowledge Management, pp. 3757–3766 (2021)
Cheng, H. T., Koc, L., Harmsen, J.: Wide & deep learning for recommender systems. In: 1st Workshop on Deep Learning for Recommender Systems, pp.7–10 (2016)
Guo, H., Tang, R., Ye, Y.: DeepFM: a factorization-machine based neural network for CTR prediction. arXiv preprint arXiv:1703.04247 (2017)
Acknowledgement
This work was supported by the National Natural Science Foundation of China (Nos. 62077038, 61672405, 62176196 and 62271374).
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2024 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Fang, Y., Mu, C., Liu, Y. (2024). AutoShape: Automatic Design of Click-Through Rate Prediction Models Using Shapley Value. In: Liu, F., Sadanandan, A.A., Pham, D.N., Mursanto, P., Lukose, D. (eds) PRICAI 2023: Trends in Artificial Intelligence. PRICAI 2023. Lecture Notes in Computer Science(), vol 14326. Springer, Singapore. https://doi.org/10.1007/978-981-99-7022-3_3
Download citation
DOI: https://doi.org/10.1007/978-981-99-7022-3_3
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-99-7021-6
Online ISBN: 978-981-99-7022-3
eBook Packages: Computer ScienceComputer Science (R0)