CsiNet-Pytorch

KO | EN

gitlite — search

#python #typescript #golang #linux #docker #macos #java #awesome #kubernetes #awesome-list #vue #pytorch #mcp #open-source

(★ 31)

Pytorch reproduction of CsiNet

CsiNet-Pytorch Latest Version Download

Download Latest Version (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# CsiNet-Pytorch  
Pytorch implementation of CsiNet (C. Wen, 2018)  
and online learning (Y. Cui, 2022)

[1] Chao-Kai Wen, Wan-Ting Shih, and Shi Jin, “Deep learning for massive MIMO CSI feedback,” IEEE Wireless Communications Letters, 2018.   
[2] Yiming Cui, Jiajia Guo, Chao-Kai Wen, Shi Jin, and Shuangfeng Han, "Unsupervised Online Learning in Deep Learning-Based Massive MIMO CSI Feedback", Unsupervised Online Learning in Deep Learning-Based Massive MIMO CSI Feedback, 2022.

some of functions are from https://github.com/WilliamYangXu/CSITransformer.
  
**data**:  
You can download QuaDRiGa in https://quadriga-channel-model.de/ and generate channel.  
      Please put the channel data csv file in ./filepath folder.  
      Sample datasets are available in https://drive.google.com/drive/folders/1cWkGpdYPxrHSN55phGgyJ4YGHFZgSCS9?usp=sharing
      
**run**:   
open 'main.py'  
adjust parameters (dimension of codeword, the number of epochs, ...)  
     and run 'main.py'.  
     
  
**explanation of each python file**  
- data.py: generates data, returns train dataloader and test dataloader  
- train.py: run train epochs and evaluations.  
- model.py: neural network models of encoer and decoder  
- main.py: run file

**update 20230116**  
Here is the online learning with different scenarios added.  
The model architecture is same as CsiNet(2018), not the online learning paper(2022)  
**What is online learning?**  
In the real time communication scenarios, each BS experience different environment (i.e., different probability density of channel).    
Therefore, it is helpful to do fine tuning with each BS's data.  
Since UE cannot transmit the whole channel matrix to BS,   
BS doesn't have knowledge about data label.  
Moreover, BS can't deliver the gradient flow to UE, so that the encoder in UE cannot be updated.  
Therefore, the online learning only learns decoder part, without knowledge of channel label.

CsiNet-Pytorch

CsiNet-Pytorch Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

CsiNet-Pytorch

CsiNet-Pytorch Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

Download README