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
-
Can LLMs Access their World Knowledge for Event Prediction?
Can LLMs Access their World Knowledge for Event Prediction?
Z. Wu, Z. Zhang, H. Hajimirsadeghi, N. Dvornik, and A. Pashevich. EMNLP
Publications
-
Jump Start or False Start? A Theoretical and Empirical Evaluation of LLM-initialized Bandits
Jump Start or False Start? A Theoretical and Empirical Evaluation of LLM-initialized Bandits
A. Bailey, X. Zhu, R. Aoki, Y. Cao, and K. Wilson. TMLR
Publications
-
Exploring Model Invariance with Discrete Search for Ultra-Low-Bit Quantization
Exploring Model Invariance with Discrete Search for Ultra-Low-Bit Quantization
Y. Wen, Y. Cao, and L. Mou. CIKM
Publications