Ranking Neighborhood and Class Prototype Contrastive Learning for Time Series

Chixuan Wei,Jidong Yuan,Yi Zhang,Zhongyang Yu,Yanze Liu,Haiyang Liu
DOI: https://doi.org/10.1109/tbdata.2024.3495509
2024-01-01
IEEE Transactions on Big Data
Abstract:Time series are often complex and rich in information but sparsely labeled and therefore challenging to model. Existing contrastive learning methods conduct augmentations and maximize their similarity. However, they ignore the similarity of adjacent timestamps and suffer from the problem of sampling bias. In this paper, we propose a self-supervised framework for learning generalizable representations of time series, called $\mathbf {R}$ anking n $\mathbf {E}$ ighborhood and cla $\mathbf {S}$ s prototyp $\mathbf {E}$ contr $\mathbf {A}$ stive $\mathbf {L}$ earning (RESEAL). It exploits information about similarity ranking to learn an embedding space, ensuring that positive samples are ranked according to their temporal order. Additionally, RESEAL introduces a class prototype contrastive learning module. It contrasts time series representations and their corresponding centroids as positives against truly negative pairs from different clusters, mitigating the sampling bias issue. Extensive experiments conducted on several multivariate and univariate time series tasks (i.e., classification, anomaly detection, and forecasting) demonstrate that our representation framework achieves significant improvement over existing baselines of self-supervised time series representation
What problem does this paper attempt to address?