DES-JEPA: Learning a Shared Vocabulary for High-Dimensional Discrete Event Sequences
Organizations: BMW Group · University of Glasgow
Abstract
Temporal foundation models such as Chronos and TimesFM generalize across domains through a shared numerical observation space. In contrast, discrete event sequences (DESs), including operational event logs or patient clinical-event trajectories, have domain-specific event vocabularies whose labels lack a consistent correspondence across domains. Beyond this lack of consistency, the number of distinct events in real-world settings is high enough to challenge both classical temporal point process modeling and current in-context foundation inference models such as FIM-PP. Critically, current foundation models for DESs are limited to a small, fixed number of event types, restricting their applicability to realistic, high-dimensional event streams. To address this, we propose two complementary modules: (i) DES-JEPA, a joint-embedding predictive architecture that learns a shared discrete vocabulary from synthetic stochastic processes with unaligned event vocabularies; and (ii) DES-PFN, a discrete event sequence prior-fitted network that reuses the learned vocabulary for in-context next-event prediction. Without task-specific fine-tuning, this framework achieves competitive zero-shot performance on eight unseen real-world datasets and scales up to 700 distinct event types.
Figures & tables
| RMTPP [ 26 ] | NHP [ 27 ] | SAHP [ 28 ] | |||||||
| Dataset | few-shot | full | few-shot | full | few-shot | full | FIM-PP [ 12 ] | DES-PFN (ours) | |
| retweet [ 29 ] | 3 | 0.464 0.021 | 0.533 | 0.235 0.011 | 0.591 | 0.531 0.020 | 0.471 | 0.462 0.022 | 0.520 0.023 |
| stackoverflow [ 29 ] | 22 | 0.009 0.003 | 0.411 | 0.007 0.002 | 0.420 | 0.413 0.035 | 0.420 | 0.441 0.032 | 0.356 0.030 |
| taobao [ 29 ] | 17 | 0.227 0.049 | 0.559 | 0.490 0.049 | 0.522 | 0.433 0.041 | 0.465 | 0.516 0.040 | 0.420 0.057 |
| amazon [ 29 ] | 16 | 0.151 0.028 | 0.299 | 0.089 0.006 | 0.319 | 0.193 0.031 | 0.254 | 0.207 0.041 | 0.241 0.035 |
| HDFS [ 30 , 31 ] | 14 | – | 0.074 § | – | 0.063 § | – | 0.103 § | 0.266 0.018 | 0.200 0.028 |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
Sequence Modeling
Sequence modeling studies how to represent, predict, classify, and generate ordered data such as text, time series, events, and biological sequences.
See all 253 papers