Improved weight initialization for deep and narrow feedforward neural network

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Lee, Hyunwoo, Kim, Yunho, Yang, Seung Yeop, Choi, Hayoung
Formato: Preprint
Publicado: 2023
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913293336051712
author Lee, Hyunwoo
Kim, Yunho
Yang, Seung Yeop
Choi, Hayoung
author_facet Lee, Hyunwoo
Kim, Yunho
Yang, Seung Yeop
Choi, Hayoung
contents Appropriate weight initialization settings, along with the ReLU activation function, have become cornerstones of modern deep learning, enabling the training and deployment of highly effective and efficient neural network models across diverse areas of artificial intelligence. The problem of \textquotedblleft dying ReLU," where ReLU neurons become inactive and yield zero output, presents a significant challenge in the training of deep neural networks with ReLU activation function. Theoretical research and various methods have been introduced to address the problem. However, even with these methods and research, training remains challenging for extremely deep and narrow feedforward networks with ReLU activation function. In this paper, we propose a novel weight initialization method to address this issue. We establish several properties of our initial weight matrix and demonstrate how these properties enable the effective propagation of signal vectors. Through a series of experiments and comparisons with existing methods, we demonstrate the effectiveness of the novel initialization method.
format Preprint
id arxiv_https___arxiv_org_abs_2311_03733
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Improved weight initialization for deep and narrow feedforward neural network
Lee, Hyunwoo
Kim, Yunho
Yang, Seung Yeop
Choi, Hayoung
Machine Learning
Neural and Evolutionary Computing
Appropriate weight initialization settings, along with the ReLU activation function, have become cornerstones of modern deep learning, enabling the training and deployment of highly effective and efficient neural network models across diverse areas of artificial intelligence. The problem of \textquotedblleft dying ReLU," where ReLU neurons become inactive and yield zero output, presents a significant challenge in the training of deep neural networks with ReLU activation function. Theoretical research and various methods have been introduced to address the problem. However, even with these methods and research, training remains challenging for extremely deep and narrow feedforward networks with ReLU activation function. In this paper, we propose a novel weight initialization method to address this issue. We establish several properties of our initial weight matrix and demonstrate how these properties enable the effective propagation of signal vectors. Through a series of experiments and comparisons with existing methods, we demonstrate the effectiveness of the novel initialization method.
title Improved weight initialization for deep and narrow feedforward neural network
topic Machine Learning
Neural and Evolutionary Computing
url https://arxiv.org/abs/2311.03733