Business Purchase Prediction Based on XAI and LSTM Neural Networks

Abstract

The black-box nature of neural networks is an obstacle to the adoption of systems based on them, mainly due to a lack of understanding and trust by end users. Providing explanations of the model’s predictions should increase trust in the system and make peculiar decisions easier to examine. In this paper, an architecture of a machine learning time series prediction system for business purchase prediction based on neural networks and enhanced with Explainable artificial intelligence (XAI) techniques is proposed. The architecture is implemented on an example of a system for predicting the following purchases for time series using Long short-term memory (LSTM) neural networks and Shapley additive explanations (SHAP) values. The developed system was evaluated with three different LSTM neural networks for predicting the next purchase day, with the most complex network producing the best results across all metrics. Explanations generated by the XAI module are provided with the prediction results to the user to allow him to understand the system’s decisions. Another benefit of the XAI module is the possibility to experiment with different prediction models and compare input feature effects.