DUET: Dual Clustering Enhanced Multivariate Time Series Forecasting
Xiangfei Qiu, Xingjian Wu, Yan Lin, Chenjuan Guo, Jilin Hu, Bin Yang
TL;DR
DUET addresses two core challenges in multivariate time series forecasting: heterogeneous temporal patterns from Temporal Distribution Shift and complex, intertwined channel relationships. It introduces a Temporal Clustering Module to partition time series into distributions and apply distribution-specific pattern extractors, and a Channel Clustering Module that learns frequency-domain channel relations with sparsification via a mask. The Fusion Module combines these components through a masked attention mechanism, enabling robust forecasting. Extensive experiments on 25 real-world datasets across 10 domains demonstrate state-of-the-art performance, highlighting DUET's effectiveness in handling temporal heterogeneity and channel noise, with publicly available code for reproducibility.
Abstract
Multivariate time series forecasting is crucial for various applications, such as financial investment, energy management, weather forecasting, and traffic optimization. However, accurate forecasting is challenging due to two main factors. First, real-world time series often show heterogeneous temporal patterns caused by distribution shifts over time. Second, correlations among channels are complex and intertwined, making it hard to model the interactions among channels precisely and flexibly. In this study, we address these challenges by proposing a general framework called DUET, which introduces dual clustering on the temporal and channel dimensions to enhance multivariate time series forecasting. First, we design a Temporal Clustering Module (TCM) that clusters time series into fine-grained distributions to handle heterogeneous temporal patterns. For different distribution clusters, we design various pattern extractors to capture their intrinsic temporal patterns, thus modeling the heterogeneity. Second, we introduce a novel Channel-Soft-Clustering strategy and design a Channel Clustering Module (CCM), which captures the relationships among channels in the frequency domain through metric learning and applies sparsification to mitigate the adverse effects of noisy channels. Finally, DUET combines TCM and CCM to incorporate both the temporal and channel dimensions. Extensive experiments on 25 real-world datasets from 10 application domains, demonstrate the state-of-the-art performance of DUET.
