Here below are the codes for paper B. Wang, R. Li, J. Zhu, Z. Zhao, and H. Zhang, “Knowledge enhanced semantic communication receiver,” IEEE Commun. Lett.. This is a research project and by definition is unstable. Please write to us if you find something not correct or strange. We are sharing the codes under the condition that reproducing full or part of codes must cite the related papers.
The dataset can be downloaded from https://gitlab.com/shimorina/webnlg-dataset/-/tree/master/release_v3.0.
train_base.py: training the baseline semantic communication system
train_knowledge.py: training the knowledge extractor module
train_full.py: training the complete system
train_UT_*.py: training with a Universal Transformer-based semantic communication system