This is a pytorch implementation of the WWW 2021 paper "Unsupervised Semantic Association Learning with Latent Label Inference".
python3.6.0+
pytorch 1.7.0+
transformers
tensorboard
Download the data, and change the datapath in the configs file.
To train a new model, run the following command:
python main.py --dataset='config name' --main=wsd.py/qa.py --do_train
If this work is helpful, please cite as:
@inproceedings{zhang2021unsupervised,
title={Unsupervised Semantic Association Learning with Latent Label Inference},
author={Zhang, Yanzhao and Zhang, Richong and Kim, Jaein and Liu, Xudong and Mao, Yongyi},
booktitle={Proceedings of the Web Conference 2021},
pages={4010--4019},
year={2021}
}
MIT