Robust Large Margin Deep Neural Networks

By Jure Sokolić, Raja Giryes, Guillermo Sapiro, Miguel Raul Dias Rodrigues. 2016

349 citations recorded by the source corpus.

The generalization error of deep neural networks via their classification margin is studied in this paper. Our approach is based on the Jacobian matrix of a deep neural network and can be applied to networks with arbitrary nonlinearities and pooling layers, and to networks with different architectures such as feed forward networks and residual networks. Our analysis leads to the conclusion that a bounded spectral norm of the network's Jacobian matrix in the neighbourhood of the training samples is crucial for a deep neural network of arbitrary depth and width to generalize well. This is a significant improvement over the current bounds in the literature, which imply that the generalization error grows with either the width or the depth of the network. Moreover, it shows that the recently proposed batch normalization and weight normalization reparametrizations enjoy good generalization properties, and leads to a novel network regularizer based on the network's Jacobian matrix. The analysis is supported with experimental results on the MNIST, CIFAR-10, LaRED, and ImageNet datasets.

Subfields: Artificial Intelligence, Computational Mechanics, Computer Vision and Pattern Recognition, Safety, Robustness & Interpretability

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Yoshua Bengio
  2. Percy Liang
  3. Jianfeng Gao
  4. Ruslan Salakhutdinov
  5. Quoc V. Le
  6. Mohit Bansal
  7. Dacheng Tao
  8. Sergey Levine

In Computational Mechanics

  1. Tengyu Ma
  2. Dacheng Tao
  3. Joan Bruna
  4. Wotao Yin
  5. K. Jia
  6. Stanley J. Osher
  7. Stefano Soatto
  8. Colin Wei

In Computer Vision and Pattern Recognition

  1. Luc Van Gool
  2. Dacheng Tao
  3. Trevor Darrell
  4. Ming-Hsuan Yang
  5. Jitendra Malik
  6. Xiaogang Wang
  7. Alan L. Yuille
  8. Bernt Schiele

In Safety, Robustness & Interpretability

  1. Matthias Hein
  2. Nicholas Carlini
  3. Francesco Croce
  4. Maksym Andriushchenko
  5. Aleksander Madry
  6. Nicolas Papernot
  7. Dimitris Tsipras
  8. Battista Biggio

Citations in the corpus