Search papers, labs, and topics across Lattice.
This paper provides a theoretical analysis of Random Network Distillation (RND) within the neural tangent kernel (NTK) framework in the infinite width limit, demonstrating its equivalence to deep ensembles and Bayesian inference. The analysis shows that RND's squared self-predictive error is equivalent to the predictive variance of a deep ensemble. Furthermore, by constructing a specific RND target function, the RND error distribution can be made to mirror the centered posterior predictive distribution of Bayesian inference, leading to a novel Bayesian RND model for posterior sampling.
Random Network Distillation, a computationally cheap uncertainty method, is theoretically equivalent to both deep ensembles and Bayesian inference under certain conditions, finally giving it a solid theoretical footing.
Uncertainty quantification is central to safe and efficient deployments of deep learning models, yet many computationally practical methods lack lacking rigorous theoretical motivation. Random network distillation (RND) is a lightweight technique that measures novelty via prediction errors against a fixed random target. While empirically effective, it has remained unclear what uncertainties RND measures and how its estimates relate to other approaches, e.g. Bayesian inference or deep ensembles. This paper establishes these missing theoretical connections by analyzing RND within the neural tangent kernel framework in the limit of infinite network width. Our analysis reveals two central findings in this limit: (1) The uncertainty signal from RND -- its squared self-predictive error -- is equivalent to the predictive variance of a deep ensemble. (2) By constructing a specific RND target function, we show that the RND error distribution can be made to mirror the centered posterior predictive distribution of Bayesian inference with wide neural networks. Based on this equivalence, we moreover devise a posterior sampling algorithm that generates i.i.d. samples from an exact Bayesian posterior predictive distribution using this modified \textit{Bayesian RND} model. Collectively, our findings provide a unified theoretical perspective that places RND within the principled frameworks of deep ensembles and Bayesian inference, and offer new avenues for efficient yet theoretically grounded uncertainty quantification methods.
