Skip to content
Beamforming design with deep learning.
Branch: master
Clone or download
Latest commit 40603b6 Jun 13, 2019
Type Name Latest commit message Commit time
Failed to load latest commit information.
Figs fig revision Jun 5, 2019
train_set/example revision for better BFNN architecture May 18, 2019 Update Jun 13, 2019 revision for better BFNN architecture May 18, 2019 add the figures May 19, 2019 revision for better BFNN architecture May 18, 2019


This is the simulation code for the paper "Beamforming Design for Large-Scale Antenna Arrays Using Deep Learning".



  • Tensorflow-gpu = 1.12.0

(Tensorflow 1.12.0 is better for debugging, while tensorflow 1.13.0 using cuda10 can run faster)

If you are confused about how to have several different tensorflows and cudas of different versions in one computer, there is a easy guide may help you (in Chinese).


After fork the repo and download the corresponding data sets and trained models, the following performance results can be easily reproduced. (the python codes is only for the blur cerves, and compared cerves should be plot via Matlab codes)

五连板后走势图 五连板后走势图

How to use:

  • run the to train the model
  • run the to test the trained model
  • Due to the space limitation of github, we provide two tiny training and testing data sets only for running the example.

Data sets and trained models

  • Thanks for the authors of [10], the simulation codes of the channel is provided here. you can use the direct URL to download it.
  • The channel estimation codes can be referred to the website of the first author of [2].
  • We have provided the data sets in our google drive, which can be directly used in our .py files.
  • Trained weights, corresponding to the shown results in the paper, is also provided in the google drive.

Due to some readers requiremnets, for Chinese people, the BaiduYun URL is also provided. (password: z9un).


  • Matlab codes of compared algorithms [4,5] can be referred to this repo. In addition, if you are interested in traditional HBF algorithms, you can kindly refer to our previous work Hybrid Beamforming for Millimeter Wave Systems Using the MMSE Criterion, we also provide specific Matlab codes for your reproduction.
  • If you have any questions, you can contact the author via for help.
  • Hopefully you can star or fork this repo if it is helpful. Thank you for your support in advance.
You can’t perform that action at this time.