RIRGAN: An end-to-end lightweight multi-task learning method for brain MRI super-resolution and denoising

被引:11
作者
Yu, Miao [1 ]
Guo, Miaomiao [1 ]
Zhang, Shuai [1 ]
Zhan, Yuefu [2 ]
Zhao, Mingkang [1 ]
Lukasiewicz, Thomas [3 ,4 ]
Xu, Zhenghua [1 ]
机构
[1] Hebei Univ Technol, Sch Hlth Sci & Biomed Engn, State Key Lab Reliabil & Intelligence Elect Equipm, Tianjin, Peoples R China
[2] Hainan Women & Childrens Med Ctr, Dept Radiol, Haikou, Peoples R China
[3] Vienna Univ Technol, Inst Log & Computat, Vienna, Austria
[4] Univ Oxford, Dept Comp Sci, Oxford, England
基金
中国国家自然科学基金;
关键词
Multi-task learning; Super-resolution and denoising; Medical image analysis; Generative adversarial network; RESOLUTION ENHANCEMENT; IMAGE; RECONSTRUCTION; NETWORK; MODEL;
D O I
10.1016/j.compbiomed.2023.107632
中图分类号
Q [生物科学];
学科分类号
07 ; 0710 ; 09 ;
摘要
A common problem in the field of deep-learning-based low-level vision medical images is that most of the research is based on single task learning (STL), which is dedicated to solving one of the situations of low resolution or high noise. Our motivation is to design a model that can perform both SR and DN tasks simultaneously, in order to cope with the actual situation of low resolution and high noise in low-level vision medical images. By improving the existing single image super-resolution (SISR) network and introducing the idea of multi-task learning (MTL), we propose an end-to-end lightweight MTL generative adversarial network (GAN) based network using residual-in-residual-blocks (RIR-Blocks) for feature extraction, RIRGAN, which can concurrently accomplish super-resolution (SR) and denoising (DN) tasks. The generator in RIRGAN is composed of several residual groups with a long skip connection (LSC), which can help form a very deep network and enable the network to focus on learning high-frequency (HF) information. The introduction of a discriminator based on relativistic average discriminator (RaD) greatly improves the discriminator's ability and makes the generated image have more realistic details. Meanwhile, the use of hybrid loss function not only ensures that RIRGAN has the ability of MTL, but also enables RIRGAN to give a more balanced attention between quantitative evaluation of metrics and qualitative evaluation of human vision. The experimental results show that the quality of the restoration image of RIRGAN is superior to the SR and DN methods based on STL in both subjective perception and objective evaluation metrics when processing medical images with low-level vision. Our RIRGAN is more in line with the practical requirements of medical practice.
引用
收藏
页数:17
相关论文
共 59 条
  • [1] Pre-Trained Image Processing Transformer
    Chen, Hanting
    Wang, Yunhe
    Guo, Tianyu
    Xu, Chang
    Deng, Yiping
    Liu, Zhenhua
    Ma, Siwei
    Xu, Chunjing
    Xu, Chao
    Gao, Wen
    [J]. 2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 12294 - 12305
  • [2] Chen Y, 2023, Multimedia Tools Appl., P1
  • [3] DARGS: Image inpainting algorithm via deep attention residuals group and semantics
    Chen, Yuantao
    Xia, Runlong
    Yang, Kai
    Zou, Ke
    [J]. JOURNAL OF KING SAUD UNIVERSITY-COMPUTER AND INFORMATION SCIENCES, 2023, 35 (06)
  • [4] RNON: image inpainting via repair network and optimization network
    Chen, Yuantao
    Xia, Runlong
    Zou, Ke
    Yang, Kai
    [J]. INTERNATIONAL JOURNAL OF MACHINE LEARNING AND CYBERNETICS, 2023, 14 (09) : 2945 - 2961
  • [5] FFTI: Image inpainting algorithm via features fusion and two-steps inpainting
    Chen, Yuantao
    Xia, Runlong
    Zou, Ke
    Yang, Kai
    [J]. JOURNAL OF VISUAL COMMUNICATION AND IMAGE REPRESENTATION, 2023, 91
  • [6] MFFN: image super-resolution via multi-level features fusion network
    Chen, Yuantao
    Xia, Runlong
    Yang, Kai
    Zou, Ke
    [J]. VISUAL COMPUTER, 2024, 40 (02) : 489 - 504
  • [7] Destruction and Construction Learning for Fine-grained Image Recognition
    Chen, Yue
    Bai, Yalong
    Zhang, Wei
    Mei, Tao
    [J]. 2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, : 5152 - 5161
  • [8] Chen YH, 2020, Arxiv, DOI [arXiv:2003.01217, DOI 10.48550/ARXIV.2003.01217]
  • [9] Da Wang Y, 2019, Arxiv, DOI arXiv:1907.07131
  • [10] Image Super-Resolution Using Deep Convolutional Networks
    Dong, Chao
    Loy, Chen Change
    He, Kaiming
    Tang, Xiaoou
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2016, 38 (02) : 295 - 307