Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Conferences, Workshops, Symposiums, and Seminars
  4. Robustness in deep learning: The good (width), the bad (depth), and the ugly (initialization)
 
conference paper

Robustness in deep learning: The good (width), the bad (depth), and the ugly (initialization)

Zhu, Zhenyu  
•
Liu, Fanghui  
•
Chrysos, Grigorios  
Show more
2022
[Proceedings of NeurIPS 2022]
36th Conference on Neural Information Processing Systems - NeurIPS 2022

We study the average robustness notion in deep neural networks in (selected) wide and narrow, deep and shallow, as well as lazy and non-lazy training settings. We prove that in the under-parameterized setting, width has a negative effect while it improves robustness in the over-parameterized setting. The effect of depth closely depends on the initialization and the training mode. In particular, when initialized with LeCun initialization, depth helps robustness with the lazy training regime. In contrast, when initialized with Neural Tangent Kernel (NTK) and He-initialization, depth hurts the robustness. Moreover, under the non-lazy training regime, we demonstrate how the width of a two-layer ReLU network benefits robustness. Our theoretical developments improve the results by Huang et al. [2021], Wu et al. [2021] and are consistent with Bubeck and Sellke [2021], Bubeck et al. [2021].

  • Files
  • Details
  • Metrics
Type
conference paper
Author(s)
Zhu, Zhenyu  
•
Liu, Fanghui  
•
Chrysos, Grigorios  
•
Cevher, Volkan  orcid-logo
Date Issued

2022

Published in
[Proceedings of NeurIPS 2022]
Total of pages

32

Subjects

ml-ai

Editorial or Peer reviewed

REVIEWED

Written at

EPFL

EPFL units
LIONS  
Event nameEvent placeEvent date
36th Conference on Neural Information Processing Systems - NeurIPS 2022

New Orleans, USA

November 28 - December 3, 2022

Available on Infoscience
October 16, 2022
Use this identifier to reference this record
https://infoscience.epfl.ch/handle/20.500.14299/191446
Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés