arXiv · 1704.05646
Effects of the optimisation of the margin distribution on generalisation in deep architectures
Abstract
Despite being so vital to success of Support Vector Machines, the principle of separating margin maximisation is not used in deep learning. We show that minimisation of margin variance and not maximisation of the margin is more suitable for improving generalisation in deep architectures. We propose the Halfway loss function that minimises the Normalised Margin Variance (NMV) at the output of a deep learning models and evaluate its performance against the Softmax Cross-Entropy loss on the MNIST, smallNORB and CIFAR-10 datasets.
Explore related subjects
Keep this discovery
Lech Szymanski, Brendan McCane, Wei Gao, Zhi-Hua Zhou. 2017-04-19. Effects of the optimisation of the margin distribution on generalisation in deep architectures. https://arxiv.org/abs/1704.05646
Cite the original work for its findings. Save a collection to share your selection of sources.