Two-Stage is Enough: A Concise Deep Unfolding Reconstruction Network for Flexible Video Compressive Sensing
Computer Network Information Center, Chinese Academy of Science(中国科学院计算机网络信息中心);University of Chinese Academy of Sciences(中国科学院大学);Westlake University(西湖大学)
We consider the reconstruction problem of video compressive sensing (VCS) under the deep unfolding/rolling structure. Yet, we aim to build a flexible and concise model using minimum stages. Different from existing deep unfolding networks used for inverse problems, where more stages are used for higher performance but without flexibility to different masks and scales, hereby we show that a 2-stage deep unfolding network can lead to the state-of-the-art (SOTA) results (with a 1.7dB gain in PSNR over the single stage model, RevSCI) in VCS. The proposed method possesses the properties of adaptation to new masks and ready to scale to large data without any additional training thanks to the advantages of deep unfolding. Furthermore, we extend the proposed model for color VCS to perform joint reconstruction and demosaicing. Experimental results demonstrate that our 2-stage model has also achieved SOTA on color VCS reconstruction, leading to a >2.3dB gain in PSNR over the previous SOTA algorithm based on plug-and-play framework, meanwhile speeds up the reconstruction by >17 times. In addition, we have found that our network is also flexible to the mask modulation and scale size for color VCS reconstruction so that a single trained network can be applied to different hardware systems. The code and models will be released to the public.
Degradation-Aware Unfolding Half-Shuffle Transformer for Spectral Compressive Imaging
Tsinghua University(清华大学);Shenzhen Institute of Future Media Technology(深圳未来媒体技术研究院);Westlake University(西湖大学);ETH Zürich(苏黎世联邦理工学院);University of Würzburg(维尔茨堡大学)
Latent Diffusion Prior Enhanced Deep Unfolding for Snapshot Spectral Compressive Imaging
Zhejiang University(浙江大学);Westlake University(西湖大学);School of Engineering, Westlake University(西湖大学工程学院);Xidian University(西安电子科技大学);School of Cyber Engineering, Xidian University(西安电子科技大学网络与工程学院);Beijing Institute of Technology(北京理工大学);School of Computer Science and Technology, Beijing Institute of Technology(北京理工大学计算机科学与技术学院)
HerosNet: Hyperspectral Explicable Reconstruction and Optimal Sampling Deep Network for Snapshot Compressive Imaging
Peking University Shenzhen Graduate School(北京大学深圳研究生院);Harbin Institute of Technology, Shenzhen(哈尔滨工业大学(深圳));Peking University(北京大学);The Chinese University of Hong Kong, Shenzhen(香港中文大学(深圳));Peng Cheng Laboratory(鹏城实验室)
D3C2-Net: Dual-Domain Deep Convolutional Coding Network for Compressive Sensing
School of Electronic and Computer Engineering, Peking University(北京大学电子与计算机工程学院);Graduate School at Shenzhen, Tsinghua University(清华大学深圳研究生院);ByteDance Inc(字节跳动公司);School of Software Engineering, Tongji University(同济大学软件学院);School of Computer Science and Technology, Harbin Institute of Technology (Shenzhen)(哈尔滨工业大学(深圳)计算机科学与技术学院)
Recursions Are All You Need: Towards Efficient Deep Unfolding Networks
King Fahd University of Petroleum and Minerals(法赫德国王石油与矿业大学);SDAIA-KFUPM Joint Research Center for Artificial Intelligence(沙特数据与人工智能局-法赫德国王石油与矿业大学人工智能联合研究中心)
Memory-augmented Deep Unfolding Network for Guided Image Super-resolution
Hefei Institute of Physical Science, Chinese Academy of Sciences(中国科学院合肥物质科学研究院);University of Science and Technology of China(中国科学技术大学);Nanjing University of Science and Technology(南京理工大学);Institute of Information Engineering, Chinese Academy of Sciences(中国科学院信息工程研究所);Xi’an Jiaotong University(西安交通大学)