AI RESEARCH

Calibrating Scientific Foundation Models with Inference-Time Stochastic Attention

arXiv CS.LG

ArXi:2604.19530v1 Announce Type: new Transformer-based scientific foundation models are increasingly deployed in high-stakes settings, but current architectures give deterministic outputs and provide limited for calibrated predictive uncertainty. We propose Stochastic Attention, a lightweight inference-time modification that randomizes attention by replacing softmax weights with normalized multinomial samples controlled by a single concentration parameter, and produces predictive ensembles without re.