LLM Reasoning 相关度: 5/10

Monte Carlo Stochastic Depth for Uncertainty Estimation in Deep Learning

Adam T. Müller, Tobias Rögelein, Nicolaj C. Stache
arXiv: 2604.12719v1 发布: 2026-04-14 更新: 2026-04-14

AI 摘要

论文研究了Monte Carlo Stochastic Depth在深度学习中不确定性估计的有效性,并进行了理论分析和实验验证。

主要贡献

  • 建立了MCSD与贝叶斯变分推断的理论联系
  • 首次在目标检测任务上全面评估了MCSD的性能
  • 验证了MCSD作为一种高效贝叶斯近似工具的有效性

方法论

通过理论推导将MCSD与变分推断联系,并在COCO数据集上对比MCSD与MCD、MCDB在YOLO和RT-DETR上的表现。

原文摘要

The deployment of deep neural networks in safety-critical systems necessitates reliable and efficient uncertainty quantification (UQ). A practical and widespread strategy for UQ is repurposing stochastic regularizers as scalable approximate Bayesian inference methods, such as Monte Carlo Dropout (MCD) and MC-DropBlock (MCDB). However, this paradigm remains under-explored for Stochastic Depth (SD), a regularizer integral to the residual-based backbones of most modern architectures. While prior work demonstrated its empirical promise for segmentation, a formal theoretical connection to Bayesian variational inference and a benchmark on complex, multi-task problems like object detection are missing. In this paper, we first provide theoretical insights connecting Monte Carlo Stochastic Depth (MCSD) to principled approximate variational inference. We then present the first comprehensive empirical benchmark of MCSD against MCD and MCDB on state-of-the-art detectors (YOLO, RT-DETR) using the COCO and COCO-O datasets. Our results position MCSD as a robust and computationally efficient method that achieves highly competitive predictive accuracy (mAP), notably yielding slight improvements in calibration (ECE) and uncertainty ranking (AUARC) compared to MCD. We thus establish MCSD as a theoretically-grounded and empirically-validated tool for efficient Bayesian approximation in modern deep learning.

标签

不确定性估计 随机深度 贝叶斯推断 目标检测

arXiv 分类

cs.LG stat.ML