FLDRL-in-Wireless-Communication

February 16, 2021 ยท View on GitHub

  • Simulation code for Paper:
    Lyutianyang Zhang1 , Hao Yin1, Zhanke Zhou, Sumit Roy, Yaping Sun, Enhancing WiFi Multiple Access Performance with Federated Deep Reinforcement Learning, VTC2020-Fall.
    1 Both authors contribute equally to this work.
  • Cite our work:
@INPROCEEDINGS{FrmaVTC2020,
  author={L. {Zhang} and H. {Yin} and Z. {Zhou} and S. {Roy} and Y. {Sun}},
  booktitle={IEEE 92nd Vehicular Technology Conference (VTC2020-Fall)}, 
  title={Enhancing {WiFi} Multiple Access Performance with Federated Deep Reinforcement Learning}
  }

Contributors: Hao Yin, Zhanke Zhou

The paper can be found https://ieeexplore.ieee.org/document/9348485

Simulations

Author Notes:

  • Please check config.py for model loading and saving setups.

    • self.saveModel = False
      self.loadModel = True
      
  • Run python3 test_CSMA_DQN_withModelAllocation.py to proceed training.

  • Throughput is about 5.2-5.4

Training log

Number of StationMax Avg ThroughputTotal training epoch
55.4510w
105.4613w
205.2822w