Customer Segmentation and Personalized Recommendation Based on Machine Learning

Authors

  • Lingyun Li

DOI:

https://doi.org/10.54097/nwdtkx22

Keywords:

Customer Segmentation, Personalized Recommendation, Machine Learning, Deep Learning, Recommendation System.

Abstract

Today, with the rapid development of digitalization and networking, enterprises and institutions can collect customer data on an unprecedented scale, including demographic information, consumption behavior records, interaction logs, and social media information. These data provide a rich foundation for in-depth understanding of customer behavior and prediction of future demands. However, in the face of high-dimensional, multi-modal and dynamic data, how to effectively extract information, conduct refined customer segmentation and provide efficient personalized recommendations has become a research hotspot of common concern in both academia and industry. This paper systematically reviews the research progress of machine learning in customer segmentation and personalized recommendation in the past five years (2020-2025), covering aspects such as data feature construction, unsupervised and supervised clustering methods, gradient boosting tree models (LightGBM, CatBoost), deep neural networks (Transformer, GNN), core architectures and hybrid strategies of recommendation systems, model evaluation methods, interpretability and privacy protection. At the same time, it combines actual cases from industries such as retail, e-commerce, finance and healthcare to analyze the application effects and challenges of algorithms in real business, and looks forward to future development trends such as real-time personalization, multi-modal fusion, federated learning and green AI.

References

[1]Arthur, D. and Vassilvitskii, S.: “k-means++: The advantages of careful seeding,” in Proc. 18th Annual ACM-SIAM Symposium on Discrete Algorithms (SODA), 2007, pp. 1027–1035.

[2]Prokhorenkova, L., Gusev, G., Vorobev, A., Dorogush, A. and Gulin, A.: “CatBoost: unbiased boosting with categorical features,” in Proc. 32nd Int. Conf. on Neural Information Processing Systems (NeurIPS), 2018, pp. 6638–6648.

[3]Hamilton, W., Ying, Z. and Leskovec, J.: “Inductive representation learning on large graphs,” in Proc. 31st Int. Conf. on Neural Information Processing Systems (NeurIPS), 2017, pp. 1025–1035.

[4]Sun, F., Liu, J., Wu, Z. and Wang, H.: “BERT4Rec: Sequential recommendation with bidirectional encoder representations from transformer,” in Proc. 28th ACM Int. Conf. on Information and Knowledge Management (CIKM), 2019, pp. 1441–1450.

[5]He, K., Zhang, X., Ren, S. and Sun, J: “Deep residual learning for image recognition,” in Proc. IEEE Conf. on Computer Vision and Pattern Recognition (CVPR), 2016, pp. 770–778.

[6]Zhang, F., Duan, J. and Chen, Z.: “Feature engineering for purchase prediction in e-commerce,” Expert Systems with Applications, vol. 95, pp. 30–42, 2018.

[7]Chen, T. and Guestrin, C.: “XGBoost: A scalable tree boosting system,” in Proc. 22nd ACM SIGKDD Int. Conf. on Knowledge Discovery and Data Mining, 2016, pp. 785–794.

[8]Ke, G., Meng, Q., Finley, T., Wang, T., Chen, W., Ma, W., Ye, Q. and Liu, T. Y.: “LightGBM: A highly efficient gradient boosting decision tree,” in Proc. 31st Conf. on Neural Information Processing Systems (NeurIPS), 2017, pp. 3146–3154.

[9]He, X., Liao, L., Zhang, H., Nie, L., Hu, X. and Chua, T. S.: “Neural collaborative filtering,” in Proc. 26th Int. Conf. on World Wide Web (WWW), 2017, pp. 173–182.

[10][10] Ester, M., Kriegel, H.-P. Sander, J. and Xu, X.: “A density-based algorithm for discovering clusters in large spatial databases with noise,” in Proc. 2nd Int. Conf. on Knowledge Discovery and Data Mining (KDD), 1996, pp. 226–231.

[11]Mikolov, T., Chen, K., Corrado, G. and Dean, J.: “Efficient estimation of word representations in vector space,” in Proc. 1st Int. Conf. on Learning Representations (ICLR), 2013.

[12]Devlin, J., Chang, M. W., Lee, K. and Toutanova, K.: “BERT: Pre-training of deep bidirectional transformers for language understanding,” in Proc. 2019 Conf. of the North American Chapter of the Association for Computational Linguistics (NAACL), 2019, pp. 4171–4186.

[13]Cui, S., Yang, X. and Guo, Y.: “User interest modeling with time decay for personalized recommendation,” Information Sciences, vol. 512, pp. 1106–1122, 2020.

[14]Cheng, H., Koc, L., Harmsen, J. et al.: “Wide & deep learning for recommender systems,” in Proc. 1st Workshop on Deep Learning for Recommender Systems, 2016, pp. 7–10.

[15]Hutter, F., Kotthoff, L. and Vanschoren, J.: Automated Machine Learning: Methods, Systems, Challenges. Springer, 2019.

[16]Liu, Z., Zhang, Y. and Yang, Q.: “Feature engineering for large-scale click-through rate prediction in Alibaba,” Data Mining and Knowledge Discovery, vol. 34, no. 3, pp. 1079–1099, 2020.

[17]Chen, J., Liu, M. and Liu, K.: “Real-time feature engineering for large-scale recommendation systems at ByteDance,” IEEE Transactions on Big Data, vol. 7, no. 4, pp. 852–863, 2021.

[18]Kingma, D. P. and Welling, M.: “Auto-encoding variational Bayes,” in Proc. 2nd Int. Conf. on Learning Representations (ICLR), 2014.

[19]Ying, R., He, R., Chen, K., Eksombatchai, P., Hamilton, W. L. and Leskovec, J.: “Graph convolutionnal neural networks for web-scale recommender systems,” in Proc. 24th ACM SIGKDD Int. Conf. on Knowledge Discovery & Data Mining (KDD), 2018, pp. 974–983.

[20]Sarwar, B., Karypis, G., Konstan, J. and Riedl, J.: “Item-based collaborative filtering recommendation algorithms,” in Proc. 10th Int. Conf. on World Wide Web (WWW), 2001, pp. 285–295.

[21]Koren, Y.: “Factorization meets the neighborhood: a multifaceted collaborative filtering model,” in Proc. 14th ACM SIGKDD Int. Conf. on Knowledge Discovery and Data Mining, 2008, pp. 426–434.

[22]Adomavicius, G. and Tuzhilin, A.: “Toward the next generation of recommender systems: A survey of the state-of-the-art and possible extensions,” IEEE Transactions on Knowledge and Data Engineering, vol. 17, no. 6, pp. 734–749, 2005.

[23]Lundberg, S. and Lee, S. I.: “A unified approach to interpreting model predictions,” in Proc. 31st Conf. on Neural Information Processing Systems (NeurIPS), 2017, pp. 4765–4774.

[24]Finn, C., Abbeel, P. and Levine, S.: “Model-agnostic meta-learning for fast adaptation of deep networks,” in Proc. 34th Int. Conf. on Machine Learning (ICML), 2017, pp. 1126–1135.

Downloads

Published

15-03-2026

Issue

Section

Articles

How to Cite

Li, L. (2026). Customer Segmentation and Personalized Recommendation Based on Machine Learning. Mathematical Modeling and Algorithm Application, 9(1), 196-205. https://doi.org/10.54097/nwdtkx22