TIC能有效衡量接近NTK的DNN泛化能力,但超出该范围则失效。
Takeuchi's Information Criteria as Generalization Measures for DNNs Close to NTK Regime
- 用TIC评估DNN泛化性能,仅在接近NTK regime时有效
- 5000+模型实验显示:近NTK时TIC与泛化差距强相关
- 适用于超参数优化中的模型筛选,优于现有方法
泛化性能度量在机器学习中备受关注,但针对深度神经网络(DNNs)这类统计奇异模型,建立可靠的泛化度量仍具挑战。本研究聚焦于田口信息准则(Takeuchi's Information Criterion, TIC),探究其在何种条件下可有效解释DNN的泛化差距。理论分析表明,TIC在接近神经正切核(NTK)区域具有适用性。我们对12种架构、包括VGG-16在内的大型模型,在4个数据集上训练超过5000个DNN,并采用多种计算成本可控的TIC近似方法估计其值,评估精度与开销权衡。实验结果表明,当模型接近NTK regime时,估计的TIC值与泛化差距高度相关;但理论与实证均证明,超出该区域后相关性消失。此外,我们验证了TIC在超参数优化中比现有方法更优的模型筛选能力。
原文摘要 · Abstract (English)
Generalization measures have been studied extensively in the machine learning community to better characterize generalization gaps. However, establishing a reliable generalization measure for statistically singular models such as deep neural networks (DNNs) is difficult due to their complex nature. This study focuses on Takeuchi's information criterion (TIC) to investigate the conditions under which this classical measure can effectively explain the generalization gaps of DNNs. Importantly, the developed theory indicates the applicability of TIC near the neural tangent kernel (NTK) regime. In a series of experiments, we trained more than 5,000 DNN models with 12 architectures, including large models (e.g., VGG-16), on four datasets, and estimated the corresponding TIC values to examine the relationship between the generalization gap and the TIC estimates. We applied several TIC approximation methods with feasible computational costs and assessed the accuracy trade-off. Our experimental results indicate that the estimated TIC values correlate well with the generalization gap under conditions close to the NTK regime. However, we show both theoretically and empirically that outside the NTK regime such correlation disappears. Finally, we demonstrate that TIC provides better trial pruning ability than existing methods for hyperparameter optimization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。