A general image fusion framework using multi-task semi-supervised learning

被引:22
作者
Wang, Wu [1 ]
Deng, Liang-Jian [1 ]
Vivone, Gemine [2 ,3 ]
机构
[1] Univ Elect Sci & Technol China, Sch Math Sci, Chengdu 611731, Sichuan, Peoples R China
[2] Natl Council Res, Inst Methodol Environm Anal, I-85050 Tito, Italy
[3] NBFC Natl Biodivers Future Ctr, I-90133 Palermo, Italy
基金
中国国家自然科学基金;
关键词
Image fusion; Multi-task; Semi-supervised learning; Laplacian pyramid; Fusion rule; Remote sensing; Medical images; ADVERSARIAL NETWORK; QUALITY ASSESSMENT; ALGORITHM; GAN; MODEL; NEST; MFF;
D O I
10.1016/j.inffus.2024.102414
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Existing image fusion methods primarily focus on solving single -task fusion problems, overlooking the potential information complementarity among multiple fusion tasks. Additionally, there has been no prior research in the field of image fusion that explores the mixed training of labeled and unlabeled data for different fusion tasks. To address these gaps, this paper introduces a novel multi -task semi-supervised learning approach to construct a general image fusion framework. This framework not only facilitates collaborative training for multiple fusion tasks, thereby achieving effective information complementarity among datasets from different fusion tasks, but also promotes the (unsupervised) learning of unlabeled data via the (supervised) learning of labeled data. Regarding the specific network module, we propose a so-called pseudo-siamese Laplacian pyramid transformer (PSLPT), which can effectively distinguish information at different frequencies in source images and discriminatively fuse features from distinct frequencies. More specifically, we take datasets of four typical image fusion tasks into the same PSLPT for weight updates, yielding the final general fusion model. Extensive experiments demonstrate that the obtained general fusion model exhibits promising outcomes for all four image fusion tasks, both visually and quantitatively. Moreover, comprehensive ablation and discussion experiments corroborate the effectiveness of the proposed method. The code is available at https: //github.com/wwhappylife/A-general-image-fusion-framework-using-multi-task-semi-supervised-learning.
引用
收藏
页数:16
相关论文
共 72 条
[31]  
Liu J., 2023, IEEE Trans. Neural Netw. Learn. Syst.
[32]   A Bilevel Integrated Model With Data-Driven Layer Ensemble for Multi-Modality Image Fusion [J].
Liu, Risheng ;
Liu, Jinyuan ;
Jiang, Zhiying ;
Fan, Xin ;
Luo, Zhongxuan .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2021, 30 :1261-1274
[33]   Multi-focus image fusion with a deep convolutional neural network [J].
Liu, Yu ;
Chen, Xun ;
Peng, Hu ;
Wang, Zengfu .
INFORMATION FUSION, 2017, 36 :191-207
[34]   Swin Transformer: Hierarchical Vision Transformer using Shifted Windows [J].
Liu, Ze ;
Lin, Yutong ;
Cao, Yue ;
Hu, Han ;
Wei, Yixuan ;
Zhang, Zheng ;
Lin, Stephen ;
Guo, Baining .
2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, :9992-10002
[35]   Multi-Modal Image Fusion via Deep Laplacian Pyramid Hybrid Network [J].
Luo, Xing ;
Fu, Guizhong ;
Yang, Jiangxin ;
Cao, Yanlong ;
Cao, Yanpeng .
IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY, 2023, 33 (12) :7354-7369
[36]   SESF-Fuse: an unsupervised deep model for multi-focus image fusion [J].
Ma, Boyuan ;
Zhu, Yu ;
Yin, Xiang ;
Ban, Xiaojuan ;
Huang, Haiyou ;
Mukeshimana, Michele .
NEURAL COMPUTING & APPLICATIONS, 2021, 33 (11) :5793-5804
[37]   SwinFusion: Cross-domain Long-range Learning for General Image Fusion via Swin Transformer [J].
Ma, Jiayi ;
Tang, Linfeng ;
Fan, Fan ;
Huang, Jun ;
Mei, Xiaoguang ;
Ma, Yong .
IEEE-CAA JOURNAL OF AUTOMATICA SINICA, 2022, 9 (07) :1200-1217
[38]   Pan-GAN: An unsupervised pan -sharpening method for remote sensing image fusion [J].
Ma, Jiayi ;
Yu, Wei ;
Chen, Chen ;
Liang, Pengwei ;
Guo, Xiaojie ;
Jiang, Junjun .
INFORMATION FUSION, 2020, 62 :110-120
[39]   DDcGAN: A Dual-Discriminator Conditional Generative Adversarial Network for Multi-Resolution Image Fusion [J].
Ma, Jiayi ;
Xu, Han ;
Jiang, Junjun ;
Mei, Xiaoguang ;
Zhang, Xiao-Ping .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2020, 29 :4980-4995
[40]   Perceptual Quality Assessment for Multi-Exposure Image Fusion [J].
Ma, Kede ;
Zeng, Kai ;
Wang, Zhou .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2015, 24 (11) :3345-3356