COAST: COntrollable Arbitrary-Sampling NeTwork for Compressive Sensing

• Published in: IEEE transactions on image processing Vol. 30; pp. 6066 - 6080
• Main Authors: You, Di, Zhang, Jian, Xie, Jingfen, Chen, Bin, Ma, Siwei
• Format: Journal Article
• Language: English
• Published: New York IEEE 2021; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: Compressed sensing; compressive sensing (CS); Computational modeling; deep unfolding network; Image reconstruction; Inverse problems; Magnetic resonance imaging; neural networks; Optimization; optimization-inspired; Sampling; Task analysis; Training
• ISSN: 1057-7149; 1941-0042
• DOI: 10.1109/TIP.2021.3091834

Recent deep network-based compressive sensing (CS) methods have achieved great success. However, most of them regard different sampling matrices as different independent tasks and need to train a specific model for each target sampling matrix. Such practices give rise to inefficiency in computing and suffer from poor generalization ability. In this paper, we propose a novel COntrollable Arbitrary-Sampling neTwork, dubbed COAST, to solve CS problems of arbitrary-sampling matrices (including unseen sampling matrices) with one single model. Under the optimization-inspired deep unfolding framework, our COAST exhibits good interpretability. In COAST, a random projection augmentation (RPA) strategy is proposed to promote the training diversity in the sampling space to enable arbitrary sampling, and a controllable proximal mapping module (CPMM) and a plug-and-play deblocking (PnP-D) strategy are further developed to dynamically modulate the network features and effectively eliminate the blocking artifacts, respectively. Extensive experiments on widely used benchmark datasets demonstrate that our proposed COAST is not only able to handle arbitrary sampling matrices with one single model but also to achieve state-of-the-art performance with fast speed.