We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:


Current browse context:


Change to browse by:

References & Citations

DBLP - CS Bibliography


(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Computer Science > Machine Learning

Title: Neural networks learn to magnify areas near decision boundaries

Abstract: We study how training molds the Riemannian geometry induced by neural network feature maps. At infinite width, neural networks with random parameters induce highly symmetric metrics on input space. Feature learning in networks trained to perform classification tasks magnifies local areas along decision boundaries. These changes are consistent with previously proposed geometric approaches for hand-tuning of kernel methods to improve generalization.
Comments: 53 pages, many figures
Subjects: Machine Learning (cs.LG); Disordered Systems and Neural Networks (cond-mat.dis-nn); Machine Learning (stat.ML)
Cite as: arXiv:2301.11375 [cs.LG]
  (or arXiv:2301.11375v1 [cs.LG] for this version)

Submission history

From: Jacob Zavatone-Veth [view email]
[v1] Thu, 26 Jan 2023 19:43:16 GMT (20018kb,D)

Link back to: arXiv, form interface, contact.