Estimation of the Learning Coefficient Using Empirical Loss
Tatsuyoshi Takio, Joe Suzuki
TL;DR
The paper addresses estimating the learning coefficient $\lambda$, a key parameter governing the asymptotic behavior of WAIC/WBIC in both regular and singular models. It introduces a novel estimator based on the Empirical Loss $T_n$ that combines WBIC with $T_n$ as $\hat{\lambda}_T=(WBIC_n-nT_n)/\log n$, eliminating the need for multiple inverse-temperature settings. The authors prove consistency and demonstrate through simulations that the estimator achieves lower bias and notably lower variance, especially in singular-model scenarios and when MCMC outliers occur. These results enhance the reliability and practicality of WBIC-based model-selection criteria in complex Bayesian settings and deepen understanding of singular model behavior.
Abstract
The learning coefficient plays a crucial role in analyzing the performance of information criteria, such as the Widely Applicable Information Criterion (WAIC) and the Widely Applicable Bayesian Information Criterion (WBIC), which Sumio Watanabe developed to assess model generalization ability. In regular statistical models, the learning coefficient is given by d/2, where d is the dimension of the parameter space. More generally, it is defined as the absolute value of the pole order of a zeta function derived from the Kullback-Leibler divergence and the prior distribution. However, except for specific cases such as reduced-rank regression, the learning coefficient cannot be derived in a closed form. Watanabe proposed a numerical method to estimate the learning coefficient, which Imai further refined to enhance its convergence properties. These methods utilize the asymptotic behavior of WBIC and have been shown to be statistically consistent as the sample size grows. In this paper, we propose a novel numerical estimation method that fundamentally differs from previous approaches and leverages a new quantity, "Empirical Loss," which was introduced by Watanabe. Through numerical experiments, we demonstrate that our proposed method exhibits both lower bias and lower variance compared to those of Watanabe and Imai. Additionally, we provide a theoretical analysis that elucidates why our method outperforms existing techniques and present empirical evidence that supports our findings.
