CoLES: Contrastive Learning for Event Sequences with Self-Supervision

Autor: Babaev, Dmitrii, Kireev, Ivan, Ovsov, Nikita, Ivanova, Mariya, Gusev, Gleb, Nazarov, Ivan, Tuzhilin, Alexander
Rok vydání: 2020
Předmět:
Druh dokumentu: Working Paper
DOI: 10.1145/3514221.3526129
Popis: We address the problem of self-supervised learning on discrete event sequences generated by real-world users. Self-supervised learning incorporates complex information from the raw data in low-dimensional fixed-length vector representations that could be easily applied in various downstream machine learning tasks. In this paper, we propose a new method "CoLES", which adapts contrastive learning, previously used for audio and computer vision domains, to the discrete event sequences domain in a self-supervised setting. We deployed CoLES embeddings based on sequences of transactions at the large European financial services company. Usage of CoLES embeddings significantly improves the performance of the pre-existing models on downstream tasks and produces significant financial gains, measured in hundreds of millions of dollars yearly. We also evaluated CoLES on several public event sequences datasets and showed that CoLES representations consistently outperform other methods on different downstream tasks.
Comment: SIGMOD'22
Databáze: arXiv