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


Current browse context:


Change to browse by:

References & Citations


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

Statistics > Machine Learning

Title: Improved Generalization Bounds of Group Invariant / Equivariant Deep Networks via Quotient Feature Spaces

Abstract: Numerous invariant (or equivariant) neural networks have succeeded in handling invariant data such as point clouds and graphs. However, a generalization theory for the neural networks has not been well developed, because several essential factors for the theory, such as network size and margin distribution, are not deeply connected to the invariance and equivariance. In this study, we develop a novel generalization error bound for invariant and equivariant deep neural networks. To describe the effect of invariance and equivariance on generalization, we develop a notion of a \textit{quotient feature space}, which measures the effect of group actions for the properties. Our main result proves that the volume of quotient feature spaces can describe the generalization error. Furthermore, the bound shows that the invariance and equivariance significantly improve the leading term of the bound. We apply our result to specific invariant and equivariant networks, such as DeepSets (Zaheer et al. (2017)), and show that their generalization bound is considerably improved by $\sqrt{n!}$, where $n!$ is the number of permutations. We also discuss the expressive power of invariant DNNs and show that they can achieve an optimal approximation rate. Our experimental result supports our theoretical claims.
Comments: Old title: "Improved Generalization Bound of Permutation Invariant Deep Neural Networks"
Subjects: Machine Learning (stat.ML); Machine Learning (cs.LG)
Cite as: arXiv:1910.06552 [stat.ML]
  (or arXiv:1910.06552v3 [stat.ML] for this version)

Submission history

From: Masaaki Imaizumi [view email]
[v1] Tue, 15 Oct 2019 06:24:53 GMT (719kb,D)
[v2] Fri, 20 Mar 2020 12:56:51 GMT (3122kb,D)
[v3] Sat, 19 Jun 2021 15:53:54 GMT (8068kb,D)

Link back to: arXiv, form interface, contact.