PanFlowNet: A Flow-Based Deep Network for Pan-sharpening
Gang Yang
Xiangyong Cao
Wenzhe Xiao
Man Zhou
Aiping Liu
Xun Chen
Deyu Meng
[Paper]
[GitHub]
[Poster]
TL; DR: 1)Probabilistic flow model modeling of image inverse problem. 2)Achieving diverse feature extraction in pan-sharpening tasks.

Summary

We propose a flow-based deep network (i.e., PanFlowNet) for pan-sharpening. This network can accurately learn the conditional distribution of HRMS images given the corresponding LRMS and PAN images. To the best of our knowledge, this is the first attempt to learn an explicit distribution by employing the generative flow model for the pan-sharpening task.

The proposed PanFlowNet can generate diverse HRMSs given the LRMS and PAN images as well as the Gaussian noise sample and thus can alleviate the ill-posed issue to some extent. Besides, the generated HRMS images are diverse since each HRMS image focuses on a different detailed part of the ground truth.

We extend the vanilla flow model to a probabilistic multi-conditional flow model to adapt to the multi-conditionality of the pan-sharpening task. Extensive experiments over different satellite datasets demonstrate that our method can outperform existing state-of-the-art approaches both visually and quantitatively.

Please enjoy our results!



Abstract

Pan-sharpening aims to generate a high-resolution multispectral (HRMS) image by integrating the spectral information of a low-resolution multispectral (LRMS) image with the texture details of a high-resolution panchromatic (PAN) image. It essentially inherits the ill-posed nature of the super-resolution (SR) task that diverse HRMS images can degrade into an LRMS image. However, existing deep learning-based methods recover only one HRMS image from the LRMS image and PAN image using a deterministic mapping, thus ignoring the diversity of the HRMS image. In this paper, to alleviate this ill-posed issue, we propose a flow-based pan-sharpening network (\textbf{PanFlowNet}) to directly learn the \textbf{conditional distribution} of HRMS image given LRMS image and PAN image instead of learning a deterministic mapping. Specifically, we first transform this unknown conditional distribution into a given Gaussian distribution by an invertible network, and the conditional distribution can thus be explicitly defined. Then, we design an invertible Conditional Affine Coupling Block (CACB) and further build the architecture of PanFlowNet by stacking a series of CACBs. Finally, the PanFlowNet is trained by maximizing the log-likelihood of the conditional distribution given a training set and can then be used to predict diverse HRMS images. The experimental results verify that the proposed PanFlowNet can generate various HRMS images given an LRMS image and a PAN image. Additionally, the experimental results on different kinds of satellite datasets also demonstrate the superiority of our PanFlowNet compared with other state-of-the-art methods both visually and quantitatively. Code is available at Github.


Code

PanFlowNet

 [GitHub]


Paper and Supplementary Material

Gang Yang, Xiangyong Cao, Wenzhe Xiao, Man Zhou, Aiping Liu, Xun Chen, Deyu Meng.
PanFlowNet: A Flow-Based Deep Network for Pan-Sharpening.
In Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV), 2023, pp. 16857-16867.
(hosted on ICCV 2023)




Recommended citation

Zheyu Zhang, Gang Yang, Yueyi Zhang, Huanjing Yue, Aiping Liu, Yunwei Ou, Jian Gong, Xiaoyan Sun. TMFormer: Token Merging Transformer for Brain Tumor Segmentation with Missing Modalities[C]//Proceedings of the AAAI Conference on Artificial Intelligence. 2024, 38(7): 7414-7422.


Acknowledgements

This work was supported by the National Key R&D Program of China (2022YFA1004100), the National Natural Science Foundation of China (NSFC) under Grant 62272375, 61922075 and 82272070, the CAS Project for Young Scientists in Basic Research (Grant No. YSBR-067) and the Macao Science and Technology Development Fund under Grant 061/2020/A2. We acknowledge the support of GPU cluster built by MCC Lab of Information Science and Technology Institution, USTC.