Singularity-Induced Local Complexity for High-Dimensional Learning Models
Authors: Atsuyoshi Muta, Masaaki Imaizumi
Organizations: The University of Tokyo, Tokyo, Japan. · RIKEN Center for Advanced Intelligence Project, Tokyo, Japan. · Kyoto University, Kyoto, Japan.
Many overparameterized models that have achieved remarkable success are singular models. That is, the set of optimal parameters consists of multiple points, or the Hessian of the generalization error at such parameters is degenerate, reflecting redundancy in the model parametrization. Therefore, understanding the effective dimension and intrinsic model complexity, beyond the apparent dimension of the parameter space, is essential for studying the generalization ability of such overparameterized models. In this work, we focus on the real log canonical threshold (RLCT) and its localized counterpart, the local RLCT, which are fundamental quantities for characterizing the generalization of singular models. We clarify the theoretical relationship between the local RLCT and existing two key metrics related to generalization under some localization assumption: (I) the rank of the Hessian matrix of the generalization error and (II) the covering number of a family of loss functions. This result provides novel insights into effective dimension and intrinsic complexity of overparameterized models.
Figures & tables
Figure 1: (a) The definition of RLCT. The RLCT is defined as the absolute value of the largest pole of the ζ function. The ζ function is initially defined for s∈C with Re(s)>0 . and it can then be extended to the entire complex plane by analytic continuation Atiyah (1970) . Then their poles (Remark 2 ) arise on the negative real axis Kashiwara (1976) ; Watanabe (2009) . It was found that this mathematical quantity is deeply related to statistical models ( Watanabe, 2009 ) . (b) The image of ζ(s) . At each pole, ∣ζ(s)∣ diverges to infinity.
Figure 2: Hessian Eigenvalues vs. Local RLCT : rank∇2R(wi)≤2λ(wi)(i=1,2,3) holds at all displayed parameter values. The red dots indicate the values of the eigenvalues, and the orange vertical line marks 2λ(wi) . The rank of the Hessian equals the number of its positive eigenvalues at global minimizers. The parameters and their corresponding local RLCTs are given in ( 8 ) for reduced-rank regression, and in ( 9 ) for the linear neural network.
Figure 3: Hessian Eigenvalues vs. Local RLCT (GELU Network) : This plot shows the eigenvalues of the Hessian and the estimated local RLCTs λ^SGLD(wi) (orange lines) at five parameters wi(i=1,…,5) , which are chosen from a GELU network so that the output function is 0 . The number of dominant eigenvalues is roughly bounded above by 2λ(wi)(i=1,…,5) .
Figure 4: Hessian Eigenvalues vs. Local RLCT (ReLU Network) : This plot shows the eigenvalues of the Hessian and the estimated local RLCTs λ^SGLD(wi) (orange lines) at five parameters wi(i=1,…,5) , which are chosen from a ReLU network so that the output function is zero. Since ReLU is not analytic, the experimental results do not clearly support the claim of the Main Theorem.