Reflection-based Word Attribute Transfer

paper (ACL2021 SRW)

Setup

• Install packages according to requirements.txt
• Download pre-trained word2vec(GoogleNews-vectors-negative300.bin)(3.6GB) and move it to data/

$ cd data
$ wget -c https://s3.amazonaws.com/dl4j-distribution/GoogleNews-vectors-negative300.bin.gz
$ gunzip GoogleNews-vectors-negative300.bin.gz

• Download pre-trained GloVe(glove.42B.300d.txt)(5.0GB) and move it to data/
• Fix GloVe file to read it with gensim

$ cd data
$ wget http://nlp.stanford.edu/data/glove.42B.300d.zip
$ unzip glove.42B.300d.zip
$ cd ../src
$ python fix_glove_file.py

Usage

Training example

$ python train.py --attr MF --save-model

Evaluation example

$ python eval.py --attr MF --model-dir ./result/[MODEL DIRECTORY]

Transfer test

$ python trans.py --attr MF --model-dir ./result/[MODEL DIRECTORY] # demo mode
$ python trans.py --attr MF --model-dir ./result/[MODEL DIRECTORY] --src ../data/sentence_example.txt

Citation

@inproceedings{ishibashi-etal-2020-reflection,
    title = "Reflection-based Word Attribute Transfer",
    author = "Ishibashi, Yoichi  and
              Sudoh, Katsuhito  and
              Yoshino, Koichiro  and
              Nakamura, Satoshi",
    booktitle = "Proceedings of the 58th Annual Meeting of the Association for Computational Linguistics: Student Research Workshop",
    month = jul,
    year = "2020",
    address = "Online",
    publisher = "Association for Computational Linguistics",
    url = "https://www.aclweb.org/anthology/2020.acl-srw.8",
    pages = "51--58"
}

@article{ishibashi-etal-2021-reflection,
    title = "Reflection-based Word Attribute Transfer",
    author = "Ishibashi, Yoichi  and
              Sudoh, Katsuhito  and
              Yoshino, Koichiro  and
              Nakamura, Satoshi",
    journal = {Journal of Natural Language Processing},
    pages = "51--58"
    volume={28},
    number={1},
    year = {2021}
}