There are two directories in this repository i.e., graph Neural Network (GNN) MIMO detector (https://arxiv.org/pdf/2007.05703.pdf) and graph expectation propagation network (GEPNet) MIMO detector (https://arxiv.org/pdf/2201.03731.pdf). Readers are expected to read the papers before going through the code.
- python 3.7
- numpy=1.20.3
- pytorch=1.9.0
- scipy==1.7.1
- sklearn
- scikit-learn==0.24.2
- pandas==1.3.2
- matplotlib==3.4.3
Please cite this work: @article{kosasih2022gepnet, title={Graph Neural Network Aided Expectation Propagation Detector for MU-MIMO Systems}, author={Kosasih, Alva and Onasis, Vincent and Hardjawana, Wibowo and Miloslavskaya, Vera and Andrean, Victor and Leu, Jenq-Shiou and Vucetic, Branka}, journal={arXiv preprint arXiv:2201.03731}, year={2022}}