Next Article in Journal
Open Localization in 3D Package with TSV Daisy Chain Using Magnetic Field Imaging and High-Resolution Three-Dimensional X-ray Microscopy
Previous Article in Journal
Data-Driven Control Algorithm for Snake Manipulator
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

Multiple Optimizations-Based ESRFBN Super-Resolution Network Algorithm for MR Images

1
School of Electronics and Information Engineering, Harbin Institute of Technology, Harbin 150001, China
2
College of Computer Science and Engineering, Shandong University of Science and Technology, Qingdao 266590, China
*
Author to whom correspondence should be addressed.
Appl. Sci. 2021, 11(17), 8150; https://doi.org/10.3390/app11178150
Submission received: 6 August 2021 / Revised: 27 August 2021 / Accepted: 27 August 2021 / Published: 2 September 2021

Abstract

:
Magnetic resonance (MR) images can detect small pathological tissue with the size of 3–5 image pixels at an early stage, which is of great significance in the localization of pathological lesions and the diagnosis of disease. High-resolution MR images can provide clearer structural details and help doctors to analyze and diagnose the disease correctly. In this paper, MR super-resolution based on the multiple optimizations-based Enhanced Super Resolution Feed Back Network (ESRFBN) is proposed. The method realizes network optimization from the three perspectives of network structure, data characteristics and heterogeneous network integration. Firstly, a super-resolution network structure based on multi-slice input optimization is proposed to make full use of the structural similarity between samples. Secondly, aiming at the problem that the L1 or L2 loss function is based on a per-pixel comparison of differences, without considering human visual perception, the optimization method of multiple loss function cascade is proposed, which combines the L1 loss function to retain the color and brightness characteristics and the MS-SSIM loss function to retain the contrast characteristics of the high-frequency region better, so that the depth model has better characterization performance; thirdly, in view of the problem that large deep learning networks are difficult to balance model complexity and training difficulty, a heterogeneous network fusion method is proposed. For multiple independent deep super-resolution networks, the output of a single network is integrated through an additional fusion layer, which broadens the width of the network, and can effectively improve the mapping and characterization capabilities of high- and low-resolution features. The experimental results on two super-resolution scales and on MR images datasets of four human body parts show that the proposed large-sample space learning super-resolution method effectively improves the super-resolution performance.

1. Introduction

MR images can detect small pathological tissue at an early stage, which is of great significance in the localization of pathological lesions and the diagnosis of disease. As MR images with high-resolution can provide clearer structural details, which can help doctors to analyze and diagnose the disease correctly, it is more desirable to obtain high-resolution MR images in a stronger magnetic field [1]. However, the acquisition of high-resolution MR images requires a stronger magnetic field and longer radiation scanning time. Generally, on the MR imaging time, it maybe lasts about 6–12 min for one organ region [2]. In this paper, the high-resolution refers to the resolution of an image for one organ region. For example, for the same size of the organ region, the low-resolution is 512 × 512, and the high-resolution is 1024 × 1024. When the imaging process takes a longer time, the movement of the human body during the process will bring more noise [3]. The improvement of the resolution of MR images will be at the cost of a doubling of the imaging time, which greatly reduces the patient’s experience and easily brings psychological pressure to the patient. When improving the resolution of MR images, the use of software to improve the resolution of MR images in this paper is a cost-effective method compared with the hardware method of upgrading the imaging equipment. Compared with traditional interpolation methods, such as Bicubic Interpolation and Nearest Neighbor Interpolation [4], as well as reconstruction-based methods, such as Convex Set Projection [5] and Maximum Posterior [6], the dictionary learning method has stronger feature expression, and can better characterize the mapping relationship between high- and low-resolution MR images [7].
Bicubic interpolation is the interpolation method in which the value of the function at the point (x, y) can be obtained by the weighted average of the 16 nearest sampling points in the rectangular grid. The nearest neighbor interpolation method is a gray value interpolation method, which makes the gray value of the transformed pixel equal to the gray value of the input pixel closest to it. Super-resolution reconstruction technology refers to the reconstruction of a corresponding high-resolution image from one or more low-resolution images based on reconstruction. This is mainly because the dictionary learning method can introduce the image’s prior information from the trained data set, but its shallow model expression ability is limited; thus, the ability to learn the image’s prior information is limited.
In recent years, with the rapid development of deep learning technology, deep learning based super-resolution methods have been actively explored, which greatly promotes the development of image super-resolution technology. In 2014, Dong et al. proposed the Super Resolution Convolutional Neural Network (SRCNN) model [8], which applied deep learning to the field of image super-resolution for the first time. Since then, scholars have carried out studies from the aspects of network architecture, loss function type, learning principle and strategy, etc. A large number of image super-resolution methods based on deep convolutional networks have been proposed, and they have been improving the performance of super-resolution continuously [9], such as Very Deep Super Resolution (VDSR [10]), Residual Dense Network (RDN) [11], Super Resolution Generative Adversarial Network (SRGAN) [12], Deep Back-projection Network (DBPN) [13] and Super Resolution Feed Back Network (SRFBN) [14], etc., which have reached the most advanced performance on various benchmarks of image super-resolution. In previous works, the researchers developed some MR super-resolution algorithms. We describe as follows. BIC [15]: The Bicubic Interpolation (BIC) method is to use the value of 16 pixels around the pixel to be calculated to estimate the value of the unknown pixel. It is a more complex interpolation method than the Nearest Neighbor Interpolation and Bilinear Interpolation. The method uses a cubic polynomial to approximate the theoretically best interpolation function. IBP [16]: The Iterative Back Projection (IBP) is to adopt an iterative method to reduce the error between the image obtained by the simulated degradation process and the actual low-resolution image gradually, and then the error value between the two is backward projected to update the initial estimate of the high-resolution image and reconstruct the high-resolution image. POCS [5]: The Projection onto Convex Sets Approach (POCS) requires that a set of closed convex constraints be defined in a vector space within which the actual high-resolution image is contained. An estimate of a high-resolution image is defined as a point within the intersection of sets of these constraints. A high-resolution estimation image can be obtained by projecting any initial estimate onto these constrained sets. MAP [6]: The Maximum a Posterior (MAP) method is a probability-based algorithm framework, which is the most widely used method in practical application and scientific research at present, and many specific super-resolution algorithms can be classified into this probabilistic framework. The basic idea of the maximum posterior probability method is derived from conditional probability, which takes the known low-resolution image sequence as the observation result to estimate the unknown high-resolution image. JDL [7]: Compressive Sensing (CS) points out that an image can be accurately reconstructed from a set of sparse representation coefficients on an over-complete dictionary under very harsh conditions. Joint Dictionary Learning (JDL) can strengthen the similarity between the sparse representation of low-resolution and high-resolution image blocks and their corresponding real dictionaries through the joint training of low-resolution and high-resolution image block dictionaries. Therefore, the sparse representation of low-resolution image blocks and the high-resolution super-complete dictionary can work together to reconstruct the high-resolution image blocks, and then the high-resolution image blocks are connected to obtain the final complete high-resolution image. IJDL [17]: Aiming at the problem of dictionary joint cascade training only considers the joint image block pair error and does not consider the individual reconstruction error of the high- and low-resolution dictionary, Improved Joint Dictionary Learning (IJDL), which is a dictionary training method based on independent calculation of high- and low-resolution dictionary reconstruction errors, is proposed. This method optimizes the objective function, and takes the individual reconstruction errors of high- and low-resolution dictionaries into consideration, and discards the traditional cascade calculation method, effectively reducing the reconstruction error and improving the reconstruction accuracy of the image. RDN [11]: The Residual Dense Network (RDN) applies Residual Dense blocks. It can not only read the state of the previous residual dense block through a continuous memory mechanism, but also make full use of all the layers in it through local dense connection, and adaptively retain the accumulated features through Local Feature Fusion (LFF). The global residual learning is used to combine the shallow features with the deep features and make full use of the stratified features of the original low-resolution image. EDSR [18]: Enhanced Deep Residual Networks for Single Image Super-Resolution (EDSR) draws on the residual learning mechanism of the ResNet network. The input is divided into two paths after a layer of convolution. One path goes through the N-layer RDB module for convolution again, and the other leads directly to the intersection for weighted summation, and then the result is output after up-sampling and convolution. SRGAN [12]: SRGAN is a generative adversarial network. It adds a discriminator on the basis of SRResnet. Its function is to add an additional discriminator network and two losses, and to train the two networks in an alternate training way. ESRGAN [19]: Compared with SRGAN, ESRGAN increases the perceptual loss, and uses the Residual-in-Residual Dense Block (RDDB) module to replace the original Residual Block module, so that ESRGAN can obtain more realistic and natural textures. SRFBN [14]: SRFBN designed a feedback module, using the return mechanism to improve the effect of super-resolution. The advantage of a returnback is that no additional parameters are added, and multiple returnbacks are equivalent to deepening the network, refining the generated image. Although other networks have adopted similar return structures, these networks are unable to make the front layer obtain useful information from the back layer. ESRFBN [20]: For ESRFBN, the number of feature graphs and group convolutions is increased on the basis of the SRFBN network. The number of feature graphs is changed from 32 to 64, and the number of group convolution is also increased from 3 to 6.
However, these deep learning networks are mainly for natural image super-resolution tasks, not medical images or MR images. In this regard, medical image researchers have also paid attention to the progress of deep super-resolution, and transferred deep learning technology to MR images, and a number of deep-learning-based MR images super-resolution methods have emerged. For example, a Progressive Sub-band Residual learning Super-Resolution Network (PSR-SRN) is proposed in [21], which contains two parallel progressive learning streams. One stream passes through the sub-band residual learning unit to detect the missing high-frequency residuals, and the other stream focuses on the reconstruction of refined MR images. These two learning streams complement each other and learn the complex mapping between high- and low-resolution MR images. In [22], researchers proposed a new hybrid network, which improves the quality of MR images by increasing the width of the network. The hybrid block combines multi-path structure and the variation dense block, which can extract rich features from low-resolution images. In the previous work [23], the researcher combined the meta-learning technology with the Generative adversarial networks (GAN) to achieve super-resolution of MR images of any scale and high fidelity, and the meta-learning technology was used in the work [24]. Generative adversarial networks (GAN) are a deep learning model. It is one of the most promising methods of unsupervised learning on complex distribution in recent years. The model produces fairly good output through mutual game learning of two modules in the framework: generative model and discriminant model. GAN is most commonly used in image generation, such as super-resolution tasks, semantic segmentation and so on. The Super Resolution scale is the scaling factor. The scaling operation will adjust the size of the image according to the given scaling factor. In this paper, data sets with two super-resolution scales of ×2 and ×4 are generated in this paper.
However, the current deep learning technology still has three limitations when applying to MR images’ super-resolution. (1) The characteristics of MR images’ sequence are not fully considered. The imaging method of MR images is different from natural images. In the process of MR images, a series of images with similar structures can be obtained. Therefore, compared with natural images, MR sequence images can provide richer structural information and make the network more robust to noise. (2) At present, most deep networks adopt L1 or L2 loss functions, which are all based on a pixel-by-pixel comparison without considering human visual perception. At the same time, they are easy to fall into the local optimal solution, and it is difficult to obtain the optimal effect [20]. (3) The capability of feature representation of deep learning networks gradually increases with the deepening of the network. However, with the huge network structure, the training difficulty of the network also increases exponentially, which makes it difficult to achieve a good balance between model complexity and training difficulty.

2. Algorithm Architecture

Super-resolution methods based on deep learning have been successively applied to MR images. However, these methods simply use deep learning techniques to process MR image super-resolution tasks without fully considering the difference between natural images and medical images. Deep learning technology still has limitations in MR images’ super-resolution. In this paper, corresponding optimization methods are proposed for the three limitations of the application of current deep super-resolution networks in MR images.
Aiming at the problem that the characteristics of MR images’ sequence are not fully considered, this chapter proposes a multi-slice input strategy to provide more structural information for the network. The MR images are a sequence of images. Its adjacent slices have a similar structure, which can provide two-point-five dimension (2.5D) information for the network. It can obtain better results than two-dimension (2D), and has fewer parameters and higher efficiency than three-dimension (3D) convolution [25].
Aiming at the problem that the L1 or L2 loss function is based on a pixel-by-pixel comparison without considering human visual perception, where the L1 loss function is the Least Absolute value Deviation (LAD) loss function, and the L2 norm loss function is the Least Square Error (LSE) loss function. We propose a method of multi-loss function cascade optimization based on the analysis of the commonly used loss functions. By combining the characteristics of the L1 loss function to retain color and brightness and the characteristics of the MS-SSIM loss function to retain the contrast of the high-frequency region better, the depth model has better characterization performance [26].
Aiming at the problem that it is difficult for large deep learning networks to balance model complexity and training difficulty, this paper proposes a method based on heterogeneous network fusion [27]. Heterogeneous network fusion is the use of common technologies to realize the interconnection of different structures of super-resolution networks. In addition, the different structures have the different ability of MR super-resolution. Heterogeneous network fusion is to increase the super-resolution performance through applying enough of the advantages from these heterogeneous networks. For multiple independent deep super-resolution networks, the output of a single network is integrated through an additional fusion layer, or two independent networks are cascaded. This method essentially broadens the width and depth of the network, and can effectively improve the mapping and characterization capabilities of high- and low-resolution features, so as to obtain a better MR images’ super-resolution effect.
In order to solve the three limitations of deep super-resolution networks on MR images simultaneously, this paper proposes a multi-optimized ESRFBN super-resolution network algorithm for MR images based on the optimization methods of the previous three paragraphs. The algorithm architecture is shown in Figure 1. The three optimization methods are applied to the ESRFBN network at the same time, so that the multi-optimized ESRFBN network can get a better MR images’ super-resolution effect.

2.1. Cascade Optimization of Loss Function Based on Joint PSNR-SSIM

In order to improve the PSRN and SSIM indexes of MR super-resolution, a cascade optimization method based on the joint loss function of PSNR-SSIM is proposed in this section. SSIM as a loss function is not particularly sensitive to uniform deviation. This leads to changes in brightness or color, which usually become duller. However, SSIM can preserve contrast and is better than other loss functions in the high-frequency region. On the other hand, the L1 loss function can keep the brightness and color unchanged. Therefore, in order to obtain the best characteristics of the two loss functions, this section combines them to form a new loss function, and the name is Mix loss function. Its formula is as follows
L Mix = α L MS - SSIM + ( 1 α ) G σ G M L 1
The cascade optimization calculation steps of the loss function based on joint PSNR-SSIM are as follows.
  • Step 1. Calculate L MS - SSIM
SSIM/MS-SSIM is an index that measures the similarity of two images, mainly from three aspects: brightness, contrast and structure. Therefore, the SSIM index can better characterize the structural information of the image. As a loss function, it can better retain the high-frequency information of the image, so that the super-resolution image has more detailed information, which is useful for MR images in clinical diagnosis. Its expression is as follows. Where   y   is the original high-resolution image; y   is the high-resolution image output by the deep network; μ y is the mean of the image y ; μ y is the mean of the image y ; σ y 2 is the variance of the image y ; σ y 2 is the variance of the image y ; σ y y   is the covariance of the sum of the image y and y ; c 1 and c 2 are two constants, avoiding the denominator to be 0.
S S I M ( y , y ) = ( 2 μ y μ y + c 1 ) ( σ y y + c 2 ) ( μ y 2 + μ y 2 + c 1 ) ( σ y 2 + σ y 2 + c 2 )
The index range of SSIM is 0–1. The larger the SSIM is, the higher the similarity of the two images. Therefore, SSIM as the loss function can be expressed as the following formula [20].
L S S I M = 1 S S I M ( y , y )
In order to better transmit the error, the SSIM formula is deformed, and the dependence of the mean and standard deviation on the pixel p is ignored here. The mean and standard deviation are calculated using a Gaussian filter with a standard deviation G σ G .
S S I M ( y , y ) = ( 2 μ y μ y + c 1 ) ( σ y y + c 2 ) ( μ y 2 + μ y 2 + c 1 ) ( σ y 2 + σ y 2 + c 2 ) = l ( p ) · c s ( p )
At the same time, the SSIM loss function can be re-characterized as
L S S I M ( P ) = 1 S S I M ( p ~ )
p ~ represents the central pixel. Even if the network learns the weight to maximize the SSIM value of the central pixel, the learned kernel will be applied to all pixels in the image. Therefore, it is necessary to calculate the reciprocal of p ~ to other pixels q , and the derivation process is as follows
L S S I M ( P ) x ( q ) = x ( q ) S S I M ( p ~ ) = ( l ( p ~ ) x ( q ) · c s ( p ~ ) + l ( p ~ ) · c s ( p ~ ) x ( q ) )
where
l ( p ~ ) x ( q ) = 2 · G σ G ( q p ~ ) · ( μ y μ y l ( p ~ ) μ y 2 + μ y 2 + c 1 )
c s ( p ~ ) x ( q ) = 2 σ y 2 + σ y 2 + c 2 · G σ G ( q p ~ ) · [ ( y ( q ) μ y ) c s ( p ~ ) · ( y ( q ) μ y ) ]
G σ G ( q p ~ ) is the Gaussian coefficient associated with pixel q.
For MS-SSIM, it can be expressed as follows, where α = β j = 1
M S S S I M ( p ) = l M α ( p ) · j = 1 M c s j β j ( p )
Analogous to SSIM as a loss function, MS-SSIM as a loss function can be characterized as
L M S S S I M ( P ) = 1 M S S S I M ( p ~ )
  • Step 2. Calculate L l 1
Both the L1 and L2 loss function can keep the brightness and color unchanged, but the L2 loss function is more sensitive to abnormal points; thus, the effect of the model will be greatly affected the data quality.
The mechanism of the L1 loss function on the image is based on comparing the differences per pixel and then taking the absolute value. In general, it minimizes the sum of the absolute difference between the target value and the estimated value. The expression is shown below, where y ( i , j ) represents the pixel value of the original high-resolution image at position ( i , j ) ; y ( i , j ) represents the pixel value of the network output image at position ( i , j ) ; W and H represent the width and height of the image, respectively [20].
L l 1 = 1 W H i = 0 M j = 0 N | y ( i , j ) y ( i , j ) |
L l 2 = 1 W H i = 0 M j = 0 N ( y ( i , j ) y ( i , j ) ) 2
Compared with the L1 loss function, the L2 loss function magnifies the gap between the maximum error and the minimum error, and the L2 loss function is more sensitive to abnormal points. When the quality of the training data is poor, the L2 loss function will affect the model effect, resulting in inaccurate super-resolution effects, which will have a serious impact on medical diagnosis based on MR images. However, if only the L1 loss function is used, it is easy to fall into a local optimal solution.
  • Step 3. Setting of α
The initial value of α in the formula is set to 0.84. In this section, half attenuation was carried out in the 90th epoch, the 120th epoch and the 190th epoch, respectively, which can be conducive to the rapid convergence of the network.

2.2. ESRFBN Network Improvement Based on Contextual Network Fusion

The ESRFBN network improvement based on contextual network fusion is shown in Figure 2. For the input low-resolution MR images, it is divided into two channels and two input network architectures, respectively. For the first channel, the images are computed with the calculation of 3 × 3 convolution, 1 × 1 convolution and deconvolution and the up-sampling operation. Through these computer units, the high-resolution image is created. In the second channel of images, a high-resolution image is obtained through the second network computing. Then, the high-resolution images are created with the convolution-based fusion network from the two channels of MR images from the super-resolution networks. Two independent networks are used to predict high-resolution images, respectively, and the fusion layer is used to fuse the high-resolution images outputted by the independent network, then output the final super-resolution image. The expression is as follows [20].
y i = F { S 1 , S 2 , , S M } ( x i ) = j = 1 M W j S j ( x i ) + b j
Among them, W j , b j is the fusion layer parameter that being constructed. The weight of the fusion layer can be learned by fine-tuning the entire network. In this section, two independent network parameters are frozen to learn the parameters of the fusion layer. A 3 × 3 convolution layer is used as the fusion layer, and its learnable parameter is 3 × (3 × 3 + 1) = 30. In the training of the improved ESRFBN network method based on contextual network fusion, the weight of a single network is frozen, and the fusion layer is randomly initialized by a zero-mean Gaussian distribution with a standard deviation of 0.001.

2.3. Algorithm Flow

In this section, the steps of the ESRFBN super-resolution network algorithm for MR images based on multiple optimizations are mainly divided into network training and model testing. The specific steps are in the Appendix A.

2.4. Discussion

Firstly, under the large number of the training samples, the super-resolution performance is better if the super-resolution network is deeper. Accordingly, the number of the parameters of the deep super-resolution network is larger. Secondly, under the similar number of parameters and training samples, the different structures of the deep network have the different ability of super-resolution. The different performances depend on the different super-resolution mechanism. So, we have the fusion of the different structure of networks, and the fusion network can extract enough of the features of the MR images. Then, the higher super-resolution performance is obtained with the fusion super-resolution deep network. In this fusion network, we fuse the independent deep super-resolution networks, so the size of networks is not decreased. So, the numbers of parameters are similar for the original different super-resolution networks and fusion network. On the balance of the performance and the complexity, on the same size of the deep networks, the fusion network can achieve the higher performance than the original scale of deep network. For multiple independent deep super-resolution networks, the output of a single network is integrated through an additional fusion layer, which broadens the width of the network, and can effectively improve the mapping and characterization capabilities of high- and low-resolution features.

3. Experimental Results and Analysis

3.1. Experiment 1

In this section, the loss function experiments of L1, L2, SSIM, MS-SSIM and L1+MS-SSIM are performed on the double cubic down-sampling data set and K-space truncated data set. The specific experimental results are shown in Table 1, Table 2, Table 3 and Table 4. When ESRFBN uses cascading loss function L1+MS-SSIM as a loss function for training, the effect is optimal on both the bicubic decrement sampling data set and K-space truncated data set, which indicates the effectiveness of the optimization of the cascading loss function.
SSIM, as a loss function, is not particularly sensitive to uniform deviations, which will cause changes in brightness or color, and usually become duller. However, SSIM can preserve contrast and is better than other loss functions in the high-frequency region. On the other hand, the L1 loss function can keep the brightness and color unchanged. Therefore, the cascaded loss function can combine the characteristics of the two well, so that the super-resolution effect of the trained network is better.
As can be seen from Figure 3, compared with SSIM and MS-SSIM loss functions, the result of L1 and L2 only improves the pixels around the contour details, while for the result of SSIM and MS-SSIM loss functions, the feature details of high-resolution images after super-resolution are more prominent than that of L1 and L2. However, the L1+MS-SSIM loss function retains both pixel gray level information and contour feature information, and the image after super-resolution is most similar to the original image. It also indicates that the ESRFBN super-resolution network based on the L1+MS-SSIM cascading loss function optimization can effectively improve the super-resolution effect of MR images.

3.2. Experiment 2: Analysis of the Convergence of Each Loss Function Index during the Training Process

In order to enable the network to converge better during the training process and avoid falling into the local optimal solution, the cascaded loss function L1+MS-SSIM uses parameter α to balance the PSNR index and the SSIM index. As can be seen from the training convergence curve of the ESRFBN network in Figure 4, it began to converge before 80 epochs. Uncertain factors in the calculation of loss function were increased by adjusting the attenuation of the learning rate and parameter α , so as to avoid falling into the local optimal solution. The attenuation of parameter α is shown in Figure 4, which is attenuated by half at the 90th epoch, 120th epoch and 190th epoch, respectively.
In this section, in order to reflect the impact of parameter α ’s attenuation on the training process better, the difference between the evaluation indexes of the cascaded loss function (Mix) and other loss functions during the training process is used for characterization. Figure 5 shows the curve of the number of iterations and the difference in PSNR during the training of each loss function, and Figure 6 shows the curve of the number of iterations and the difference in the SSIM during the training of each loss function. It can be seen from the figure that at 90 epoch and 120 epoch, the difference between the Mix loss function and other loss functions will have obvious vibrations. This is because in the Mix loss function, the parameter α attenuated and the loss function curve oscillates, resulting in lower PSNR and SSIM values calculated by the model. However, the difference tends to stabilize after rising in the subsequent process. This shows that ESRFBN has a strong convergence ability. On the other hand, it shows that during the training process, the learning rate attenuation strategy can prevent the network from falling into the local optimal solution, so that the network model obtained finally has a better super-resolution effect.
In order to improve the PSRN and SSIM indexes of MR super-resolution, this section proposes a method based on PSNR-SSIM joint loss function cascade optimization. In order to verify the effectiveness of the proposed method, four experiments were carried out on the data sets generated by the two modes.
Experiment 1 proves that the L1+MS_SSIM cascaded loss function can obtain the optimal super-resolution effect, which proves the effectiveness of the optimization of the cascading loss function. This is mainly because the cascading loss function combines the advantages of the L1 loss function which can keep the brightness and color unchanged and the MS_SSIM loss function which can retain the contrast and is better than other loss functions in the high frequency area, so that the cascading loss function has the best characteristics of the two loss functions. The cascade loss function optimization enables the ESRFBN network to obtain a better super-resolution effect of MR images.
Through Experiment 2, it can be proved that parameter attenuation can make the difference between the Mix loss function and other loss functions vibrate significantly, so as to avoid the network falling into the local optimal solution and make the obtained network model have better super-resolution effect. At the same time, it shows that the ESRFBN network has strong convergence ability, which can make deep network converge quickly in the training process and reduce the learning difficulty of network.
The data set used in this section includes MR images’ data of the neck, head, breast and knee. The number of training set, verification set and test set are: Neck: 400, 50, 50; Head: 500, 50, 50; breast: 1800, 200, 200; knee: 800, 100, 100. Data sets with two super-resolution scales of ×2 and ×4 are generated. The model used is implemented with Pytorch 1.6.0, and is trained with dual Nvidia GeForce GTX 1080 Ti graphics cards.

3.3. Performance Evaluation

In this section, the proposed multi-optimization ESRFBN super-resolution network algorithm for MR images is compared with the super-resolution algorithm based on interpolation, reconstruction, dictionary learning and deep learning. The algorithms include BIC [15], IBP [16], POCS [5], MAP [6], JDL [7], IJDL [17], RDN [11], EDSR [18], SRGAN [12], ESRGAN [19], SRFBN [14], ESRFBN [20].
The methods mentioned in this section and the experimental results of each method are shown in Table 5 and Table 6. It can be seen from the table that the method proposed in this paper has obtained the best results in four parts of the human body and in two super-division scales, and the evaluation index value improved greatly. Compared with the suboptimal method under the scale ×2, the PSNR value is increased by 4.93, 5.90, 5.42 and 3.44, respectively; the SSIM value is increased by 0.0019, 0.0191, 0.0105 and 0.0206, respectively. Under the scale ×4, the PSNR values were increased by 5.94, 6.07, 5.65 and 4.68; the SSIM values were increased by 0.0216, 0.0422, 0.0564 and 0.0418. From the improvement of the index value, it can be seen that the MR image ESRFBN super-resolution network based on multiple optimizations proposed in this section is more suitable for MR image super-resolution tasks, and can effectively solve the limitations of deep learning networks in the application of MR images’ super-resolution tasks. The experimental results prove the effectiveness of the method proposed in this section.

4. Conclusions

This paper aims at solving the three limitations of applying a deep learning network in the super-resolution task of MR images, such as not fully considering the characteristics of MR images’ sequence, difficulty in obtaining the optimal solution of loss function and difficulty in balancing model complexity and training difficulty. Firstly, seven kinds of deep learning networks with a good super-resolution effect were used to carry out experiments under two super-resolution scales on four human body parts, namely neck, breast, knee and head. Secondly, in view of the three limitations of applying deep learning in the super-resolution of MR images, this paper proposes a multiple optimization-based ESRFBN super-resolution network algorithm for MR images, which integrates the three optimization methods. The quantitative experimental results show that the ESRFBN network is more suitable for MR images’ super-resolution task. Secondly, in view of the three limitations of applying deep learning in the super-resolution of MR images, this paper proposes a multiple optimization-based ESRFBN super-resolution network algorithm for MR images, which integrates the three optimization methods. Quantitative experimental results show that the ESRFBN super-resolution network based on multiple optimization of MR images is superior to the super-resolution algorithms based on interpolation, reconstruction, dictionary learning and other super-resolution algorithms based on a deep learning network, and it can greatly improve the super-resolution effect of MR images, proving that the method proposed in this paper is effective.

Author Contributions

Conceptualization, J.L.; Formal analysis, H.L. and M.S.; Investigation, H.L., M.S. and J.-S.P.; Methodology, H.L. and J.L.; Validation, J.-S.P.; Writing—original draft, H.L., M.S. and J.L.; Writing—review & editing, J.-S.P. All authors have read and agreed to the published version of the manuscript.

Funding

This work is supported by the National Science Foundation of China under Grant No. 61671170 and No.61872085, Science and Technology Foundation of National Defense Key Laboratory of Science and Technology on Parallel and Distributed Processing Laboratory (PDL) under Grant No. 6142110180406.

Institutional Review Board Statement

Ethical review and approval were not applicable for this study, due to the data used in the experiment is open source.

Data Availability Statement

We have not used specific data from other sources for the simulations of the results. The two popular MRI data sets in this paper, fastMRI Dataset and IXI Dataset, are free to download with the website: https://fastmri.org/ and http://www.brain-development.org/ (accessed on 31 November 2020).

Conflicts of Interest

The authors declare no conflict of interest.

Appendix A

Algorithm A1 Multiple optimization ESRFBN super-resolution network training process.
Algorithm A1 ESRFBN super-resolution network training based on multiple optimization.
Input: Multi-slice low-resolution image set X : ( x t 1 , x t , x t + 1 ) m , Multi-slice high-resolution image set Y : ( y t 1 , y t , y t + 1 ) m
Output: ESRFBN network model parameters θ 1 , EDSR network model parameters θ 2
1: Hyper-parameter settings: Batch: batch_size, Number of iterations: epoch
2: Pre-trained model loading
3.1: Independent network model training
  for 1 to epoch do
   for 1 to m/batch_size do
    for 1 to batch_size do
    (1) ESRFBN network forward calculation
     Y 1 = f ESRFBN ( X ; θ 1 ) , where Y 1 represents the output result of ESRFBN network
    Error calculation
     L ( θ 1 ) = L M i x ( Y Y 1 ) , where L Mix = α L MS - SSIM + ( 1 α ) G σ G M L 1 , α = 0.84
     Parameter optimization
      θ 1 = A d a m ( L ( θ 1 ) )
    (2) EDSR network forward calculation
     Y 2 = f EDSR ( X ; θ 2 ) , where Y represents the output result of EDSR network
    Error calculation L ( θ 2 ) = L M i x ( Y Y 2 ) , where L Mix = α L MS - SSIM + ( 1 α ) G σ G M L 1 , α = 0.84
    Parameter optimization
      θ 2 = A d a m ( L ( θ 2 ) )
    end for
   end for
  end for
3.2: Fusion layer parameter learning
  (1) The parameters of the fusion layer are randomly initialized by a zero-mean Gaussian distribution with a standard deviation of 0.001.
  (2) Independent network outputs their respective super-resolution results
     Y 1 = f ESRFBN ( X ; θ 1 )
     Y 2 = f EDSR ( X ; θ 2 )
  (3) The fusion layer performs the fusion of the output results of each network
     Y 3 = f F u s i o n ( Y 1 , Y 2 ; θ 3 ) , where θ 3 represents the parameters of the fusion layer, a total of 30 parameters
  (4) Error calculation
     L ( θ 3 ) = L M i x ( Y Y 3 ) , where L Mix = α L MS - SSIM + ( 1 α ) G σ G M L 1 , α = 0.84
  (5) Update of fusion layer parameters
     θ 3 = A d a m ( L ( θ 3 ) )
4: Model parameter output θ 1 , θ 2 , θ 3
Algorithm A2 Multiple optimization ESRFBN super-resolution network test process.
Algorithm A2 ESRFBN super-resolution network test based on multiple optimizations.
Input: Multi-slice low-resolution image set X : ( x t 1 , x t , x t + 1 ) k
   Multi-slice high-resolution image set Y : ( y t 1 , y t , y t + 1 ) k
   Deep model parameters θ 1 , θ 2 , θ 3
Output: Evaluation value PSNR, SSIM.
1: Model parameter loading
2: Image super-resolution
   for 1 to k do
   (1) Independent network outputs their respective super-resolution results
     Y 1 i = f ESRFBN ( X ; θ 1 )
     Y 2 i = f EDSR ( X ; θ 2 )
   (2) The fusion layer performs the fusion of the output results of each network
     Y 3 i = f F u s i o n ( Y 1 i , Y 2 i ; θ 3 )
   end for
3: PSNR, SSIM calculation
   PSNR a v g = 1 k i = 1 k p s n r ( Y 3 i , Y i )
   SSIM a v g = 1 k i = 1 k s s i m ( Y 3 i , Y i )
4: Output the value of PSNR, SSIM

References

  1. El Hakimi, W.; Wesarg, S. Accurate super-resolution reconstruction for CT and MR images. In Proceedings of the 26th IEEE International Symposium on Computer-Based Medical Systems, Porto, Portugal, 20–22 June 2013; pp. 445–448. [Google Scholar]
  2. Zhao, C.; Lu, J.; Li, K.; Wang, X.; Wang, H.; Zhang, M. Optimization of scanning time for clinical application of resting fMRI. Chin. J. Mod. Neurol. Dis. 2011, 3, 51–55. [Google Scholar]
  3. Kaur, P.; Sao, A.K.; Ahuja, C.K. Super Resolution of Magnetic Resonance Images. J. Imaging 2021, 7, 101. [Google Scholar] [CrossRef]
  4. Ramasubramanian, V.; Paliwal, K.K. An efficient approximation-elimination algorithm for fast nearest-neighbour search based on a spherical distance coordinate formulation. Pattern Recognit. Lett. 1992, 13, 471–480. [Google Scholar] [CrossRef]
  5. Stark, H.; Oskoui, P. High-resolution image recovery from image-plane arrays, using convex projections. J. Opt. Soc. Am. A Opt. Image Sci. 1989, 6, 1715. [Google Scholar] [CrossRef] [PubMed]
  6. Schultz, R.R.; Stevenson, R.L. Improved definition image expansion. In Proceedings of the IEEE International Conference on Acoustics, San Francisco, CA, USA, 23–26 March 1992. [Google Scholar]
  7. Yang, J.; Wright, J.; Huang, T.S.; Ma, Y. Image Super-Resolution Via Sparse Representation. IEEE Trans. Image Process. 2010, 19, 2861–2873. [Google Scholar] [CrossRef] [PubMed]
  8. Dong, C.; Loy, C.C.; He, K.; Tang, X. Image Super-Resolution Using Deep Convolutional Networks. IEEE Trans. Pattern Anal. Mach. Intell. 2016, 38, 295–307. [Google Scholar] [CrossRef] [PubMed] [Green Version]
  9. Wang, Z.; Chen, J.; Hoi, S. Deep Learning for Image Super-resolution: A Survey. IEEE Trans. Pattern Anal. Mach. Intell. 2020, 99, 1. [Google Scholar] [CrossRef] [PubMed] [Green Version]
  10. Kim, J.; Lee, J.K.; Lee, K.M. Accurate Image Super-Resolution Using Very Deep Convolutional Networks. In Proceedings of the IEEE Conference on Computer Vision & Pattern Recognition, Las Vegas, NV, USA, 27–30 June 2016; pp. 1646–1654. [Google Scholar]
  11. Zhang, Y.; Tian, Y.; Kong, Y.; Zhong, B.; Fu, Y. Residual Dense Network for Image Super-Resolution. In Proceedings of the 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, Salt Lake City, UT, USA, 18–23 June 2018; pp. 2472–2481. [Google Scholar]
  12. Ledig, C.; Theis, L.; Huszar, F.; Caballero, J.; Cunningham, A.; Acosta, A.; Aitken, A.; Tejani, A.; Totz, J.; Wang, Z.; et al. Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network. In Proceedings of the 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Honolulu, HI, USA, 21–26 July 2017; pp. 5892–5900. [Google Scholar]
  13. Haris, M.; Shakhnarovich, G.; Ukita, N. Deep Back-Projection Networks for Super-Resolution. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Salt Lake City, UT, USA, 18–22 June 2018. [Google Scholar]
  14. Li, Z.; Yang, J.; Liu, Z.; Yang, X.; Jeon, G.; Wu, W. Feedback Network for Image Super-Resolution. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Long Beach, CA, USA, 16–20 June 2019. [Google Scholar]
  15. Keys, R. Cubic convolution interpolation for digital image processing. IEEE Trans. Acoust. Speech Signal Process. 1981, 29, 1153–1160. [Google Scholar] [CrossRef] [Green Version]
  16. Irani, M.; Peleg, S. Improving resolution by image registration. CVGIP Graph. Model. Image Process. 1991, 53, 231–239. [Google Scholar] [CrossRef]
  17. Li, J.B.; Liu, H.; Pan, J.S.; Yao, H. Training samples-optimizing based dictionary learning algorithm for MR sparse superresolution reconstruction. Biomed. Signal Process. Control 2018, 39, 177–184. [Google Scholar] [CrossRef]
  18. Lim, B.; Son, S.; Kim, H.; Nah, S.; Lee, K.M. Enhanced Deep Residual Networks for Single Image Super-Resolution. In Proceedings of the 2017 IEEE Conference on Computer Vision and Pattern Recognition Workshops (CVPRW), Honolulu, HI, USA, 21–26 July 2017. [Google Scholar]
  19. Wang, X.; Yu, K.; Wu, S.; Gu, J.; Liu, Y.; Dong, C.; Qiao, Y.; Loy, C.C. ESRGAN: Enhanced Super-Resolution Generative Adversarial Networks. In Proceedings of the European Conference on Computer Vision (ECCV) Workshops, Munich, Germany, 8–14 September 2018. [Google Scholar]
  20. Liu, H.; Liu, J.; Li, J.; Pan, J.S.; Yu, X. PSR: Unified Framework of Parameter-Learning-Based MR Image Superresolution. J. Healthc. Eng. 2021, 2021, 5591660. [Google Scholar] [PubMed]
  21. Xue, X.; Wang, Y.; Li, J.; Jiao, Z.; Ren, Z.; Gao, X. Progressive Sub-Band Residual-Learning Network for MR Image Super Resolution. IEEE J. Biomed. Health Inform. 2019, 24, 377–386. [Google Scholar] [CrossRef] [PubMed]
  22. Zheng, Y.; Zhen, B.; Chen, A.; Qi, F.; Hao, X.; Qiu, B. A hybrid convolutional neural network for super-resolution reconstruction of MR images. Med. Phys. 2020, 47, 3013–3022. [Google Scholar] [CrossRef] [PubMed]
  23. Tan, C.; Zhu, J. Arbitrary Scale Super-Resolution for Brain MRI Images. In Proceedings of the IFIP International Conference on Artificial Intelligence Applications and Innovations, Neos Marmaras, Greece, 5–7 June 2020. [Google Scholar]
  24. Hu, X.; Mu, H.; Zhang, X.; Wang, Z.; Tan, T.; Sun, J. Meta-SR: A magnification-arbitrary network for super-resolution. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Long Beach, CA, USA, 16–20 June 2019; pp. 1575–1584. [Google Scholar]
  25. Xu, J.; Gong, E.; Pauly, J.; Zaharchuk, G. 200x Low-dose PET Reconstruction using Deep Learning. arXiv 2017, arXiv:1712.04119. [Google Scholar]
  26. Zhao, H.; Gallo, O.; Frosio, I.; Kautz, J. Loss Functions for Image Restoration with Neural Networks. IEEE Trans. Comput. Imaging 2016, 3, 47–57. [Google Scholar] [CrossRef]
  27. Ren, H.; El-Khamy, M.; Lee, J. Image Super Resolution Based on Fusing Multiple Convolution Neural Networks. In Proceedings of the Computer Vision & Pattern Recognition Workshops, Honolulu, HI, USA, 21–26 July 2017. [Google Scholar]
Figure 1. The architecture of deep super-resolution algorithm based on multiple optimizations.
Figure 1. The architecture of deep super-resolution algorithm based on multiple optimizations.
Applsci 11 08150 g001
Figure 2. The improved ESRFBN network method based on contextual network fusion.
Figure 2. The improved ESRFBN network method based on contextual network fusion.
Applsci 11 08150 g002
Figure 3. Knee: the super-resolution graph of each loss function of the ESRFBN network at scale 4.
Figure 3. Knee: the super-resolution graph of each loss function of the ESRFBN network at scale 4.
Applsci 11 08150 g003
Figure 4. Attenuation of parameter α .
Figure 4. Attenuation of parameter α .
Applsci 11 08150 g004
Figure 5. The scale of the knee part is 4: The PSNR difference between the training process of the Mix loss function and other loss functions.
Figure 5. The scale of the knee part is 4: The PSNR difference between the training process of the Mix loss function and other loss functions.
Applsci 11 08150 g005
Figure 6. The scale of the knee is 4: The difference between the training process of the Mix loss function and other loss functions SSIM.
Figure 6. The scale of the knee is 4: The difference between the training process of the Mix loss function and other loss functions SSIM.
Applsci 11 08150 g006
Table 1. PSNR values of different loss functions (bicubic downsampling).
Table 1. PSNR values of different loss functions (bicubic downsampling).
DataScaleBicubicL1L2SSIMMS-SSIML1+MS-SSIM
neck×225.4437.8940.4140.2440.1540.47
×421.5830.2830.9030.5030.6730.92
breast×228.5635.1341.4642.9042.8342.97
×420.5030.7832.8832.6032.9132.93
knee×230.6038.3940.1240.0440.1240.20
×422.6931.4232.6332.2932.5132.64
head×220.3239.4238.5538.1938.2839.39
×418.6034.0432.9732.2031.9534.11
Note: The numbers in bold are the best results.
Table 2. SSIM values of different loss functions (double cubic downsampling).
Table 2. SSIM values of different loss functions (double cubic downsampling).
DataScaleBicubicL1L2SSIMMS-SSIML1+MS-SSIM
neck×20.69280.98760.99000.99000.98980.9897
×40.54420.93080.93490.93670.93580.9392
breast×20.88480.94920.97630.98020.97970.9816
×40.48300.89790.90310.90700.90750.9082
knee×20.91850.97600.96330.96410.96370.9774
×40.56490.89870.88440.88810.88670.8873
head×20.60250.96050.95920.95970.95930.9609
×40.56230.91280.90640.90640.90110.9609
Table 3. PSNR values of different loss functions (K-space truncation downsampling).
Table 3. PSNR values of different loss functions (K-space truncation downsampling).
DataScaleBicubicL1L2SSIMMS-SSIML1+MS-SSIM
neck×215.8024.1624.3223.9824.2424.36
×49.2918.5419.7618.8519.2619.77
breast×213.5231.7531.8231.7831.9631.98
×47.6825.6225.9425.1525.9825.98
knee×211.8331.4931.5230.8131.0931.73
×45.9726.0926.3124.8625.4826.49
head×26.2927.8929.8726.9827.9728.76
×42.9520.7520.4320.9718.5320.98
Note: The numbers in bold are the best results.
Table 4. SSIM values of different loss functions (K-space truncation downsampling).
Table 4. SSIM values of different loss functions (K-space truncation downsampling).
DataScaleBicubicL1L2SSIMMS-SSIML1+MS-SSIM
neck×20.65000.89230.89790.89970.89800.8998
×40.33990.82010.82550.82920.82680.8315
breast×20.49670.91230.92110.92060.91890.9213
×40.04870.80270.80780.81110.81780.8185
knee×20.36470.90080.90720.90730.90540.9078
×40.09190.78400.79080.78580.78640.7864
head×20.31260.80650.82110.89970.80750.9609
×40.20810.64340.63360.65030.65080.6539
Note: The numbers in bold are the best results.
Table 5. PSNR values of MR images ESRFBN super-resolution network based on multiple optimization.
Table 5. PSNR values of MR images ESRFBN super-resolution network based on multiple optimization.
DataScaleBICIBPPOCSMAPJDLIJDLRDNEDSR DBPNSRGANESRGANSRFBNESRFBNProposed
neck×225.4429.3528.8630.0135.9536.1229.0537.3435.7437.6936.1638.0337.8942.96
×421.5824.7224.2825.2728.5328.8224.5329.2529.2828.7829.1130.2330.2836.22
breast×228.5630.0629.7631.2133.3434.0833.5034.5731.9434.0632.9134.2835.1341.03
×420.5021.5622.3823.0927.8729.1623.6830.0028.1327.9926.3130.6830.7836.85
knee×230.6032.8033.0134.3635.9436.8736.3437.4937.6736.9336.0037.9538.3943.81
×422.6924.7425.2526.6628.4830.1130.0228.3431.2929.6328.7531.3531.4237.07
head×220.3228.2228.0431.2433.4335.1138.0538.0333.8235.8035.8939.1939.4242.86
×418.6020.3522.2924.6828.6528.6029.7032.2229.3930.9531.3233.6034.0438.72
Note: The numbers in bold are the best results.
Table 6. SSIM values of MR images ESRFBN super-resolution network based on multiple optimization.
Table 6. SSIM values of MR images ESRFBN super-resolution network based on multiple optimization.
DataScaleBICIBPPOCSMAPJDLIJDLRDNEDSR DBPNSRGANESRGANSRFBNESRFBNProposed
neck×20.69280.80240.82570.84580.91550.93860.88240.98610.98000.98680.98210.98790.98800.9899
×40.54420.75890.80160.82740.90010.93180.71440.91640.91820.90930.91580.93040.93080.9524
breast×20.88480.89430.90100.90460.92000.93080.93730.94470.92430.93320.92560.94280.94920.9683
×40.48300.75920.77590.79000.85940.87470.61180.86310.83510.81370.79000.87350.89790.9401
knee×20.91850.92310.92100.93420.95910.96600.96410.97190.97270.96700.96340.97440.97600.9865
×40.56490.64910.65070.68750.84760.85220.85970.89680.89400.86430.90770.89640.90540.9641
head×20.60250.85220.84830.84270.90180.92740.95650.95650.90620.92340.92410.96000.96050.9811
×40.56230.72950.75350.76240.83580.86010.79620.88860.79060.85000.86110.90890.91280.9546
Note: The numbers in bold are the best results.
Publisher’s Note: MDPI stays neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Share and Cite

MDPI and ACS Style

Liu, H.; Shao, M.; Pan, J.-S.; Li, J. Multiple Optimizations-Based ESRFBN Super-Resolution Network Algorithm for MR Images. Appl. Sci. 2021, 11, 8150. https://doi.org/10.3390/app11178150

AMA Style

Liu H, Shao M, Pan J-S, Li J. Multiple Optimizations-Based ESRFBN Super-Resolution Network Algorithm for MR Images. Applied Sciences. 2021; 11(17):8150. https://doi.org/10.3390/app11178150

Chicago/Turabian Style

Liu, Huanyu, Mingmei Shao, Jeng-Shyang Pan, and Junbao Li. 2021. "Multiple Optimizations-Based ESRFBN Super-Resolution Network Algorithm for MR Images" Applied Sciences 11, no. 17: 8150. https://doi.org/10.3390/app11178150

Note that from the first issue of 2016, this journal uses article numbers instead of page numbers. See further details here.

Article Metrics

Back to TopTop