RATSF: Empowering Customer Service Volume Management through Retrieval-Augmented Time-Series Forecasting

Tianfeng Wang,Gaojie Cui
2024-06-16
Abstract:An efficient customer service management system hinges on precise forecasting of service volume. In this scenario, where data non-stationarity is pronounced, successful forecasting heavily relies on identifying and leveraging similar historical data rather than merely summarizing periodic patterns. Existing models based on RNN or Transformer architectures may struggle with this flexible and effective utilization. To tackle this challenge, we initially developed the Time Series Knowledge Base (TSKB) with an advanced indexing system for efficient historical data retrieval. We also developed the Retrieval Augmented Cross-Attention (RACA) module, a variant of the cross-attention mechanism within Transformer's decoder layers, designed to be seamlessly integrated into the vanilla Transformer architecture to assimilate key historical data segments. The synergy between TSKB and RACA forms the backbone of our Retrieval-Augmented Time Series Forecasting (RATSF) framework. Based on the above two components, RATSF not only significantly enhances performance in the context of Fliggy hotel service volume forecasting but also adapts flexibly to various scenarios and integrates with a multitude of Transformer variants for time-series forecasting. Extensive experimentation has validated the effectiveness and generalizability of this system design across multiple diverse contexts.
Machine Learning,Information Retrieval
What problem does this paper attempt to address?
The paper primarily aims to address key challenges in customer service quality prediction, particularly in improving prediction accuracy in scenarios where data non-stationarity is significant. Specifically, the paper proposes a new framework called RATSF (Retrieval-Augmented Time-Series Forecasting) to tackle the following issues: 1. **Limitations of Existing Models**: Traditional time-series forecasting models based on RNN or Transformer architectures have limitations in utilizing historical data, making it difficult to effectively identify and leverage similar historical data patterns. 2. **Handling Non-Stationary Data**: In scenarios such as hotel service volume forecasting, data is influenced by various factors like order status, weather, and international events, leading to non-stationary characteristics. This makes it challenging for traditional univariate time-series forecasting methods to be effectively applied. 3. **Need for Flexibility and Adaptability**: There is a need for a general and flexible time-series forecasting system design that can efficiently utilize historical sequence information to address complex forecasting challenges. To solve the above problems, the paper contributes in the following aspects: - **Time-Series Knowledge Base (TSKB)**: Developed a time-series knowledge base for efficient storage and indexing of historical data. It employs an advanced indexing system that can quickly retrieve relevant historical data segments. - **Retrieval-Augmented Cross-Attention (RACA) Module**: This is a variant of the cross-attention mechanism in the Transformer decoder layer, designed to be seamlessly integrated into the standard Transformer architecture to facilitate the integration of key historical data segments. - **RATSF Framework**: The combination of TSKB and RACA forms the core of the RATSF framework, which not only significantly improves the performance of Fliggy hotel service volume forecasting but also flexibly adapts to various scenarios and integrates with multiple Transformer variants for time-series forecasting. Experimental results show that in real-world application scenarios (such as Fliggy hotel service volume forecasting) and multiple public datasets, RATSF demonstrates significant performance improvements compared to baselines and other methods. For example, in the Fliggy application, the prediction error was reduced from approximately 15% to about 8%, significantly lowering personnel management costs. Additionally, the paper validates the effectiveness and importance of each component of RATSF through a series of ablation studies.