arXiv · 2306.16090
Empirical Loss Landscape Analysis of Neural Network Activation Functions
Abstract
Activation functions play a significant role in neural network design by enabling non-linearity. The choice of activation function was previously shown to influence the properties of the resulting loss landscape. Understanding the relationship between activation functions and loss landscape properties is important for neural architecture and training algorithm design. This study empirically investigates neural network loss landscapes associated with hyperbolic tangent, rectified linear unit, and exponential linear unit activation functions. Rectified linear unit is shown to yield the most convex loss landscape, and exponential linear unit is shown to yield the least flat loss landscape, and to exhibit superior generalisation performance. The presence of wide and narrow valleys in the loss landscape is established for all activation functions, and the narrow valleys are shown to correlate with saturated neurons and implicitly regularised network configurations.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Anna Sergeevna Bosman, Andries Engelbrecht, Marde Helbig. 2023-06-28. Empirical Loss Landscape Analysis of Neural Network Activation Functions. https://doi.org/10.1145/3583133.3596321
Cite the original work for its findings. Save a collection to share your selection of sources.