Proceedings

PyTorch-Lifestream: Learning Embeddings on Discrete Event Sequences

Download BibTeX Cite

Proceedings of the Thirty-Fourth International Joint Conference on Artificial Intelligence. Demo Track. Pages 11104-11108.

https://doi.org/10.24963/ijcai.2025/1272

Abstract

The domain of event sequences is widely applied in various industrial tasks in banking, healthcare, etc., where temporal tabular data processing is required. This paper introduces PyTorch-Lifestream, the first open-source library specially designed to handle event sequences. It supports scenarios with multimodal data and offers a variety of techniques for learning embeddings of event sequences and end-to-end model training. Furthermore, PyTorch-Lifestream efficiently implements state-of-the-art methods for event sequence analysis and adapts approaches from similar domains, thus enhancing the versatility and performance of sequence-based models for a wide range of applications, including financial risk scoring, campaigning, user ID matching, churn prediction, fraud detection, medical diagnostics, and recommender systems.

  1. [Alexandrov et al., 2020] Alexander Alexandrov, Kon- stantinos Benidis, Michael Bohlke-Schneider, Valentin Flunkert, Jan Gasthaus, Tim Januschowski, Danielle C. Maddix, Syama Rangapuram, David Salinas, Jasper Schulz, Lorenzo Stella, Ali Caner Türkmen, and Yuyang Wang. GluonTS: Probabilistic and Neural Time Series Modeling in Python. Journal of Machine Learning Research, 21(116):1-6, 2020.
  2. [Apache Arrow, 2016] Apache Software Foundation Apache Arrow. Apache arrow, 2016.
  3. [Babaev et al., 2022] Dmitrii Babaev, Nikita Ovsov, Ivan Kireev, Maria Ivanova, Gleb Gusev, Ivan Nazarov, and Alexander Tuzhilin. CoLES: Contrastive learning for event sequences with self-supervision. In Proceedings of the 2022 International Conference on Management of Data, SIGMOD/PODS '22. ACM, June 2022.
  4. [Bacry et al., 2017] E. Bacry, M. Bompaire, S. Gaïffas, and S. Poulsen. tick: a Python library for statistical learning, with a particular emphasis on time-dependent modeling. ArXiv e-prints, July 2017.
  5. [Baevski et al., 2020] Alexei Baevski, Henry Zhou, Abdel- rahman Mohamed, and Michael Auli. wav2vec 2.0: a framework for self-supervised learning of speech repre- sentations. In Proceedings of the 34th International Con- ference on Neural Information Processing Systems, NIPS '20, Red Hook, NY, USA, 2020.
  6. Curran Associates Inc. [Bardes et al., 2021]
  7. Adrien Bardes, Jean Ponce, and Yann LeCun. Vicreg: Variance-invariance-covariance regu- larization for self-supervised learning. arXiv preprint arXiv:2105.04906, 2021.
  8. [Bazarova et al., 2025] Alexandra Bazarova, Maria Koval- eva, Ilya Kuleshov, Evgenia Romanenkova, Alexander Stepikin, Aleksandr Yugay, Dzhambulat Mollaev, Ivan Kireev, Andrey Savchenko, and Alexey Zaytsev. Learning transactions representations for information management in banks: Mastering local, global, and external knowl- edge. International Journal of Information Management Data Insights, 5(1):100323, 2025.
  9. [Boosters, 2025a] Boosters. Alfa Battle 2 Overview, 2025. [Online; accessed 11-February-2025].
  10. [Boosters, 2025b] Boosters. Rosbank Challenge 1 Overview, 2025. [Online; accessed 11-February-2025].
  11. [Bromley et al., 1993] Jane Bromley, Isabelle Guyon, Yann LeCun, Eduard Säckinger, and Roopak Shah. Signature verification using a "Siamese" time delay neural network. Advances in neural information processing systems, 6, 1993.
  12. [Chen et al., 2020] Ting Chen, Simon Kornblith, Moham- mad Norouzi, and Geoffrey E. Hinton. A simple frame- work for contrastive learning of visual representations. ArXiv, abs/2002.05709, 2020.
  13. [Clark, 2020] K Clark. Electra: Pre-training text encoders as discriminators rather than generators. arXiv preprint arXiv:2003.10555, 2020.
  14. [Devlin et al., 2019] Jacob Devlin, Ming-Wei Chang, Ken- ton Lee, and Kristina Toutanova. BERT: Pre-training of deep bidirectional transformers for language understand- ing. In Jill Burstein, Christy Doran, and Thamar Solorio, editors, Proceedings of the 2019 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies, Volume 1 (Long and Short Papers), pages 4171-4186, Minneapo- lis, Minnesota, June 2019. Association for Computational Linguistics.
  15. [Gorishniy et al., 2022] Yury Gorishniy, Ivan Rubachev, and Artem Babenko. On embeddings for numerical features in tabular deep learning. Advances in Neural Information Processing Systems, 35:24991-25004, 2022.
  16. [Hoffer and Ailon, 2015] Elad Hoffer and Nir Ailon. Deep metric learning using triplet network. In Similarity-based pattern recognition: third international workshop, SIM- BAD 2015, Copenhagen, Denmark, October 12-14, 2015. Proceedings 3, pages 84-92. Springer, 2015.
  17. [Karpukhin et al., 2024] Ivan Karpukhin, Foma Shipilov, and Andrey Savchenko. Hotpp benchmark: Are we good at the long horizon events forecasting?, 2024.
  18. [Liu et al., 2019] Yinhan Liu, Myle Ott, Naman Goyal, Jingfei Du, Mandar Joshi, Danqi Chen, Omer Levy, Mike Lewis, Luke Zettlemoyer, and Veselin Stoyanov. Roberta: A robustly optimized bert pretraining approach, 2019. [Mollaev et al., 2024] Dzhambulat Mollaev, Alexander Kostin, Maria Postnova, Ivan Karpukhin, Ivan A Kireev, Gleb Gusev, and Andrey Savchenko. Multimodal bank- ing dataset: Understanding client needs through event sequences, 2024.
  19. [Musgrave et al., 2020] Kevin Musgrave, Serge J. Belongie, and Ser-Nam Lim. Pytorch metric learning. ArXiv, abs/2008.09164, 2020.
  20. [ODS.AI, 2025] ODS.AI. Sberbank Sirius Lesson Competi- tion, 2025. [Online; accessed 11-February-2025].
  21. [Oord et al., 2018] Aaron van den Oord, Yazhe Li, and Oriol Vinyals. Representation learning with contrastive predic- tive coding. arXiv preprint arXiv:1807.03748, 2018.
  22. [Padhi et al., 2021] Inkit Padhi, Yair Schiff, Igor Melnyk, Mattia Rigotti, Youssef Mroueh, Pierre Dognin, Jerret Ross, Ravi Nair, and Erik Altman. Tabular transformers for modeling multivariate time series. In Proceedings of International Conference on Acoustics, Speech and Signal Processing (ICASSP), pages 3565-3569. IEEE, 2021.
  23. [Pedregosa et al., 2011] Fabian Pedregosa, Gaël Varoquaux, Alexandre Gramfort, Vincent Michel, Bertrand Thirion, Olivier Grisel, Mathieu Blondel, Peter Prettenhofer, Ron Weiss, Vincent Dubourg, et al. Scikit-learn: Machine learning in python. Journal of machine learning research, 12(Oct):2825-2830, 2011.
  24. [Radford et al., 2019] Alec Radford, Jeffrey Wu, Rewon Child, David Luan, Dario Amodei, Ilya Sutskever, et al. Language models are unsupervised multitask learners. OpenAI blog, 1(8):9, 2019.
  25. [Tavenard et al., 2020] Romain Tavenard, Johann Faouzi, Gilles Vandewiele, Felix Divo, Guillaume Androz, Chester Holtz, Marie Payne, Roman Yurchak, Marc Rußwurm, Kushal Kolar, and Eli Woods. Tslearn, a ma- chine learning toolkit for time series data. Journal of Ma- chine Learning Research, 21(118):1-6, 2020.
  26. [Vasilev et al., 2024] Alexey Vasilev, Anna Volodkevich, Denis Kulandin, Tatiana Bysheva, and Anton Klenitskiy. Replay: a recommendation framework for experimenta- tion and production use. In 18th ACM Conference on Rec- ommender Systems, RecSys '24, page 1191-1194. ACM, October 2024.
  27. [Wang, 2020] Phil Wang. x-transformers: A flexible trans- former implementation in pytorch, 2020. GitHub reposi- tory.
  28. [Wolf, 2019] T Wolf. Huggingface's transformers: State- of-the-art natural language processing. arXiv preprint arXiv:1910.03771, 2019.
  29. [Yugay and Zaytsev, 2024] Aleksandr Yugay and Alexey Za- ytsev. Uniting contrastive and generative learning for event sequences models. ArXiv, abs/2408.09995, 2024.
  30. [Zbontar et al., 2021] Jure Zbontar, Li Jing, Ishan Misra, Yann LeCun, and Stéphane Deny. Barlow twins: Self- supervised learning via redundancy reduction. In Inter- national conference on machine learning, pages 12310- 12320. PMLR, 2021.
  31. [Zhao et al., 2022] Wayne Xin Zhao, Yupeng Hou, Xingyu Pan, Chen Yang, Zeyu Zhang, Zihan Lin, Jingsen Zhang, Shuqing Bian, Jiakai Tang, Wenqi Sun, Yushuo Chen, Lanling Xu, Gaowei Zhang, Zhen Tian, Changxin Tian, Shanlei Mu, Xinyan Fan, Xu Chen, and Ji-Rong Wen. Recbole 2.0: Towards a more up-to-date recommendation library. In CIKM, pages 4722-4726. ACM, 2022.
  32. [Zong et al., 2024] Yongshuo Zong, Oisin Mac Aodha, and Timothy Hospedales. Self-supervised multimodal learn- ing: A survey. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2024.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards