Vision Transformer Approach to Customer Churn Prediction Radar Chart Image Classification for Non-subscription Based E-commerce

Open Access
Authors
Publication date 2025
Host editors
  • Pari Delir Haghighi
  • Michal Greguš
  • Gabriele Kotsis
  • Ismail Khalil
Book title Information Integration and Web Intelligence
Book subtitle 26th International Conference, iiWAS 2024, Bratislava, Slovak Republic, December 2–4, 2024 : proceedings
ISBN
  • 9783031780929
ISBN (electronic)
  • 9783031780936
Series Lecture Notes in Computer Science
Event 26th International Conference on Information Integration and Web Intelligence, iiWAS 2024
Volume | Issue number II
Pages (from-to) 75–80
Publisher Cham: Springer
Organisations
  • Faculty of Science (FNWI) - Informatics Institute (IVI)
Abstract

Having a ready-to-use dependent churn variable in the non-subscription-based e-commerce domain is not straightforward, as customers do not formally cancel their status as a customer but simply stop placing orders. In this study, a k-means clustering algorithm with RFM variable segmentation is used to assist in classifying customers as churned or not churned. Next, a novel approach is adopted by transforming tabular customer data into radar chart images which are then fed into a pre-trained Vision Transformer model for supervised image classification. The results of the study indicate that while the Vision Transformer model show significant increase in precision and F1-score evaluation metrics, its performance in MCC, recall, and AUCROC is comparable to that of CatBoost and XGBoost. This study demonstrates the potential of transforming tabular data into images for training Vision Transformers, for use in customer churn prediction within the non-subscription-based e-commerce domain.

Document type Conference contribution
Language English
Published at https://doi.org/10.1007/978-3-031-78093-6_6
Downloads
978-3-031-78093-6_6 (Final published version)
Permalink to this page
Back