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

Download:

Current browse context:

math.NA

Change to browse by:

References & Citations

Bookmark

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

Mathematics > Numerical Analysis

Title: A neuron-wise subspace correction method for the finite neuron method

Abstract: In this paper, we propose a novel algorithm called Neuron-wise Parallel Subspace Correction Method (NPSC) for training ReLU neural networks for numerical solution of partial differential equations (PDEs). Despite of extremely extensive research activities in applying neural networks for numerical PDEs, there is still a serious lack of training algorithms that can be used to obtain approximation with adequate accuracy. Based on recent results on the spectral properties of linear layers and landscape analysis for single neuron problems, we develop a special type of subspace correction method that deals with the linear layer and each neuron in the nonlinear layer separately. An optimal preconditioner that resolves the ill-conditioning of the linear layer is presented, so that the linear layer is trained in a uniform number of iterations with respect to the number of neurons. In each single neuron problem, a good local minimum is found by a superlinearly convergent algorithm, avoiding regions where the loss function is flat. Performance of the proposed method is demonstrated through numerical experiments for function approximation problems and PDEs.
Comments: 22 pages, 6 figures
Subjects: Numerical Analysis (math.NA)
MSC classes: 65D15, 65N22, 65N30, 65N55, 68T07
Cite as: arXiv:2211.12031 [math.NA]
  (or arXiv:2211.12031v2 [math.NA] for this version)

Submission history

From: Jongho Park [view email]
[v1] Tue, 22 Nov 2022 06:06:48 GMT (476kb,D)
[v2] Sun, 8 Jan 2023 08:19:02 GMT (554kb,D)

Link back to: arXiv, form interface, contact.