Vector Quantization for Deep-Learning-Based CSI Feedback in Massive MIMO Systems (2403.07355v2)
Abstract: This paper presents a finite-rate deep-learning (DL)-based channel state information (CSI) feedback method for massive multiple-input multiple-output (MIMO) systems. The presented method provides a finite-bit representation of the latent vector based on a vector-quantized variational autoencoder (VQ-VAE) framework while reducing its computational complexity based on shape-gain vector quantization. In this method, the magnitude of the latent vector is quantized using a non-uniform scalar codebook with a proper transformation function, while the direction of the latent vector is quantized using a trainable Grassmannian codebook. A multi-rate codebook design strategy is also developed by introducing a codeword selection rule for a nested codebook along with the design of a loss function. Simulation results demonstrate that the proposed method reduces the computational complexity associated with VQ-VAE while improving CSI reconstruction performance under a given feedback overhead.
- C.-K. Wen, W.-T. Shih, and S. Jin, “Deep learning for massive MIMO CSI feedback,” IEEE Wireless Commun. Lett., vol. 7, no. 5, pp. 748–751, Oct. 2018.
- Z. Lu, J. Wang, and J. Song, “Multi-resolution CSI feedback with deep learning in massive MIMO system,” in Proc. IEEE Int. Conf. Commun. (ICC), June 2020, pp. 1–6.
- J. Guo, C.-K. Wen, S. Jin, and G. Y. Li, “Convolutional neural network-based multiple-rate compressive sensing for massive MIMO CSI feedback: Design, simulation, and analysis,” IEEE Trans. Wireless Commun., vol. 19, no. 4, pp. 2827–2840, Apr. 2020.
- D. J. Love, R. W. Heath, Jr., V. K. Lau, D. Gesbert, B. D. Rao, and M. Andrews, “An overview of limited feedback in wireless communication systems,” IEEE J. Sel. Areas Commun., vol. 26, no. 8, pp. 1341–1365, Oct. 2008.
- Z. Qin, J. Fan, Y. Liu, Y. Gao, and G. Y. Li, “Sparse representation for wireless communications: A compressive sensing approach,” IEEE Signal Process. Mag., vol. 35, no. 3, pp. 40–58, May 2018.
- X. Liang, H. Chang, H. Li, X. Gu, and L. Zhang, “Changeable rate and novel quantization for CSI feedback based on deep learning,” IEEE Trans. Wireless Commun., vol. 21, no. 12, pp. 10100–10114, Dec. 2022.
- Z. Liu, L. Zhang, and Z. Ding, “An efficient deep learning framework for low rate massive MIMO CSI reporting,” IEEE Trans. Commun., vol. 68, no. 8, pp. 4761–4772, Aug. 2020.
- V. Rizzello, M. Nerini, M. Joham, B. Clerckx, and W. Utschick, “User-driven adaptive CSI feedback with ordered vector quantization,” IEEE Wireless Commun. Lett., vol. 12, no. 11, pp. 1956–1960, Nov. 2023.
- J. Yang, X. Chen, H. Zou, D. Wang, Q. Xu, and L. Xie, “EfficientFi: toward large-scale lightweight WiFi sensing via CSI compression,” IEEE Internet of Things Journal, vol. 9, no. 15, pp. 13086–13095, Aug., 2022.
- A. van den Oord, O. Vinyals, and K. Kavukcuoglu, “Neural discrete representation learning,” Adv. Neural Inf. Process. Syst., Dec. 2017, pp. 6306–6315.
- M. Malka, S. Ginzach, and N. Shlezinger, “Learning multi-rate vector quantization for remote deep inference,” in Proc. IEEE Int. Conf. Acoustics, Speech, Sig. Process. Workshops (ICASSPW), June 2023, pp. 1–5.
- N. Shlezinger and Y. C. Eldar, “Deep task-based quantization,” Entropy, vol. 23, no. 1, 104, Jan. 2021.
- Y. Linde, A. Buzo, and R. Gray, “An algorithm for vector quantizer design,” IEEE Trans. Commun., vol. 28, no. 1, pp. 84–95, Jan. 1980.
- X. Liang, Z. Jia, X. Gu, and L. Zhang, “Towards better low-rate deep learning-based CSI feedback: a test channel-based approach,” IEEE Trans. Wireless Commun., early access. doi: 10.1109/TWC.2024.3354238.