Physics-informed Supervised Residual Learning for Electromagnetic ModelingThanks: This preprint has been published in IEEE Transactions on Antennas and Propagation on 01 March 2023. Please cite the final published version as [T. Shan et al., ”Physics-Informed Supervised Residual Learning for Electromagnetic Modeling,” in IEEE Transactions on Antennas and Propagation, vol. 71, no. 4, pp. 3393-3407, April 2023, doi: 10.1109/TAP.2023.3245281]. The link is https://ieeexplore.ieee.org/abstract/document/10057211.Thanks: Digital Object Identifier 10.1109/TAP.2023.3245281Thanks: This work was supported in part by the China Postdoctoral Science Foundation under Grant 2022M711764, in part by the National Natural Science Foundation of China under Grant 61971263, the Institute for Precision Medicine, Tsinghua University, the Biren Technology, and the BGP Inc.Thanks: Tao Shan, Jinhong Zeng, Rui Guo, Maokun Li, Fan Yang and Shenheng Xu are with the State Key Laboratory on Microwave and Digital Communications, Beijing National Research Center for Information Science and Technology (BNRist), Department of Electronic Engineering, Tsinghua University, Beijing 100084, China. (e-mail:maokunli@tsinghua.edu.cn). Thanks: Xiaoqian Song is with National Institute of Metrology, Beijing, 100013, China.
Abstract
In this study, physics-informed supervised residual learning (PhiSRL) is proposed to enable an effective, robust, and general deep learning framework for 2D electromagnetic (EM) modeling. Based on the mathematical connection between the fixed-point iteration method and the residual neural network (ResNet), PhiSRL aims to solve a system of linear matrix equations. It applies convolutional neural networks (CNNs) to learn updates of the solution with respect to the residuals. Inspired by the stationary and non-stationary iterative scheme of the fixed-point iteration method, stationary and non-stationary iterative physics-informed ResNets (SiPhiResNet and NiPhiResNet) are designed to solve the volume integral equation (VIE) of EM scattering. The effectiveness and universality of PhiSRL are validated by solving VIE of lossless and lossy scatterers with the mean squared errors (MSEs) converging to (SiPhiResNet) and (NiPhiResNet). Numerical results further verify the generalization ability of PhiSRL.
Index Terms:
Physics-informed Supervised Residual Learning, Volume Integral Equation, Residual Neural Network, Fixed-point Iteration Method, Electromagnetic Scattering, Deep Learning, Electromagnetic modeling.I Introduction
Computational electromagnetics (CEM) is focused on efficient and accurate numerical algorithms to model electromagnetic phenomena governed by Maxwell’s equations[1]. CEM has wide applications in the scientific research and engineering[2, 3, 4, 5]. There exist a variety of numerical algorithms including the finite difference method (FDM)[1], the finite element method (FEM)[6] and the method of moments (MoM)[7, 8]. A common approach of these algorithms is to discretize and convert Maxwell’s equations into a system of linear matrix equations that can be solved by fixed-point iteration method[9], conjugate gradient method[10], etc. The matrix equations usually have a large number of unknowns that is computationally expensive.
The computational complexity can be reduced by fast algorithms, such as conjugate gradient-fast Fourier transform[11], adaptive integral method[12], multilevel fast multipole algorithm[13], etc. On the other hand, acceleration can be performed by dividing the computation into offline and online parts, such as the reduced basis method[14], the characteristic basis function[15] and the model-order reduction method[16]. Machine learning (ML) techniques are also applied to speed up the online computation[17, 18, 19].
Recently, deep learning (DL) is developing rapidly[20]. It improves the performance in image[21], speech[22] and video[23] processing. It has also been applied to physics and engineering, like intuitive physics[24, 25], computational fluid dynamics[26, 27], Hamiltonian mechanics[28], simulations of Symplectic integration and Lagrangians[29, 30]. In the field of CEM, DL is also studied to improve the computational efficiency by training offline to accelerate online computation[31, 32, 33, 34]. In EM imaging, various DL-based methods have been proposed to enhance the computational efficiency and accuracy of inversion[35, 36, 37, 38, 39, 40, 41]. DL is also integrated to assist the design of microwave devices, such as microwave circuit design[42], synthesis of array antennas[43], coding programmable metasurfaces[44], metasurface imager[45], etc. Despite the recent progress of applying DL to CEM, it is still challenging to design effective deep neural network (DNN) architectures for EM computation. Most works are fully data-driven and considering DNN as ”black-box” approximators without the inner reasoning [46, 31]. The architectures of DNNs are determined by either hyperparameter optimization[47] or trial and error method, which is practical but unexplainable and costly[47, 48].
Research has been reported to interpret DNNs by relating them to the theory of ordinary differential equations (ODEs) or partial differential equations (PDEs)[49, 50]. On one hand, DL can help reduce the curse of dimensionalities in solving PDEs, such as Schrödinger equation[51], Navier-stokes equation[52], Poisson’s equation[53], Burger’s equation[54], etc. The automatic differentiation of DL can also be used to solve PDEs by embedding them into the loss of DNNs[55, 56]. On the other hand, it allows to better understand the inner reasoning of DNNs inspired by PDEs and corresponding numerical algorithms[57, 58, 48]. The design of DNNs is guided by the numerical methods of PDEs to obtain improved performance and generalization ability, such as multi-grid method[59, 60], hierarchical matrices[61], Runge-Kutta method[62], etc. The properties of DNNs can be analyzed by PDE theory[49, 63] and the PDE interpretation of DNNs further bridges the gap between DL and PDEs[57, 48].
In this work, we propose the physics-informed supervised residual learning (PhiSRL) that enables an effective, robust, and general framework for solving EM wave equations. It is based on the ResNet[64]. The mathematical link between the fixed-point iteration method and ResNet is investigated. PhiSRL is designed based on this link. In each iteration of PhiSRL, a parameterized function based on CNN is learned to modify the candidate solution to minimize the calculated residual. PhiSRL can be considered as a learned iterative solver of matrix equations and can also be expanded to solve other PDEs. Inspired by the stationary and non-stationary iterative schemes of the fixed-point iteration method, we design two different neural network architectures of PhiSRL including stationary and non-stationary iterative physics-informed ResNets (SiPhiResNet and NiPhiResNet). They are validated by solving EM scattering by 2D dielectric scatterers and achieve good computing precisions. In case of lossless scatterers, the MSEs of SiPhiResNet and NiPhiResNet converge to and . For lossy scatterers, the MSEs converge to and . Compared to MoM, by dividing the entire computation into offline and online parts, PhiSRL demonstrates a significant reduction in online computing time after offline training. Meanwhile, the high-performance computing platform GPU enables massively parallel computation of PhiSRL, which guarantees its computational efficiency. Furthermore, numerical results demonstrate that PhiSRL can generalize to solve EM scattering by a wide variety of geometries and frequencies different from the ones in the training data set. The preliminary results have been reported in [65].
This paper is organized as follows. Section II investigates the mathematical connections between ResNet and fixed-point iteration method. Section III introduces the volume integral equation and the corresponding numerical methods including MoM and PhiSRL. In Section IV, the universality of PhiSRL is validated by applying SiPhiResNet and NiPhiResNet to solve VIEs. Furthermore, the generalization abilities are verified. Observations and discussions are summarized in Section V.
II Physics-informed Supervised Residual Learning
The equations in EM modeling can be expressed as:
| (1) |
where denotes the field distribution, is the coordinate defined in , is a function represents the excitation, represents integral or differential operator, and denotes parameters in the definition of . Boundary conditions should be included to uniquely determine the solution :
| (2) |
where denotes an operator on the boundary and is the value on the boundary. In EM modeling, Eq. (1) and Eq. (2) are converted into a linear system of matrix equations:
| (3) |
Eq. (3) can be solved by direct matrix inversion, i.e, LU decomposition or iterative methods such as the conjugate gradient method [10]. Here we focus on the fixed-point iteration method[9].
II-A Fixed-point Iteration Method
The fixed-point iteration method can be expressed as[10]:
| (4) |
where is an approximation of in Eq. (3), denotes the -th candidate solution, and the operator denotes the matrix-vector multiplication . Eq. (4) is the stationary iterative scheme of the fixed-pointed iteration method where is kept the same at each iteration. The non-stationary iterative scheme can be written as:
| (5) |
Two widely used iterative methods are Richardson and Jacobi methods. Both of them can be written in the form of Eq. (4)[9].
II-B Physics-informed Supervised Residual Learning
ResNet is an effective neural network architectures in image processing[64]. It is a stacked structure composed of a family of modular blocks named as residual blocks[64]. The residual blocks enable ResNet to be easily extended to deep architectures. ResNet have been analyzed based on the theory of ODEs/PDEs and the corresponding numerical algorithms[66], such as forward Euler discretization of ODEs[49, 50], transport equation of control problem[63], dynamical systems[67].
The typical structure of a single residual block is depicted in Figure 1 that can be expressed as[64]:
| (6) | ||||
| (7) |
where and are input and output of the -th residual block; denotes the linear projection; denotes the nonlinear activation; denotes the transform function of the residual and is the parameter set of . It is pointed out in [68] that ResNet can achieve better performance and improved generalization ability by taking and as the identity mappings. As shown in Figure 1, Eq. (6) and Eq. (7) can be combined as:
| (8) |
It can be observed that the identity residual block Eq. (8) has a similar update equation as the fixed-point iteration method eqs. 5 and 6. Motivated by this link, PhiSRL is proposed by embedding the fixed-point iteration method into the identity residual block. The stationary scheme can be written as:
| (9) |
and the nonstationary one is:
| (10) |
where and denote the CNNs and the corresponding parameters. Figure 1 illustrates the structure of a PhiSRL block. As shown in Figure 1, the residual is calculated first regarding the input :
| (11) |
In each iteration of PhiSRL, CNNs are trained to predict the modification of the candidate solution with the calculated residual as input. The training process enables CNNs in PhiSRL to learn effective updates of the solution.
II-C Stationary Iterative Physics-informed ResNets
SiPhiResNet is designed based on the stationary iterative scheme of the fixed-point iteration method. Formulated as Eq. (9), the CNNs of SiPhiResNet share the same structure and parameter set, which can be considered as the recurrent neural network (RNN). The update equation of SiPhiResNet is (as shown in Figure 2):
| (12) |
where denotes the concatenation operation, , and denote the candidate solution, the corresponding residual and modification. SiPhiResNet requires the shared CNNs to possess strong learning capacities to handle the different mappings between residuals and modifications in every iteration.
adopts U-Net[69] as shown in Figure 2. U-Net is an effective network in image segmentation with a good learning ability[69]. U-Net has almost the same computing process with the V-cycle scheme of the multi-grid method[60]. It is noted that U-Net shares the same parameter set through the iterative process. With the RNN-like structure, the output is also concatenated with the residual as input to the next iteration. Here, plays a role of the hidden state that is widely applied in RNNs. Thus, the channels of input and output can be denoted as and . The value of depends on the linear equation systems, for example, will be in the real-valued equations and in the complex-valued equations.
II-D Non-stationary Iterative Physics-informed ResNets
Inspired by the non-stationary scheme of the fixed-point iteration method, NiPhiResNet is designed by using independent CNNs to update the candidate solution in each iteration. The CNNs share the same structure but owns independent and different parameter sets. It can be considered as the unrolled structure of the SiPhiResNet, as shown in Figure 3. According to Eq. (10), the specific iteration update equation of NiPhiResNet can be written as:
| (13) |
With as the input of the -iteration CNN, the computation can be described as:
| (14) |
where , , denote the tanh nonlinear activation, batch normalization and convolution, is the number of layers of the CNN and is set as 5 in this work. The independent CNN of each iteration tackles the learning task of a single iteration. Therefore, the CNN does not require a complicated structure. The specific hyperparameters of the CNN are mostly determined by author’s experience and trials. It should be emphasized that the nonlinearity function needs to be capable of providing both positive and negative values. The output has the same size as the input due to the fully convolutional operations.
II-E Stability Analysis of Physics-informed ResNets
Both SiPhiResNet and NiPhiResNet are built on top of ResNets and they inherit the properties of ResNet. The stability analysis of ResNet has been investigated in [68] and a similar analysis can be applied to SiPhiResNet and NiPhiResNet. Here, we take NiPhiResNet as an example. With the update equation of NiPhiResNet described as Eq. (13), we can define the loss function as MSE and can be written as:
| (15) |
where and are ground truth and the prediction at the -th iteration, is the Frobenius norm. Then, based on the chain rule, the back propagation of with respect to the -th input can be written as[68]:
| (16) | ||||
where is the input of the -th iteration. By removing the residual connections, Eq. (13) can be expressed as:
| (17) |
Then the back propagation of can be re-written as:
| (18) |
For gradient vanishing, the following assumption can be made:
| (19) |
Adding the residual connections convert Eq. (19) into:
| (20) |
Then, the limit converges. Since is finite, we can conclude that converges to a non-zero value according to Eq. (16). This demonstrates that the residual connections can alleviate the problem of gradient vanishing and improve the training stability[68]. The same conclusion can be derived for SiPhiResNet as well.
III Volume Integral Equation
The effectiveness and universality of PhiSRL are verified by solving the VIE[1, 70]. VIE computes the scattered field of the dielectric scatterers in free space , as shown in Figure 4. The electrical properties of dielectric scatterers are assumed to only vary along the lateral axes. The total electric field is induced when the dielectric scatterer is illuminated by the incident field . Taking transverse magnetic (TM) mode into account, the relationship between and satisfies:
| (21) | ||||
where is the wavenumber, is the Green’s function defined in [1], and is the contrast:
| (22) |
where , , , , denote vacuum permittivity, relative permittivity, conductivity, angular frequency and the position vector in free space respectively. The scattered field can be calculated by:
| (23) |
where is the position vector of the receiver and is the Green’s function defined in the observation domain [1].
III-A Method of Moments
To determine the scattered field with respect to the incident field , Eq. (21) needs to be solved by MoM for the total field in . By discretizing domain into subdomains, Eq. (21) can be written as[1, 70]:
| (24) | ||||
where and are the indices of subdomains in , is the Hankel function of the second kind. Then Eq. (24) can be converted as a linear equation system:
| (25) |
where is an identity matrix and is a matrix of size .
III-B Physics-informed Supervised Residual Learning
SiPhiResNet is first applied to solve Eq. (25). The -th update equation of SiPhiResNet refers to Eq. (12). Then, NiPhiResNet is applied to solve Eq. (25) and its -th update equation follows Eq. (13). It is noted that the calculation of the residual is based on Eq. (25):
| (26) |
In this paper, the is taken as the initial guess of PhiSRL and it can be regarded as the input of PhiSRL. The and are needed for the evaluation of in Eq. (26). They can be viewed as the components of PhiSRL. The objective function of them is the MSE that can be written as:
| (27) |
where is the total field computed by PhiSRL, is the total field by MoM, is the Frobenius norm and is the number of elements in the matrix of .
IV Numerical Results and Analysis
In this section, we verify PhiSRL by applying both SiPhiResNet and NiPhiResNet to solve 2D VIE. The model setup is illustrated in Figure 4 and the size of domain is . is discretized with uniform grids. One transmitter is placed outside and its distance from the center of is 1.67. The frequency of incident field is 3GHz and different incident angles are taken into account. It is noted that the coefficient matrix and right-hand side term of Eq. (25) vary with and the direction of incident wave .
IV-A Lossless scatterers
The angle of incident wave is randomly selected from , , , , , , , , as shown in Figure 5. The contrast of lossless scatterers can be derived from Eq. (22):
| (28) |
Three cylinders with random positions and radii are located in . The real parts of their contrasts vary from 0 to 1, 1 to 2 and 0 to 2 while the imaginary parts are 0, as shown in Figure 6.
Both SiPhiResNet and NiPhiResNet are implemented in Pytorch, and the computing platform is one Nvidia V100 GPU. The stochastic optimizer of SiPhiResNet and NiPhiResNet is Adam[71]. The learning rate is initialized as 0.002 and multiplied by 0.8 every 20 epochs. MoM is applied to compute the field in . A total of 40000 models are solved and form the data set, of which 32000 are as the training data set and 8000 are as the testing data set.
IV-A1 SiPhiResNet
SiPhiResNet is assumed to have three iterations and the average MSE converges below , as shown in Figure 7. The training and testing MSE agree well with each other, which indicates little overfitting. The little overfitting can be reduced by stopping early or adding regularizations. The U-Net of SiPhiResNet shares the same structure and parameter space throughout the whole iterative process. This means that U-Net needs to learn a wide variety of mappings between residuals and modifications in all iterations. Thus, the optimization of U-Net is not stable enough and fluctuates several times as reflected in Figure 7. Figure 8 demonstrates the detailed comparisons of total field computed by SiPhiResNet () and MoM () that are randomly selected from the testing data set. It can be observed that and are in a good agreement with low level of absolute error. The updated total field of each iteration is depicted in Figure 9 for a better insight into SiPhiResNet. The initial guess is the incident field. The difference between and decreases with the increase of iterations. The MSE of each iteration is also denoted in Figure 9. Furthermore, the histogram of mean absolute error (MAE) between and is shown in Figure 10. The MAE’s mean and standard deviation (std) of training and testing data sets agree well with each other that is consistent with the MSE convergence curve in Figure 7.
| Method | MAE-R* (mean/stds) | MAE-I** (mean/stds) |
|---|---|---|
| SiPhiResNet | ||
| NiPhiResNet |
- *
MAE of total field real parts
- **
MAE of total field imaginary parts
IV-A2 NiPhiResNet
NiPhiResNet consists of seven independent PhiSRL blocks and the convergence curve of MSE is shown in Figure 11. With a good agreement, training and testing MSEs decrease steadily and converge below . NiPhiResNet demonstrates better computing precisions than SiPhiResNet. Although NiPhiResNet has more iterations, the CNN of each iteration owns simpler structure than the U-Net of SiPhiResNet. It indicates that it is better to use an independent CNN to learn update rules in a single iteration. Figure 12 shows two results of total fields solved by NiPhiResNet and they are randomly chosen from the testing data set. Total fields solved by NiPhiResNet have good precisions with different incident angles and contrast distributions. Figure 13 illustrates the updated total field of each iteration. The total field of the first iteration is a rough approximation of ground truth, and it is refined with the increase of iterations. Figure 14 plots NiPhiResNet’s MAE histogram of training and testing data sets. The corresponding means and stds are in a good agreement. Small values of means and stds indicate stable computing precisions of NiPhiResNet.
IV-A3 Generalization Ability on Contrast Shape
The generalization ability of SiPhiResNet and NiPhiResNet on unseen contrast shapes is verified. Eight types of unseen contrast shapes are considered as shown in Figure 15. The range of contrasts is and 40 samples of each shape are generated for the validation data set. Table I shows the MAEs of total field solved by SiPhiResNet and NiPhiResNet in the validation data set. Figure 16 plots the samples of total fields solved by SiPhiResNet and NiPhiResNet. Both of them maintain a small error level on the generalization validation data set.
IV-A4 Generalization Ability on Incident Frequency
The training and testing data set of SiPhiResNet and NiPhiResNet are generated by fixing the incident frequency at 3GHz. To verify the generalization ability of SiPhiResNet and NiPhiResNet on different incident frequencies, we consider 12 frequencies of incident field: 2GHz, 2.3GHz, 2.6GHz, 2.9GHz, 3.1GHz, 3.2GHz, 3.4GHz, 3.7GHz, 4GHz, 4.5GHz, 5GHz. Eight types of contrast shapes and eight directions of incident waves are taken into account, as shown in Figure 15 and Figure 5. We generate 320 samples for each incident frequency. Figure 17 shows the MSE of SiPhiResNet and NiPhiResNet at different incident frequencies. Both of them keep good computing precisions across a broad range of incident frequencies, which validates their generalization ability on unseen incident frequencies.
IV-B Lossy scatterers
In this section, we apply PhiSRL to solve VIEs with lossy scatterers to verify its universality because lossy scatterers have different interactions with incident fields compared to lossless ones. Four cylinders with random positions and radii are located in . The formulation of contrast is Eq. (22) and the ranges of contrasts can refer to Table II. Figure 18 shows six examples. The incident frequency is fixed at 3GHz and the incident angle is randomly selected in , as shown in Figure 19. 40000 data samples are generated by MoM of which 80% and 20% are for training and testing respectively. The computing platform and training setup of SiPhiResNet and NiPhiResNet are the same as the ones in the cases of lossless scatterers.
| Cylinder | Real Part | Imaginary Part |
|---|---|---|
| Cylinder 1 | ||
| Cylinder 2 | ||
| Cylinder 3 | ||
| Cylinder 4 |
IV-B1 SiPhiResNet
SiPhiResNet has three iterations and Figure 20 shows the convergence curve of MSE with the final average MSE below . The training and testing MSEs are in a good agreement. The training process has a few fluctuations at the beginning and then becomes stabilized gradually. Figure 21 shows two comparisons between MoM and SiPhiResNet computed total fields ( and ). Figure 22 shows the updated total field of three iterations. The total field of iteration is close to ground truth and it improves gradually with increase of iterations. The histogram of MAE between and is depicted in Figure 23. The means and stds of both real and imaginary parts of fields are consistent that verifies the MSE convergence curve in Figure 20.
IV-B2 NiPhiResNet
NiPhiResNet has seven iterations and the MSE convergence curve is shown in Figure 24. With the converged average MSE below , NiPhiResNet shows better computing precisions than SiPhiResNet. Figure 25 illustrates two results randomly chosen from the testing data set. The absolute error distributions show small discrepancies between and . The field of each iteration is depicted in Figure 26. The updated total field is refined gradually with iterations. Figure 27 charts the MAE histogram of total fields solved by NiPhiResNet. The MAE’s mean and std of testing data set are a little higher than the ones of training data set while the discrepancy is very small and acceptable. Small stds of MAE also validates the stable computing precisions of NiPhiResNet.
IV-B3 Impact of Data Set Size on Performance
NiPhiResNet is taken as an example to investigate the impact of data set size on the performance of NiPhiResNet. A total of 40000 data samples are generated and divided into 4 data sets of which have 10000, 20000, 30000 and 40000 data samples respectively. Each data set are separated for training and testing according to 80%-20% ratio. With different data set sizes, NiPhiResNet is trained under the same training configuration. Figure 28 shows the corresponding final losses of NiPhiResNet. Both training and testing losses decrease with the increase of the data set size. Trained with 40000 data samples, NiPhiResNet demonstrates the best performance, and the difference between training and testing losses is also smallest, which indicates NiPhiResNet is trained efficiently. The training time of each data set size is also plotted in Figure 28. It can be observed that the training time is generally proportional to the data set size.
IV-B4 Generalization Ability on Contrast Shape
The generalization abilities of SiPhiResNet and NiPhiResNet on the unseen contrast shapes are verified. Same as the cases of lossless scatterers, 8 types of unseen contrast shapes are taken into account, as shown in Figure 29. The real and imaginary parts of contrasts are randomly selected in and respectively. The validation data set has a total of 320 data samples and 40 data samples are generated for each shape type. The MAEs of total fields solved by SiPhiResNet and NiPhiResNet are compared in Table III. The MAE’s stds of SiPhiResNet are higher than the ones of NiPhiResNet that indicates NiPhiResNet has more stable computing precisions on unseen contrast shapes. Figure 30 illustrates the randomly chosen total fields solved by SiPhiResNet and NiPhiResNet.
| Method | MAE-R* (mean/stds) | MAE-I** (mean/stds) |
|---|---|---|
| SiPhiResNet | ||
| NiPhiResNet |
- *
MAE of total field real parts
- **
MAE of total field imaginary parts
IV-B5 Generalization Ability on Incident Frequency
In this section, the generalization abilities of SiPhiResNet and NiPhiResNet are verified on incident frequencies unseen at training time. The incident frequency is fixed at 3GHz when generating training and testing data sets. Here, we consider 12 incident frequencies from 2GHz to 5GHz with a step of 0.3GHz. The incident angles vary in , as shown in Figure 19. For each incident frequency, we generate 320 data samples, and the contrast shapes keep consistent with the contrast generalization validation data set, as shown in Figure 29. The MSEs of SiPhiResNet and NiPhiResNet at different incident frequencies are plotted in Figure 31. Both of them maintain a good level of computing precision across a wide band of incident frequencies.
Furthermore, we study the reasons behind the good generalization ability of PhiSRL on different incident frequencies. Two cylinders are considered with their contrasts fixed as . The incident angle is set as and two different frequencies are employed including GHz and GHz. NiPhiResNet is applied to perform predictions and Figure 32 illustrates the corresponding contrasts, true total fields, and updated residuals at different iterations. In PhiSRL, the residuals are input of CNNs and can be calculated via . The calculated residuals can effectively alleviate the effects caused by the varying incident frequencies, which is further verified in Figure 32. It can be observed in Figure 32 that the updated residuals share similar distributions at the same iteration despite the different incident frequencies. This simplifies the learning burden of CNNs and enables a better generalization ability on a wide range of incident frequencies.
IV-C Generalization on Contrast Value
The generalization ability of PhiSRL is verified on the different contrast values that are unseen at training time. Here, we take NiPhiResNet as an example. When generating training data, the range of real and imaginary parts of contrast are and respectively. Two cylinders are taken into account, of which the contrast values are out of range, including , , , and . The incident angle is fixed as . Figure 33 compares ground truth and NiPhiResNet predictions. NiPhiResNet demonstrates a stable performance on the out-of-range contrast values, which further validates its generalization ability.
IV-D Comparison with the vanilla ResNet
The performance of PhiSRL is also compared to the ResNet with identity mappings which is called as the vanilla ResNet for brevity. Due to that the vanilla ResNet applies an independent CNN in each block, NiPhiResNet is taken as an example here. The vanilla ResNet has 7 iterations and the employed CNN shares the same structure as the one in PhiSRL. It takes as input the concatenation of the contrast and incident field arrays. The input channel is 4 but the channel of CNN prediction needs to be 2. Thus, a convolution is added in the first block to change the channel from 4 to 2, as shown in Figure 34. It should be emphasized that the main difference between the vanilla ResNet and NiPhiResNet is whether to incorporate the numerical calculation of . The training configuration and data set of the vanilla ResNet are also kept the same as the ones of NiPhiResNet. After 200 training epochs, the final training and testing MSEs of the vanilla ResNet are and respectively, while those for NiPhiResNet are and . Such comparison not only validates the efficacy of PhiSRL, but also demonstrates the benefits of incorporating the numerical computation of .
| Scatterers | Method | MAE (real/imag) | Time(s)/Reduction |
|---|---|---|---|
| lossless | BiCG2 | ||
| lossless | BiCG3 | ||
| lossless | BiCG3GPU | ||
| lossless | SiPhiResNet | ||
| lossless | BiCG6 | ||
| lossless | BiCG7 | ||
| lossless | BiCG7GPU | ||
| lossless | NiPhiResNet | ||
| lossy | BiCG2 | ||
| lossy | BiCG3 | ||
| lossy | BiCG3GPU | ||
| lossy | SiPhiResNet | /63.8% | |
| lossy | BiCG6 | ||
| lossy | BiCG7 | ||
| lossy | BiCG7GPU | ||
| lossy | NiPhiResNet |
- •
BiCG2,3,6,7 denote BiCGSTAB with 2, 3, 6, 7 iterations. BiCG3GPU and BiCG7GPU denote BiCGSTAB computed on GPU. The time reductions of SiPhiResNet and NiPhiResNet are compared to BiCG3 and BiCG7 respectively.
IV-E Comparisons of Computing Performance
In this section, we compare the computing performance of SiPhiResNet, NiPhiResNet and biconjugate gradient stabilized method (BiCGSTAB). BiCGSTAB is a widely used Krylov subspace method for solving linear matrix equations[72]. It is applied to evaluate the computing performance of SiPhiResNet and NiPhiResNet. Here, we adopt the data sets employed in Section IV-A and Section IV-B and each of them consists of 40000 data samples. In BiCGSTAB, the relative residual is an important stop criterion of solving matrix equations which can be written as:
| (29) |
The total fields solved by BiCGSTAB are taken as ground truth with relative residuals below . As SiPhiResNet and NiPhiResNet have 3 and 7 iterations respectively, BiCGSTAB is applied with and iterations. Then we evaluate the MAEs between the ground truth and the total fields solved by SiPhiResNet, NiPhiResNet and BiCGSTAB, as shown in Table IV. Both SiPhiResNet and NiPhiResNet maintain a good level of computing precisions compared to BiCGSTAB. Table IV also demonstrates the computing time of SiPhiResNet, NiPhiResNet and BiCGSTAB. The computing platform of BiCGSTAB is Intel(R) Xeon(R) Gold 5118 CPU @ 2.30GHz. The BiCGSTAB owns better precisions than SiPhiResNet and NiPhiResNet, while both SiPhiResNet and NiPhiResNet demonstrate faster computing speed than BiCGSTAB. NiPhiResNet needs less computing time than SiPhiResNet due to that NiPhiResNet applies simpler and smaller CNN architectures.
Furthermore, the computing time of PhiSRL is compared to BiCGSTAB when they achieve the same MAE level. The BiCGSTAB of 2 and 6 iterations can achieve the similar MAE level with SiPhiResNet and NiPhiResNet respectively. The corresponding computational time is summarized in Table IV. It can be observed that SiPhiResNet and NiPhiResNet still maintain a faster computing speed. Besides, we implement BiCGSTAB on one Nvidia V100 GPU that is the same computing platform for PhiSRL. The problem scale is small in this paper and the communication between CPU and GPU is time-consuming. The BiCGSTAB on GPU does not show the significant acceleration, as shown in Table IV.
IV-F Discussions
The word ”physics-informed” in PhiSRL is intended to convey that the supervised residual learning is informed of physics by incorporating physical operators. Such incorporation not only guarantees the interpretability of PhiSRL, but also simplifies and specifies the learning tasks of the applied CNNs. The universality of PhiSRL is validated by solving VIEs of lossless and lossy scatterers. Two different architectures of PhiSRL are discussed including SiPhiResNet and NiPhiResNet. They both demonstrate a good computational precision after adequate training. NiPhiResNet shows better computing precisions than SiPhiResNet. It can be tentatively concluded that it is recommended to apply independent CNNs to learn the update rules in PhiSRL. Numerical results also shows that PhiSRL has a good generalization ability on contrast shape, contrast value and incident frequency. Benefiting from the parallel computing on GPUs, PhiSRL shows a great potential to achieve a significant reduction in computing time. However, the generation of data samples and the training of PhiSRL are time-consuming. The numerical calculation of residuals is based on the dense matrix-vector multiplication in this paper and it needs to be further accelerated by fast Fourier transform in the case of large-scale problems.
V Conclusion
In this paper, we propose the physics-informed supervised residual learning to enable an effective, robust and general deep learning framework for electromagnetic modeling. Motivated by the mathematical link between ResNet and fixed-point iteration method, PhiSRL is designed on top of fixed-point iteration method by applying CNNs to learn update rules at each iteration. SiPhiResNet and NiPhiResNet are proposed based on the stationary and non-stationary scheme of fixed-point iteration method. With the aim of solving a linear system of matrix equations, PhiSRL is not limited to a specific EM forward modeling problem and can be easily extended to other scenarios. The universality of PhiSRL is validated by solving VIE of lossless and lossy scatterers. The MSEs of SiPhiResNet and NiPhiResNet achieve below and when solving VIEs of lossless scatterers, and converge below and when solving VIEs of lossy scatterers. PhiSRL further proves its generalization ability on the contrast shapes, incident frequencies, out-of-range contrast values that are unseen at training time. Numerical experiments demonstrate that the non-stationary scheme of PhiSRL (NiPhiResNet) can achieve better computing precisions with an independent CNN learning update rule of single iteration.
References
- [1] J.-M. Jin, Theory and computation of electromagnetic fields. John Wiley & Sons, 2011.
- [2] E. F. Knott, J. F. Schaeffer, and M. T. Tulley, Radar cross section. SciTech Publishing, 2004.
- [3] D. Poljak, Advanced modeling in computational electromagnetic compatibility. John Wiley & Sons, 2007.
- [4] J.-M. Jin and D. J. Riley, Finite element analysis of antennas and arrays. Wiley Online Library, 2009.
- [5] N. K. Nikolova, “Microwave biomedical imaging,” Wiley Encyclopedia of Electrical and Electronics Engineering, pp. 1–22, 1999.
- [6] J.-M. Jin, The finite element method in electromagnetics. John Wiley & Sons, 2015.
- [7] R. F. Harrington, Field computation by moment methods. Wiley-IEEE Press, 1993.
- [8] W. C. Chew, M. S. Tong, and B. Hu, Integral Equation Methods for Electromagnetic and Elastic Waves . Morgan and Claypool Publishers, 2007.
- [9] O. Axelsson, Iterative solution methods. Cambridge university press, 1996.
- [10] G. H. Golub and C. F. Van Loan, “Matrix Computations. edition,” 1996.
- [11] T. Sarkar, E. Arvas, and S. Rao, “Application of FFT and the conjugate gradient method for the solution of electromagnetic radiation from electrically large and small conducting bodies,” IEEE Transactions on Antennas and Propagation, vol. 34, no. 5, pp. 635–640, 1986.
- [12] E. Bleszynski, M. Bleszynski, and T. Jaroszewicz, “AIM: Adaptive integral method for solving large-scale electromagnetic scattering and radiation problems,” Radio Science, vol. 31, no. 5, pp. 1225–1251, 1996.
- [13] V. Rokhlin, “Rapid solution of integral equations of classical potential theory,” Journal of computational physics, vol. 60, no. 2, pp. 187–207, 1985.
- [14] X. Dang, M. Li, F. Yang, and S. Xu, “Quasi-Periodic Array Modeling Using Reduced Basis Method,” IEEE Antennas and Wireless Propagation Letters, vol. 16, pp. 825–828, 2017.
- [15] V. Prakash, , and R. Mittra, “Characteristic basis function method: A new technique for efficient solution of method of moments matrix equations,” Microwave and Optical Technology Letters, vol. 36, no. 2, pp. 95–100, 2003.
- [16] W. H. Schilders, H. A. Van der Vorst, and J. Rommes, Model order reduction: theory, research aspects and applications. Springer, 2008, vol. 13.
- [17] R. Chedid and N. Najjar, “Automatic finite-element mesh generation using artificial neural networks-Part I: Prediction of mesh density,” IEEE Transactions on Magnetics, vol. 32, no. 5, pp. 5173–5178, 1996.
- [18] F. Wang and Q.-J. Zhang, “Knowledge-based neural models for microwave design,” IEEE Transactions on Microwave Theory and Techniques, vol. 45, no. 12, pp. 2333–2343, 1997.
- [19] A. Massa, A. Boni, and M. Donelli, “A classification approach based on SVM for electromagnetic subsurface sensing,” IEEE Transactions on Geoscience and Remote Sensing, vol. 43, no. 9, pp. 2084–2093, 2005.
- [20] Y. LeCun, Y. Bengio, and G. Hinton, “Deep learning,” nature, vol. 521, no. 7553, p. 436, 2015.
- [21] K. Simonyan and A. Zisserman, “Very deep convolutional networks for large-scale image recognition,” arXiv preprint arXiv:1409.1556, 2014.
- [22] G. Hinton, L. Deng, D. Yu, G. E. Dahl, A.-r. Mohamed, N. Jaitly, A. Senior, V. Vanhoucke, P. Nguyen, T. N. Sainath et al., “Deep neural networks for acoustic modeling in speech recognition: The shared views of four research groups,” IEEE Signal processing magazine, vol. 29, no. 6, pp. 82–97, 2012.
- [23] G. Ciaparrone, F. L. Sánchez, S. Tabik, L. Troiano, R. Tagliaferri, and F. Herrera, “Deep learning in video multi-object tracking: A survey,” Neurocomputing, vol. 381, pp. 61–88, 2020.
- [24] W. Li, S. Azimi, A. Leonardis, and M. Fritz, “To fall or not to fall: A visual approach to physical stability prediction,” arXiv preprint arXiv:1604.00066, 2016.
- [25] T. Kipf, E. Fetaya, K.-C. Wang, M. Welling, and R. Zemel, “Neural relational inference for interacting systems,” in International Conference on Machine Learning. PMLR, 2018, pp. 2688–2697.
- [26] J. Tompson, K. Schlachter, P. Sprechmann, and K. Perlin, “Accelerating eulerian fluid simulation with convolutional networks,” in Proceedings of the 34th International Conference on Machine Learning-Volume 70. JMLR. org, 2017, pp. 3424–3433.
- [27] Y. Xie, E. Franz, M. Chu, and N. Thuerey, “tempogan: A temporally coherent, volumetric gan for super-resolution fluid flow,” ACM Transactions on Graphics (TOG), vol. 37, no. 4, pp. 1–15, 2018.
- [28] S. Greydanus, M. Dzamba, and J. Yosinski, “Hamiltonian neural networks,” arXiv preprint arXiv:1906.01563, 2019.
- [29] Z. Chen, J. Zhang, M. Arjovsky, and L. Bottou, “Symplectic recurrent neural networks,” arXiv preprint arXiv:1909.13334, 2019.
- [30] M. Cranmer, S. Greydanus, S. Hoyer, P. Battaglia, D. Spergel, and S. Ho, “Lagrangian neural networks,” arXiv preprint arXiv:2003.04630, 2020.
- [31] A. Massa, D. Marcantonio, X. Chen, M. Li, and M. Salucci, “DNNs as applied to electromagnetics, antennas, and propagation—A review,” IEEE Antennas and Wireless Propagation Letters, vol. 18, no. 11, pp. 2225–2229, 2019.
- [32] H. M. Yao and L. Jiang, “Machine-learning-based PML for the FDTD method,” IEEE Antennas and Wireless Propagation Letters, vol. 18, no. 1, pp. 192–196, 2018.
- [33] J.-J. Sun, S. Sun, Y. P. Chen, L. Jiang, and J. Hu, “Machine-Learning-Based Hybrid Method for the Multilevel Fast Multipole Algorithm,” IEEE Antennas and Wireless Propagation Letters, vol. 19, no. 12, pp. 2177–2181, 2020.
- [34] T. Shan, R. Guo, M. Li, F. Yang, S. Xu, and L. Liang, “Application of multitask learning for 2-d modeling of magnetotelluric surveys: Te case,” IEEE Transactions on Geoscience and Remote Sensing, vol. 60, pp. 1–9, 2021.
- [35] M. Salucci, M. Arrebola, T. Shan, and M. Li, “Artificial intelligence: New frontiers in real–time inverse scattering and electromagnetic imaging,” IEEE Transactions on Antennas and Propagation, 2022.
- [36] T. Shan, Z. Lin, X. Song, M. Li, F. Yang, and S. Xu, “A new approach for solving inverse scattering problems based on physics-informed supervised residual learning,” in 2022 16th European Conference on Antennas and Propagation (EuCAP). IEEE, 2022, pp. 1–4.
- [37] Z. Wei and X. Chen, “Deep-Learning Schemes for Full-Wave Nonlinear Inverse Scattering Problems,” IEEE Transactions on Geoscience and Remote Sensing, 2018.
- [38] T. Shan, Z. Lin, X. Song, M. Li, F. Yang, and S. Xu, “Neural born iterative method for solving inverse scattering problems: 2d cases,” IEEE Transactions on Antennas and Propagation, 2022.
- [39] Z. Wei, D. Liu, and X. Chen, “Dominant-current deep learning scheme for electrical impedance tomography,” IEEE Transactions on Biomedical Engineering, vol. 66, no. 9, pp. 2546–2555, 2019.
- [40] J. Liu, H. Zhou, T. Ouyang, Q. Liu, and Y. Wang, “Physical model-inspired deep unrolling network for solving nonlinear inverse scattering problems,” IEEE Transactions on Antennas and Propagation, 2021.
- [41] Z. Ma, K. Xu, R. Song, C.-F. Wang, and X. Chen, “Learning-based fast electromagnetic scattering solver through generative adversarial network,” IEEE Transactions on Antennas and Propagation, vol. 69, no. 4, pp. 2194–2208, 2020.
- [42] K. Roy, M. A. Dolatsara, H. M. Torun, R. Trinchero, and M. Swaminathan, “Inverse design of transmission lines with deep learning,” in 2019 IEEE 28th Conference on Electrical Performance of Electronic Packaging and Systems (EPEPS). IEEE, 2019, pp. 1–3.
- [43] T. Shan, M. Li, S. Xu, and F. Yang, “Phase synthesis of beam-scanning reflectarray antenna based on deep learning technique,” Progress In Electromagnetics Research, vol. 172, pp. 41–49, 2021.
- [44] T. Shan, X. Pan, M. Li, S. Xu, and F. Yang, “Coding Programmable Metasurfaces Based on Deep Learning Techniques,” IEEE Journal on Emerging and Selected Topics in Circuits and Systems, vol. 10, no. 1, pp. 114–125, 2020.
- [45] L. Li, H. Ruan, C. Liu, Y. Li, Y. Shuang, A. Alù, C.-W. Qiu, and T. J. Cui, “Machine-learning reprogrammable metasurface imager,” Nature communications, vol. 10, no. 1, pp. 1–8, 2019.
- [46] X. Chen, Z. Wei, M. Li, and P. Rocca, “A review of deep learning approaches for inverse scattering problems (invited review),” Progress In Electromagnetics Research, vol. 167, pp. 67–81, 2020.
- [47] J. M. Hernández-Lobato, M. A. Gelbart, R. P. Adams, M. W. Hoffman, and Z. Ghahramani, “A general framework for constrained bayesian optimization using information-based search,” 2016.
- [48] L. Ruthotto and E. Haber, “Deep neural networks motivated by partial differential equations,” Journal of Mathematical Imaging and Vision, pp. 1–13, 2019.
- [49] E. Weinan, “A proposal on machine learning via dynamical systems,” Communications in Mathematics and Statistics, vol. 5, no. 1, pp. 1–11, 2017.
- [50] E. Haber and L. Ruthotto, “Stable architectures for deep neural networks,” Inverse Problems, vol. 34, no. 1, p. 014004, 2017.
- [51] J. Hermann, Z. Schätzle, and F. Noé, “Deep-neural-network solution of the electronic Schrödinger equation,” Nature Chemistry, vol. 12, no. 10, pp. 891–897, 2020.
- [52] M. Raissi, A. Yazdani, and G. E. Karniadakis, “Hidden fluid mechanics: A navier-stokes informed deep learning framework for assimilating flow visualization data,” arXiv preprint arXiv:1808.04327, 2018.
- [53] T. Shan, W. Tang, X. Dang, M. Li, F. Yang, S. Xu, and J. Wu, “Study on a Fast Solver for Poisson’s Equation Based on Deep Learning Technique,” IEEE Transactions on Antennas and Propagation, vol. 68, no. 9, pp. 6725–6733, 2020.
- [54] J. Sirignano and K. Spiliopoulos, “DGM: A deep learning algorithm for solving partial differential equations,” Journal of computational physics, vol. 375, pp. 1339–1364, 2018.
- [55] M. Raissi, P. Perdikaris, and G. E. Karniadakis, “Physics-informed neural networks: A deep learning framework for solving forward and inverse problems involving nonlinear partial differential equations,” Journal of Computational Physics, vol. 378, pp. 686–707, 2019.
- [56] L. Lu, X. Meng, Z. Mao, and G. E. Karniadakis, “DeepXDE: A deep learning library for solving differential equations,” SIAM Review, vol. 63, no. 1, pp. 208–228, 2021.
- [57] Y. Lu, A. Zhong, Q. Li, and B. Dong, “Beyond finite layer neural networks: Bridging deep architectures and numerical differential equations,” in International Conference on Machine Learning. PMLR, 2018, pp. 3276–3285.
- [58] Z. Long, Y. Lu, X. Ma, and B. Dong, “Pde-net: Learning pdes from data,” in International Conference on Machine Learning. PMLR, 2018, pp. 3208–3216.
- [59] D. Greenfeld, M. Galun, R. Basri, I. Yavneh, and R. Kimmel, “Learning to optimize multigrid PDE solvers,” in International Conference on Machine Learning. PMLR, 2019, pp. 2415–2423.
- [60] J. He and J. Xu, “MgNet: A unified framework of multigrid and convolutional neural network,” Science china mathematics, vol. 62, no. 7, pp. 1331–1354, 2019.
- [61] Y. Fan, L. Lin, L. Ying, and L. Zepeda-Nunez, “A multiscale neural network based on hierarchical matrices,” 2019.
- [62] M. Zhu, B. Chang, and C. Fu, “Convolutional Neural Networks combined with Runge-Kutta Methods,” 2019.
- [63] Z. Li and Z. Shi, “Deep residual learning and pdes on manifold,” arXiv preprint arXiv:1708.05115, 2017.
- [64] K. He, X. Zhang, S. Ren, and J. Sun, “Deep residual learning for image recognition,” in Proceedings of the IEEE conference on computer vision and pattern recognition, 2016, pp. 770–778.
- [65] T. Shan, X. Song, R. Guo, M. Li, F. Yang, and S. Xu, “Physics-informed supervised residual learning for electromagnetic modeling,” in 2021 International Applied Computational Electromagnetics Society Symposium (ACES). IEEE, 2021, pp. 1–4.
- [66] B. Chang, L. Meng, E. Haber, L. Ruthotto, D. Begert, and E. Holtham, “Reversible architectures for arbitrarily deep residual neural networks,” in Proceedings of the AAAI Conference on Artificial Intelligence, vol. 32, no. 1, 2018.
- [67] B. Chang, L. Meng, E. Haber, F. Tung, and D. Begert, “Multi-level residual networks from dynamical systems view,” arXiv preprint arXiv:1710.10348, 2017.
- [68] K. He, X. Zhang, S. Ren, and J. Sun, “Identity mappings in deep residual networks,” in European conference on computer vision. Springer, 2016, pp. 630–645.
- [69] O. Ronneberger, P. Fischer, and T. Brox, “U-net: Convolutional networks for biomedical image segmentation,” in International Conference on Medical image computing and computer-assisted intervention. Springer, 2015, pp. 234–241.
- [70] X. Chen, Computational methods for electromagnetic inverse scattering. John Wiley & Sons, 2018.
- [71] D. P. Kingma and J. Ba, “Adam: A Method for Stochastic Optimization,” CoRR, vol. abs/1412.6980, 2014.
- [72] Y. Saad, Iterative methods for sparse linear systems. SIAM, 2003.