Current browse context:
stat.ML
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: Label-Aware Neural Tangent Kernel: Toward Better Generalization and Local Elasticity
(Submitted on 22 Oct 2020 (v1), last revised 29 Oct 2020 (this version, v2))
Abstract: As a popular approach to modeling the dynamics of training overparametrized neural networks (NNs), the neural tangent kernels (NTK) are known to fall behind real-world NNs in generalization ability. This performance gap is in part due to the \textit{label agnostic} nature of the NTK, which renders the resulting kernel not as \textit{locally elastic} as NNs~\citep{he2019local}. In this paper, we introduce a novel approach from the perspective of \emph{label-awareness} to reduce this gap for the NTK. Specifically, we propose two label-aware kernels that are each a superimposition of a label-agnostic part and a hierarchy of label-aware parts with increasing complexity of label dependence, using the Hoeffding decomposition. Through both theoretical and empirical evidence, we show that the models trained with the proposed kernels better simulate NNs in terms of generalization ability and local elasticity.
Submission history
From: Shuxiao Chen [view email][v1] Thu, 22 Oct 2020 14:54:32 GMT (601kb,D)
[v2] Thu, 29 Oct 2020 17:23:44 GMT (604kb,D)
Link back to: arXiv, form interface, contact.