Dropout Robustness and Cognitive Profiling of Transformer Models via Stochastic Inference
On the map: LLM reasoning → Test-time compute and efficient thinking
Abstract (arXiv)
Transformer-based language models are widely deployed for reasoning, yet their behavior under inference-time stochasticity remains underexplored. While dropout is common during training, its inference-time effects via Monte Carlo sampling lack systematic evaluation across architectures, limiting understanding of model reliability in uncertainty-aware applications. This work analyzes dropout-induced variability across 19 transformer models using MC Dropout with 100 stochastic forward passes per sample. Dropout robustness is defined as maintaining high accuracy and stable predictions under stochastic inference, measured by standard deviation of per-run accuracies. A cognitive decomposition framework disentangles performance into memory and reasoning components. Experiments span five dropout configurations yielding 95 unique evaluations on 1,000 samples. Results reveal substantial architectural variation. Smaller models demonstrate perfect prediction stability while medium-sized models exhibit notable volatility. Mid-sized models achieve the best overall performance; larger models excel at memory tasks. Critically, 53% of models suffer severe accuracy degradation under baseline MC Dropout, with task-specialized models losing up to 24 percentage points, indicating unsuitability for uncertainty quantification in these architectures. Asymmetric effects emerge: high dropout reduces memory accuracy by 27 percentage points while reasoning degrades only 1 point, suggesting memory tasks rely on stable representations that dropout disrupts. 84% of models demonstrate memory-biased performance. This provides the first comprehensive MC Dropout benchmark for transformers, revealing dropout robustness is architecture-dependent and uncorrelated with scale. The cognitive profiling framework offers actionable guidance for model selection in uncertainty-aware applications.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 24 of its 32 references that are in the map. First posted on arXiv 18 Mar 2026.
Builds on
24 of 32 references.
References · 24
- 2017Attention Is All You Needcited by 60,361
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2017On Calibration of Modern Neural Networkscited by 4,960
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
- 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,148
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,926
- 2022Training Compute-Optimal Large Language Modelscited by 3,019
- 2017What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?cited by 2,352
12 more
- 2020DeBERTa: Decoding-enhanced BERT with Disentangled Attentioncited by 2,181
- 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,822
- 2019SuperGLUE: A Stickier Benchmark for General-Purpose Language Understanding Systemscited by 1,729
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
- 2019BERT Rediscovers the Classical NLP Pipelinecited by 1,059
- 2020A Primer in BERTology: What we know about how BERT workscited by 815
- 2019Is BERT Really Robust? A Strong Baseline for Natural Language Attack on Text Classification and Entailmentcited by 671
- 2021Datasets: A Community Library for Natural Language Processingcited by 530
- 2020Calibration of Pre-trained Transformerscited by 315
- 2020Pretrained Transformers Improve Out-of-Distribution Robustnesscited by 294
- 2018DropBlock: A regularization method for convolutional networkscited by 262
- 2016Zoneout: Regularizing RNNs by Randomly Preserving Hidden Activationscited by 54
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026At the Edge of Understanding: Sparse Autoencoders Trace The Limits of Transformer Generalization
- 2026Geometric and Behavioral Stratification in Transformer Residual Streams
- 2026LEAP: Layer-wise Exit-Aware Pretraining for Efficient Transformer Inference
- 2025From Condensation to Rank Collapse: A Two-Stage Analysis of Transformer Training Dynamics
- 2025Measuring Uncertainty in Transformer Circuits with Effective Information Consistency
- 2026How Uncertainty Estimation Scales with Sampling in Reasoning Models
- 2026What Does Layer-Importance Reveal About Transformers and State-Space Models?
- 2026Align, Unify, Suppress, Route: A Coherentist View of Transformer Computation