Mathematics (Feb 2022)

Local Linear Approximation Algorithm for Neural Network

  • Mudong Zeng,
  • Yujie Liao,
  • Runze Li,
  • Agus Sudjianto

DOI
https://doi.org/10.3390/math10030494
Journal volume & issue
Vol. 10, no. 3
p. 494

Abstract

Read online

This paper aims to develop a new training strategy to improve efficiency in estimation of weights and biases in a feedforward neural network (FNN). We propose a local linear approximation (LLA) algorithm, which approximates ReLU with a linear function at the neuron level and estimate the weights and biases of one-hidden-layer neural network iteratively. We further propose the layer-wise optimized adaptive neural network (LOAN), in which we use the LLA to estimate the weights and biases in the LOAN layer by layer adaptively. We compare the performance of the LLA with the commonly-used procedures in machine learning based on seven benchmark data sets. The numerical comparison implies that the proposed algorithm may outperform the existing procedures in terms of both training time and prediction accuracy.

Keywords