English

Deep Layer-wise Networks Have Closed-Form Weights

Machine Learning 2022-02-10 v6 Machine Learning

Abstract

There is currently a debate within the neuroscience community over the likelihood of the brain performing backpropagation (BP). To better mimic the brain, training a network one layer at a time\textit{one layer at a time} with only a "single forward pass" has been proposed as an alternative to bypass BP; we refer to these networks as "layer-wise" networks. We continue the work on layer-wise networks by answering two outstanding questions. First, do they have a closed-form solution?\textit{do they have a closed-form solution?} Second, how do we know when to stop adding more layers?\textit{how do we know when to stop adding more layers?} This work proves that the kernel Mean Embedding is the closed-form weight that achieves the network global optimum while driving these networks to converge towards a highly desirable kernel for classification; we call it the Neural Indicator Kernel\textit{Neural Indicator Kernel}.

Keywords

Cite

@article{arxiv.2006.08539,
  title  = {Deep Layer-wise Networks Have Closed-Form Weights},
  author = {Chieh Wu and Aria Masoomi and Arthur Gretton and Jennifer Dy},
  journal= {arXiv preprint arXiv:2006.08539},
  year   = {2022}
}

Comments

This version will be published in AIStats 2022

R2 v1 2026-06-23T16:20:34.222Z