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

Download:

Current browse context:

stat.ML

Change to browse by:

References & Citations

Bookmark

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

Statistics > Machine Learning

Title: Dying ReLU and Initialization: Theory and Numerical Examples

Abstract: The dying ReLU refers to the problem when ReLU neurons become inactive and only output 0 for any input. There are many empirical and heuristic explanations of why ReLU neurons die. However, little is known about its theoretical analysis. In this paper, we rigorously prove that a deep ReLU network will eventually die in probability as the depth goes to infinite. Several methods have been proposed to alleviate the dying ReLU. Perhaps, one of the simplest treatments is to modify the initialization procedure. One common way of initializing weights and biases uses symmetric probability distributions, which suffers from the dying ReLU. We thus propose a new initialization procedure, namely, a randomized asymmetric initialization. We prove that the new initialization can effectively prevent the dying ReLU. All parameters required for the new initialization are theoretically designed. Numerical examples are provided to demonstrate the effectiveness of the new initialization procedure.
Subjects: Machine Learning (stat.ML); Machine Learning (cs.LG); Probability (math.PR)
DOI: 10.4208/cicp.OA-2020-0165
Cite as: arXiv:1903.06733 [stat.ML]
  (or arXiv:1903.06733v3 [stat.ML] for this version)

Submission history

From: Yeonjong Shin [view email]
[v1] Fri, 15 Mar 2019 18:23:55 GMT (373kb,D)
[v2] Tue, 12 Nov 2019 23:15:23 GMT (278kb,D)
[v3] Wed, 21 Oct 2020 19:19:02 GMT (279kb,D)

Link back to: arXiv, form interface, contact.