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.

1. python 3.7
2. numpy=1.20.3
3. pytorch=1.9.0
4. scipy==1.7.1
5. sklearn
6. scikit-learn==0.24.2
7. pandas==1.3.2
8. 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}}