Event prediction is central to applications from e-commerce to finance, yet real-world event streams are highly non-stationary, making deployment of frozen pre-trained models brittle. We propose Event prediction Test-Time Adaptation (ETTA), a model-agnostic lightweight framework that enables event-prediction models to adapt on-the-fly during inference, without retraining the model itself. Unlike model fine-tuning at test-time, ETTA introduces compact temporal and logit calibration modules that calibrate the inter-event timestamps and event type logits respectively. These adapters can be seamlessly applied to both neural temporal point process (TPP) models and large language models (LLMs). On five real-world benchmarks, ETTA consistently improves TPP models, achieving up to 20.3% reduction in RMSE and 4.9% higher accuracy, while for LLM-based event prediction it yields up to 23.1% RMSE reduction and 9.0% accuracy improvement. Together, these results establish test-time adaptation, and ETTA in particular, as a powerful, general framework for robust event prediction under distribution shift.
Bibtex
@inproceedings{
grover2026testtime,
title={Test-Time Adaptation for Event Prediction via Lightweight Adapters},
author={Shivam Grover and Hossein Hajimirsadeghi and Zhitian Zhang and Edward J. Smith and Alexander Pashevich},
booktitle={Catch, Adapt, and Operate: Monitoring ML Models Under Drift Workshop},
year={2026},
url={https://openreview.net/forum?id=VHtCpZc9L7}
}
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