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

被引:18
作者
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 [J].
Chen, Hanting ;
Wang, Yunhe ;
Guo, Tianyu ;
Xu, Chang ;
Deng, Yiping ;
Liu, Zhenhua ;
Ma, Siwei ;
Xu, Chunjing ;
Xu, Chao ;
Gao, Wen .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :12294-12305
[2]  
Chen Y., 2023, Multimedia Tools Appl.
[3]   DARGS: Image inpainting algorithm via deep attention residuals group and semantics [J].
Chen, Yuantao ;
Xia, Runlong ;
Yang, Kai ;
Zou, Ke .
JOURNAL OF KING SAUD UNIVERSITY-COMPUTER AND INFORMATION SCIENCES, 2023, 35 (06)
[4]   RNON: image inpainting via repair network and optimization network [J].
Chen, Yuantao ;
Xia, Runlong ;
Zou, Ke ;
Yang, Kai .
INTERNATIONAL JOURNAL OF MACHINE LEARNING AND CYBERNETICS, 2023, 14 (09) :2945-2961
[5]   FFTI: Image inpainting algorithm via features fusion and two-steps inpainting [J].
Chen, Yuantao ;
Xia, Runlong ;
Zou, Ke ;
Yang, Kai .
JOURNAL OF VISUAL COMMUNICATION AND IMAGE REPRESENTATION, 2023, 91
[6]   MFFN: image super-resolution via multi-level features fusion network [J].
Chen, Yuantao ;
Xia, Runlong ;
Yang, Kai ;
Zou, Ke .
VISUAL COMPUTER, 2024, 40 (02) :489-504
[7]   Destruction and Construction Learning for Fine-grained Image Recognition [J].
Chen, Yue ;
Bai, Yalong ;
Zhang, Wei ;
Mei, Tao .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :5152-5161
[8]  
Chen YH, 2020, Arxiv, DOI arXiv:2003.01217
[9]  
Da Wang Y, 2019, Arxiv, DOI arXiv:1907.07131
[10]   Image Super-Resolution Using Deep Convolutional Networks [J].
Dong, Chao ;
Loy, Chen Change ;
He, Kaiming ;
Tang, Xiaoou .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2016, 38 (02) :295-307