English

Pretext Training Algorithms for Event Sequence Data

Machine Learning 2024-02-19 v1 Artificial Intelligence

Abstract

Pretext training followed by task-specific fine-tuning has been a successful approach in vision and language domains. This paper proposes a self-supervised pretext training framework tailored to event sequence data. We introduce a novel alignment verification task that is specialized to event sequences, building on good practices in masked reconstruction and contrastive learning. Our pretext tasks unlock foundational representations that are generalizable across different down-stream tasks, including next-event prediction for temporal point process models, event sequence classification, and missing event interpolation. Experiments on popular public benchmarks demonstrate the potential of the proposed method across different tasks and data domains.

Keywords

Cite

@article{arxiv.2402.10392,
  title  = {Pretext Training Algorithms for Event Sequence Data},
  author = {Yimu Wang and He Zhao and Ruizhi Deng and Frederick Tung and Greg Mori},
  journal= {arXiv preprint arXiv:2402.10392},
  year   = {2024}
}