skip to main content
10.1145/3479239.3485672acmconferencesArticle/Chapter ViewAbstractPublication PagesmswimConference Proceedingsconference-collections
short-paper

Scalable Machine Learning Algorithms to Design Massive MIMO Systems

Published: 22 November 2021 Publication History

Abstract

Machine learning is a highly promising tool to design the physical layer of wireless communication systems, but its scaling properties for this purpose have not been widely studied. Machine learning algorithms are typically evaluated to learn SISO communications and low modulation orders, whereas current wireless standards use MIMO and high-order modulation schemes to increase capacity. The memory requirements of current Machine learning algorithms for wireless communications increase exponentially with the number of antennas and thus they cannot be used for advanced physical layers and massive MIMO. In this paper, we study the requirements of end-to-end Machine learning models for large-scale MIMO systems, determine the bottlenecks of the architecture, and design different solutions that vastly reduce overhead and allow training higher MIMO and modulation orders. We show that by training the autoencoder in a bit-wise manner, the memory requirements are reduced by several orders of magnitude, which is a critical step for Machine learning-based physical layer design in practical scenarios. Additionally, our design also improves performance over the classical autoencoder for MIMO.

References

[1]
Faycc al Ait Aoudia and Jakob Hoydis. 2019. Model-Free Training of End-to-End Communication Systems. IEEE Journal on Selected Areas in Communications, Vol. 37, 11 (2019), 2503--2516.
[2]
Sebastian Cammerer, Faycc al Ait Aoudia, Sebastian Dörner, Maximilian Stark, Jakob Hoydis, and Stephan Ten Brink. 2020. Trainable communication systems: Concepts and prototype. IEEE Transactions on Communications, Vol. 68, 9 (2020), 5489--5503.
[3]
Sebastian Dörner, Sebastian Cammerer, Jakob Hoydis, and Stephan ten Brink. 2017. Deep Learning Based Communication Over the Air. IEEE Journal of Selected Topics in Signal Processing, Vol. 12, 1 (2017), 132--143.
[4]
Yanjie Gao, Yu Liu, Hongyu Zhang, Zhengxian Li, Yonghao Zhu, Haoxiang Lin, and Mao Yang. 2020. Estimating gpu memory consumption of deep learning models. In Proceedings of the 28th ACM Joint Meeting on European Software Engineering Conference and Symposium on the Foundations of Software Engineering. 1342--1352.
[5]
Dolores Garc'ia Mart'i, Joan Palacios Beltrán, Jesús Omar Lacruz, and Joerg Widmer. 2020. A Mixture Density Channel Model for Deep Learning-Based Wireless Physical Layer Design. In Proceedings of the 23rd International ACM Conference on Modeling, Analysis and Simulation of Wireless and Mobile Systems. 53--62.
[6]
Jakob Hoydis, Faycc al Ait Aoudia, Alvaro Valcarce, and Harish Viswanathan. 2020. Towards a 6G AI-Native Air Interface. arXiv preprint arXiv:2012.08285 (2020).
[7]
Paulius Micikevicius, Sharan Narang, Jonah Alben, Gregory Diamos, Erich Elsen, David Garcia, Boris Ginsburg, Michael Houston, Oleksii Kuchaiev, Ganesh Venkatesh, et al. 2017. Mixed precision training. preprint arXiv:1710.03740 (2017).
[8]
Timothy J O'Shea, Tugba Erpek, and T Charles Clancy. 2017. Deep learning based MIMO communications. arXiv preprint arXiv:1707.07980 (2017).
[9]
Timothy J O'Shea, Tamoghna Roy, Nathan West, and Benjamin C Hilburn. 2018. Physical Layer Communications System Design Over-The-Air Using Adversarial Networks. In 2018 26th European Signal Processing Conference (EUSIPCO). IEEE, 529--532.
[10]
Timothy O'Shea and Jakob Hoydis. 2017. An Introduction to Deep Learning for the Physical Layer. IEEE Transactions on Cognitive Communications and Networking, Vol. 3, 4 (2017), 563--575.
[11]
Timothy J O'Shea, Tamoghna Roy, and Nathan West. 2019. Approximating the Void: Learning Stochastic Channel Models from Observation with Variational Generative Adversarial Networks. In 2019 International Conference on Computing, Networking and Communications (ICNC). IEEE, 681--686.
[12]
Jinxiang Song, Christian H"ager, Jochen Schröder, Tim O'Shea, and Henk Wymeersch. 2020. Benchmarking End-to-end Learning of MIMO Physical-Layer Communication. arXiv preprint arXiv:2005.09718 (2020).
[13]
Andrew J. Viterbi. 1998. An intuitive justification and a simplified implementation of the MAP decoder for convolutional codes. IEEE Journal on Selected Areas in Communications, Vol. 16, 2 (1998), 260--264.
[14]
Hao Ye, Geoffrey Ye Li, Biing-Hwang Fred Juang, and Kathiravetpillai Sivanesan. 2018. Channel Agnostic End-to-End Learning Based Communication Systems With Conditional GAN. In 2018 IEEE Globecom Workshops (GC Wkshps). IEEE, UAE, 1--5.

Index Terms

  1. Scalable Machine Learning Algorithms to Design Massive MIMO Systems

      Recommendations

      Comments

      Information & Contributors

      Information

      Published In

      cover image ACM Conferences
      MSWiM '21: Proceedings of the 24th International ACM Conference on Modeling, Analysis and Simulation of Wireless and Mobile Systems
      November 2021
      251 pages
      ISBN:9781450390774
      DOI:10.1145/3479239
      Permission to make digital or hard copies of all or part of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for components of this work owned by others than ACM must be honored. Abstracting with credit is permitted. To copy otherwise, or republish, to post on servers or to redistribute to lists, requires prior specific permission and/or a fee. Request permissions from [email protected]

      Sponsors

      Publisher

      Association for Computing Machinery

      New York, NY, United States

      Publication History

      Published: 22 November 2021

      Permissions

      Request permissions for this article.

      Check for updates

      Author Tags

      1. MIMO
      2. machine learning
      3. neural networks
      4. physical layer

      Qualifiers

      • Short-paper

      Conference

      MSWiM '21
      Sponsor:

      Acceptance Rates

      Overall Acceptance Rate 398 of 1,577 submissions, 25%

      Contributors

      Other Metrics

      Bibliometrics & Citations

      Bibliometrics

      Article Metrics

      • 0
        Total Citations
      • 89
        Total Downloads
      • Downloads (Last 12 months)8
      • Downloads (Last 6 weeks)5
      Reflects downloads up to 27 Jan 2025

      Other Metrics

      Citations

      View Options

      Login options

      View options

      PDF

      View or Download as a PDF file.

      PDF

      eReader

      View online with eReader.

      eReader

      Figures

      Tables

      Media

      Share

      Share

      Share this Publication link

      Share on social media