Hidden unit specialization in layered neural networks: ReLU vs. sigmoidal activation

Elisa Oostwal, Michiel Straat, Michael Biehl*

*Bijbehorende auteur voor dit werk

OnderzoeksoutputAcademicpeer review

35 Citaten (Scopus)
123 Downloads (Pure)

Samenvatting

By applying concepts from the statistical physics of learning, we study layered neural networks of rectified linear units (ReLU). The comparison with conventional, sigmoidal activation functions is in the center of interest. We compute typical learning curves for large shallow networks with K hidden units in matching student teacher scenarios. The systems undergo phase transitions, i.e. sudden changes of the generalization performance via the process of hidden unit specialization at critical sizes of the training set. Surprisingly, our results show that the training behavior of ReLU networks is qualitatively different from that of networks with sigmoidal activations. In networks with K≥3 sigmoidal hidden units, the transition is discontinuous: Specialized network configurations co-exist and compete with states of poor performance even for very large training sets. On the contrary, the use of ReLU activations results in continuous transitions for all K. For large enough training sets, two competing, differently specialized states display similar generalization abilities, which coincide exactly for large hidden layers in the limit K→∞. Our findings are also confirmed in Monte Carlo simulations of the training processes.
Originele taal-2English
Artikelnummer125517
TijdschriftPhysica A: Statistical Mechanics and its Applications
Volume564
Vroegere onlinedatumnov.-2020
DOI's
StatusPublished - feb.-2021

Vingerafdruk

Duik in de onderzoeksthema's van 'Hidden unit specialization in layered neural networks: ReLU vs. sigmoidal activation'. Samen vormen ze een unieke vingerafdruk.

Citeer dit