Bert B.A. de Vries

Improved Depth Estimation of Bayesian Neural Networks

2024 · arXiv (Cornell University)

This paper proposes improvements over earlier work by Nazareth and Blei (2022) for estimating the depth of Bayesian neural networks. Here, we propose a discrete truncated normal distribution over the network depth to independently learn …