Analysis of medical images super-resolution via a wavelet pyramid recursive neural network constrained by wavelet energy entropy.

Journal: Neural networks : the official journal of the International Neural Network Society
PMID:

Abstract

Recently, multi-resolution pyramid-based techniques have emerged as the prevailing research approach for image super-resolution. However, these methods typically rely on a single mode of information transmission between levels. In our approach, a wavelet pyramid recursive neural network (WPRNN) based on wavelet energy entropy (WEE) constraint is proposed. This network transmits previous-level wavelet coefficients and additional shallow coefficient features to capture local details. Besides, the parameter of low- and high-frequency wavelet coefficients within each pyramid level and across pyramid levels is shared. A multi-resolution wavelet pyramid fusion (WPF) module is devised to facilitate information transfer across network pyramid levels. Additionally, a wavelet energy entropy loss is proposed to constrain the reconstruction of wavelet coefficients from the perspective of signal energy distribution. Finally, our method achieves the competitive reconstruction performance with the minimal parameters through an extensive series of experiments conducted on publicly available datasets, which demonstrates its practical utility.

Authors

  • Yue Yu
    Department of Mathematics, Lehigh University, Bethlehem, PA, USA.
  • Kun She
    School of Information and Software Engineering, University of Electronic Science and Technology of China, Chengdu, China.
  • Kaibo Shi
    School of Automation Engineering, University of Electronic Science and Technology of China, Chengdu, Sichuan 611731, PR China. Electronic address: skbs111@163.com.
  • Xiao Cai
    School of Information and Software Engineering, University of Electronic Science and Technology of China, Chengdu, 610054, Sichuan, China. Electronic address: caixiao327327@163.com.
  • Oh-Min Kwon
    School of Electrical Engineering, Chungbuk National University, Cheongju 28644, South Korea. Electronic address: madwind@cbnu.ac.kr.
  • YengChai Soh
    School of Electrical and Electronic Engineering, Nanyang Technological University, Singapore, 639798, Singapore. Electronic address: eycsoh@ntu.edu.sg.