Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Soft Contrastive Learning for Time Series

About

Contrastive learning has shown to be effective to learn representations from time series in a self-supervised way. However, contrasting similar time series instances or values from adjacent timestamps within a time series leads to ignore their inherent correlations, which results in deteriorating the quality of learned representations. To address this issue, we propose SoftCLT, a simple yet effective soft contrastive learning strategy for time series. This is achieved by introducing instance-wise and temporal contrastive loss with soft assignments ranging from zero to one. Specifically, we define soft assignments for 1) instance-wise contrastive loss by the distance between time series on the data space, and 2) temporal contrastive loss by the difference of timestamps. SoftCLT is a plug-and-play method for time series contrastive learning that improves the quality of learned representations without bells and whistles. In experiments, we demonstrate that SoftCLT consistently improves the performance in various downstream tasks including classification, semi-supervised learning, transfer learning, and anomaly detection, showing state-of-the-art performance. Code is available at this repository: https://github.com/seunghan96/softclt.

Seunghan Lee, Taeyoung Park, Kibok Lee• 2023

Related benchmarks

TaskDatasetResultRank
Remaining Useful Life EstimationC-MAPSS FD002 (test)
RMSE17.72
44
Remaining Useful Life predictionNASA C-MAPSS FD001 (test)
RMSE11.56
41
Time-series classificationPAMAP2
Accuracy71.38
40
Time-series classificationHarth
Accuracy90.13
30
Time-series classificationSKODA
Accuracy98.82
30
Time-series classificationsleep
Accuracy84.85
30
Time-series classificationWISDM 2
Accuracy62.29
30
Time-series classificationUCR Archive all datasets
Wins64
21
Remaining Useful Life EstimationC-MAPSS FD003 (test)
RMSE12.21
21
Remaining Useful Life EstimationC-MAPSS FD004 (test)
RMSE19.75
21
Showing 10 of 16 rows

Other info

Follow for update