mlsanityData to 14 Oct 2026

Dropout Robustness and Cognitive Profiling of Transformer Models via Stochastic Inference

arXiv 2603.17811 ↗First posted 18 Mar 2026cs.LG cs.AIcited by 0

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

  1. 2017Attention Is All You Needcited by 60,361
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  3. 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
  4. 2020Scaling Laws for Neural Language Modelscited by 6,586
  5. 2017On Calibration of Modern Neural Networkscited by 4,960
  6. 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
  7. 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
  8. 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,148
  9. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
  10. 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,926
  11. 2022Training Compute-Optimal Large Language Modelscited by 3,019
  12. 2017What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?cited by 2,352
12 more
  1. 2020DeBERTa: Decoding-enhanced BERT with Disentangled Attentioncited by 2,181
  2. 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,822
  3. 2019SuperGLUE: A Stickier Benchmark for General-Purpose Language Understanding Systemscited by 1,729
  4. 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
  5. 2019BERT Rediscovers the Classical NLP Pipelinecited by 1,059
  6. 2020A Primer in BERTology: What we know about how BERT workscited by 815
  7. 2019Is BERT Really Robust? A Strong Baseline for Natural Language Attack on Text Classification and Entailmentcited by 671
  8. 2021Datasets: A Community Library for Natural Language Processingcited by 530
  9. 2020Calibration of Pre-trained Transformerscited by 315
  10. 2020Pretrained Transformers Improve Out-of-Distribution Robustnesscited by 294
  11. 2018DropBlock: A regularization method for convolutional networkscited by 262
  12. 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

  1. 2026At the Edge of Understanding: Sparse Autoencoders Trace The Limits of Transformer Generalization
  2. 2026Geometric and Behavioral Stratification in Transformer Residual Streams
  3. 2026LEAP: Layer-wise Exit-Aware Pretraining for Efficient Transformer Inference
  4. 2025From Condensation to Rank Collapse: A Two-Stage Analysis of Transformer Training Dynamics
  5. 2025Measuring Uncertainty in Transformer Circuits with Effective Information Consistency
  6. 2026How Uncertainty Estimation Scales with Sampling in Reasoning Models
  7. 2026What Does Layer-Importance Reveal About Transformers and State-Space Models?
  8. 2026Align, Unify, Suppress, Route: A Coherentist View of Transformer Computation