We present a diffusion based model for asynchronous time series prediction, where the goal is to predict the next inter event time and event type. To address the inherent uncertainty of future events, we introduce ReDiTT, a retrieval augmented conditional diffusion transformer that operates in latent space. ReDiTT retrieves structurally similar latent sequences from a memory bank during both training and inference and incorporates them as reference conditions through cross attention. This retrieval based conditioning allows the model to attend to relevant temporal dynamics and provides global structural guidance for generation. As a result, ReDiTT stabilizes long horizon forecasting and improves sample diversity. Experiments on seven real world datasets demonstrate state of the art performance on next event prediction and long horizon forecasting.
Bibtex
@misc{lyu2026redittretrievalaugmentedconditional,
title={ReDiTT: Retrieval Augmented Conditional Diffusion Transformers for Asynchronous Time Series},
author={Saiyue Lyu and Zhitian Zhang and Ruizhi Deng and Thibaut Durand},
year={2026},
eprint={2607.12391},
archivePrefix={arXiv},
primaryClass={cs.LG},
url={https://arxiv.org/abs/2607.12391},
}
Related Research
-
Score-based Outlier Generation via Controlling the Radon-Nikodym Derivative
Score-based Outlier Generation via Controlling the Radon-Nikodym Derivative
A. Mukherjee, T. Milne, K. Y. C. Lui, S. Hazlewood, and J. Liu. IEEE Conference on Decision and Control
Publications
-
Investigating Action Embeddings for More Efficient Off-Policy Evaluation
Investigating Action Embeddings for More Efficient Off-Policy Evaluation
M. Wabartha, K. Wilson, R. David Evans, H. Sharifi, and T. Sylvain. RecSys CONSEQUENCES Workshop
Publications
-
FACTS: Fast, Accurate, and Privacy-Compliant TableSummarization via Offline Template Generation
FACTS: Fast, Accurate, and Privacy-Compliant TableSummarization via Offline Template Generation
Y. Yuan, M. Amin Shabani, and S. Liu. Workshop on Generative AI in Finance at NeurIPS
Publications