Advertisement

Adaptive Measurement Network for CS Image Reconstruction

  • Xuemei Xie
  • Yuxiang Wang
  • Guangming Shi
  • Chenye Wang
  • Jiang Du
  • Xiao Han
Conference paper
Part of the Communications in Computer and Information Science book series (CCIS, volume 772)

Abstract

Conventional compressive sensing (CS) reconstruction is very slow for its characteristic of solving an optimization problem. Convolutional neural network can realize fast processing while achieving comparable results. While CS image recovery with high quality not only depends on good reconstruction algorithms, but also good measurements. In this paper, we propose an adaptive measurement network in which measurement is obtained by learning. The new network consists of a fully-connected layer and ReconNet. The fully-connected layer which has low-dimension output acts as measurement. We train the fully-connected layer and ReconNet simultaneously and obtain adaptive measurement. Because the adaptive measurement fits dataset better, in contrast with random Gaussian measurement matrix, under the same measurement rate, it can extract the information of scene more efficiently and get better reconstruction results. Experiments show that the new network outperforms the original one.

Keywords

Compressive sensing Image reconstruction Deep learning Adaptive measurement 

Notes

Acknowledgements

This work is supported by the National Natural Science Foundation of China (Grant No. 61472301, 61632019) and the Foundation for Innovative Research Groups of the National Natural Science Foundation of China (No. 61621005).

References

  1. 1.
    Kašin, B.S.: The widths of certain finite-dimensional sets and classes of smooth functions. Izv. Akad. Nauk SSSR Ser. Mat. 41(2), 334–351 (1977)MathSciNetGoogle Scholar
  2. 2.
    Candès, E., Romberg, J.: Sparsity and incoherence in compressive sampling. Inverse Prob. 23(3), 969–985 (2007)MathSciNetCrossRefzbMATHGoogle Scholar
  3. 3.
    Rauhut, H.: Random sampling of sparse trigonometric polynomials. Found. Comput. Math. 22(6), 737–763 (2008)MathSciNetzbMATHGoogle Scholar
  4. 4.
    Candes, E.J., Romberg, J., Tao, T.: Robust uncertainty principles: exact signal reconstruction from highly incomplete frequency information. IEEE Trans. Inf. Theory 52(2), 489–509 (2004)MathSciNetCrossRefzbMATHGoogle Scholar
  5. 5.
    Duarte, M.F., Wakin, M.B., Baraniuk, R.G.: Wavelet-domain compressive signal reconstruction using a Hidden Markov Tree model. In: IEEE International Conference on Acoustics, Speech and Signal Processing, pp. 5137–5140. IEEE (2008)Google Scholar
  6. 6.
    Peyré, G., Bougleux, S., Cohen, L.: Non-local regularization of inverse problems. In: Forsyth, D., Torr, P., Zisserman, A. (eds.) ECCV 2008. LNCS, vol. 5304, pp. 57–68. Springer, Heidelberg (2008).  https://doi.org/10.1007/978-3-540-88690-7_5 CrossRefGoogle Scholar
  7. 7.
    Baraniuk, R.G., Cevher, V., Duarte, M.F., et al.: Model-based compressive sensing. IEEE Trans. Inf. Theory 56(4), 1982–2001 (2010)MathSciNetCrossRefzbMATHGoogle Scholar
  8. 8.
    Li, C., Yin, W., Jiang, H., et al.: An efficient augmented Lagrangian method with applications to total variation minimization. Comput. Optim. Appl. 56(3), 507–530 (2013)MathSciNetCrossRefzbMATHGoogle Scholar
  9. 9.
    Kim, Y., Nadar, M.S., Bilgin, A.: Compressed sensing using a Gaussian Scale Mixtures model in wavelet domain. In: IEEE International Conference on Image Processing, pp. 3365–3368. IEEE (2010)Google Scholar
  10. 10.
    Kulkarni, K., Lohit, S., Turaga, P., Kerviche, R., Ashok, A.: Reconnet: non-iterative reconstruction of images from compressively sensed measurements. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 449–458 (2016)Google Scholar
  11. 11.
    Mousavi, A., Patel, A.B., Baraniuk, R.G.: A deep learning approach to structured signal recovery. In: 2015 53rd Annual Allerton Conference on Communication, Control, and Computing (Allerton), pp. 1336–1343. IEEE, September 2015Google Scholar
  12. 12.
    Yao, H., Dai, F., Zhang, D., Ma, Y., Zhang, S., Zhang, Y.: DR\(^{2}\)-Net: Deep Residual Reconstruction Network for Image Compressive Sensing. arXiv preprint arXiv:1702.05743 (2017)
  13. 13.
    Mousavi, A., Baraniuk, R.G.: Learning to invert: signal recovery via deep convolutional networks. arXiv preprint arXiv:1701.03891 (2017)
  14. 14.
    Dong, C., Loy, C.C., He, K., Tang, X.: Learning a deep convolutional network for image super-resolution. In: Fleet, D., Pajdla, T., Schiele, B., Tuytelaars, T. (eds.) ECCV 2014. LNCS, vol. 8692, pp. 184–199. Springer, Cham (2014).  https://doi.org/10.1007/978-3-319-10593-2_13 Google Scholar

Copyright information

© Springer Nature Singapore Pte Ltd. 2017

Authors and Affiliations

  • Xuemei Xie
    • 1
  • Yuxiang Wang
    • 1
  • Guangming Shi
    • 1
  • Chenye Wang
    • 1
  • Jiang Du
    • 1
  • Xiao Han
    • 1
  1. 1.Xidian UniversityXi’anChina

Personalised recommendations