29
May
[Submitted on 20 May 2024] View a PDF of the paper titled On margin-based generalization prediction in deep neural networks, by Coenraad Mouton View PDF Abstract:Understanding generalization in deep neural networks is an active area of research. A promising avenue of exploration has been that of margin measurements: the shortest distance to the decision boundary for a given sample or that sample's representation internal to the network. Margin-based complexity measures have been shown to be correlated with the generalization ability of deep neural networks in some circumstances but not others. The reasons behind the success or failure of these metrics…