Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces ReDiTT, a retrieval augmented conditional diffusion transformer designed for predicting asynchronous time series by forecasting both the next inter-event time and event type. By leveraging a memory bank of structurally similar latent sequences, ReDiTT enhances its predictive capabilities through cross attention, allowing it to effectively capture relevant temporal dynamics. Experimental results across seven real-world datasets show that ReDiTT achieves state-of-the-art performance in long horizon forecasting and improves sample diversity compared to existing methods.
Retrieval-augmented conditioning in ReDiTT stabilizes long horizon forecasting and enhances sample diversity in asynchronous time series prediction.
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. Our code is available at https://github.com/BorealisAI/ReDiTT.