7167@AAAI

Total: 1

#1 VECA: A Method for Detecting Overfitting in Neural Networks (Student Abstract) [PDF] [Copy] [Kimi]

Authors: Liangzhu Ge ; Yuexian Hou ; Yaju Jiang ; Shuai Yao ; Chao Yang

Despite their widespread applications, deep neural networks often tend to overfit the training data. Here, we propose a measure called VECA (Variance of Eigenvalues of Covariance matrix of Activation matrix) and demonstrate that VECA is a good predictor of networks' generalization performance during the training process. Experiments performed on fully-connected networks and convolutional neural networks trained on benchmark image datasets show a strong correlation between test loss and VECA, which suggest that we can calculate the VECA to estimate generalization performance without sacrificing training data to be used as a validation set.