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
-
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