A Nonoverlapping Domain Decomposition Method for Extreme Learning Machines: Elliptic Problems

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Lee, Chang-Ock, Lee, Youngkyu, Ryoo, Byungeun
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918002962726912
author Lee, Chang-Ock
Lee, Youngkyu
Ryoo, Byungeun
author_facet Lee, Chang-Ock
Lee, Youngkyu
Ryoo, Byungeun
contents Extreme learning machine (ELM) is a methodology for solving partial differential equations (PDEs) using a single hidden layer feed-forward neural network. It presets the weight/bias coefficients in the hidden layer with random values, which remain fixed throughout the computation, and uses a linear least squares method for training the parameters of the output layer of the neural network. It is known to be much faster than Physics informed neural networks. However, classical ELM is still computationally expensive when a high level of representation is desired in the solution as this requires solving a large least squares system. In this paper, we propose a nonoverlapping domain decomposition method (DDM) for ELMs that not only reduces the training time of ELMs, but is also suitable for parallel computation. In numerical analysis, DDMs have been widely studied to reduce the time to obtain finite element solutions for elliptic PDEs through parallel computation. Among these approaches, nonoverlapping DDMs are attracting the most attention. Motivated by these methods, we introduce local neural networks, which are valid only at corresponding subdomains, and an auxiliary variable at the interface. We construct a system on the variable and the parameters of local neural networks. A Schur complement system on the interface can be derived by eliminating the parameters of the output layer. The auxiliary variable is then directly obtained by solving the reduced system after which the parameters for each local neural network are solved in parallel. A method for initializing the hidden layer parameters suitable for high approximation quality in large systems is also proposed. Numerical results that verify the acceleration performance of the proposed method with respect to the number of subdomains are presented.
format Preprint
id arxiv_https___arxiv_org_abs_2406_15959
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle A Nonoverlapping Domain Decomposition Method for Extreme Learning Machines: Elliptic Problems
Lee, Chang-Ock
Lee, Youngkyu
Ryoo, Byungeun
Numerical Analysis
Computational Engineering, Finance, and Science
Machine Learning
65N55 (Primary), 35J25, 68T07 (Secondary)
Extreme learning machine (ELM) is a methodology for solving partial differential equations (PDEs) using a single hidden layer feed-forward neural network. It presets the weight/bias coefficients in the hidden layer with random values, which remain fixed throughout the computation, and uses a linear least squares method for training the parameters of the output layer of the neural network. It is known to be much faster than Physics informed neural networks. However, classical ELM is still computationally expensive when a high level of representation is desired in the solution as this requires solving a large least squares system. In this paper, we propose a nonoverlapping domain decomposition method (DDM) for ELMs that not only reduces the training time of ELMs, but is also suitable for parallel computation. In numerical analysis, DDMs have been widely studied to reduce the time to obtain finite element solutions for elliptic PDEs through parallel computation. Among these approaches, nonoverlapping DDMs are attracting the most attention. Motivated by these methods, we introduce local neural networks, which are valid only at corresponding subdomains, and an auxiliary variable at the interface. We construct a system on the variable and the parameters of local neural networks. A Schur complement system on the interface can be derived by eliminating the parameters of the output layer. The auxiliary variable is then directly obtained by solving the reduced system after which the parameters for each local neural network are solved in parallel. A method for initializing the hidden layer parameters suitable for high approximation quality in large systems is also proposed. Numerical results that verify the acceleration performance of the proposed method with respect to the number of subdomains are presented.
title A Nonoverlapping Domain Decomposition Method for Extreme Learning Machines: Elliptic Problems
topic Numerical Analysis
Computational Engineering, Finance, and Science
Machine Learning
65N55 (Primary), 35J25, 68T07 (Secondary)
url https://arxiv.org/abs/2406.15959