Note This repository is no longer actively maintained by Babylon Health. For further assistance, reach out to the paper authors.
TensorFlow implementation of the models described in the Decoding Decoders paper.
This codebase builds on top of Tensorflow Skip-Thought implementation by Chris Shallue and uses SentEval from Facebook for evaluations on transfer tasks.
The aim is to study how different choices of decoders affect the performance on unsupervised similarity tasks, such as STS.
This code uses Python 2.7. Please install the requirements in requirements.txt.
You will need to obtain the BookCorpus dataset from this website.
# Comma-separated list of globs matching the input files. The format of
# the input files is assumed to be a list of newline-separated sentences, where
# each sentence is already tokenized.
INPUT_FILES=<raw input files>
# Location to save the preprocessed training and validation data.
DATA_DIR=<data directory>
# Run the preprocessing script.
python -m skip_thoughts.data.preprocess_dataset \
--input_files=${INPUT_FILES} \
--output_dir=${DATA_DIR}We added a couple of new parameters in the train.py script.
The most important ones are described here, please see the code to see additional functionality we have added.
--decoder=SEQxSKGy where x, y can be 0, 1, 2, and 3.
SEQ stands for sequence (recurrent) decoder and SKG stands for bag-of-words (BOW) decoder.
0- no decoder of this type is present1- decoder for the current sentence (Autoencoder)2- decoders for the previous and next sentences (Skip-Though/FastSent style)3- decoders for previous, current, and next sentences (Skip-Thought + Autoencoder) Note that it is possible to combine SEQ and SKG
--skipgram_encoder=True|False
TrueThe architecture has a bag-of-words (BOW) encoder.FalseThe architecture has a sequence (RNN) encoder.
Defaults to False.
# Directory containing the preprocessed data.
DATA_DIR=<data directory>
# Directory to save the model. Note: A new folder will be created in here called run_{unixtimestamp}. Into this folder, the model checkpoints will be saved. Also, the FLAGS wile, as well as its dict and json representations will be stored as `flags.pkl`, `config.pkl` and `config.json` respectively.
RUN_DIR=<run directory>
# Model decoder configuration (choose one of SEQ0SKG2 SEQ0SKG3 SEQ2SKG2 or SEQ3SKG3)
DECODER="SEQ0SKG2"
# Whether to use skipgram (BOW) encoder (choose True or False). Defaults to False.
SKIPGRAM_ENCODER=False
# Run the training script.
python -m skip_thoughts.train \
--input_file_pattern="${DATA_DIR}/train-?????-of-00100" \
--run_dir="${RUN_DIR}" \
--decoder="${DECODER}" \
--skipgram_encoder="${SKIPGRAM_ENCODER}"This will train a model with an RNN encoder and 2 BOW decoders.
You will need to download the pretrained Google News word2vec vectors, found here. Please see the SkipThought readme for more details on vocab expansion.
MODEL_DIR=<path to model>
SKIP_THOUGHTS_VOCAB=<path to skipthoughts vocab>
W2VMODEL=<path to W2V model>
LOG_FILE=<path to log file>
python -m skip_thoughts.vocabulary_expansion \
--skip_thoughts_model="${MODEL_DIR}" \
--skip_thoughts_vocab="${SKIP_THOUGHTS_VOCAB}" \
--word2vec_model="${W2VMODEL}" \
--output_dir="${MODEL_DIR}" \
> "${LOG_FILE}" 2>&1You will need to clone the SentEval repo and download the data as instructed there.
Then copy our scripts from sent_eval/evaluation to the examples directory to run.
The SentEval evaluation scripts either use the encoder output (which we confusingly call context here), or the unrolled decoder (which we less confusingly call unroll for the similarity and transfer tasks.
The similarity scripts
similarity_context.py and
similarity_unroll.py
run the STS* tasks (STS12, STS13, STS14, STS15 and STS16) of SentEval.
The transfer scripts
transfer_context.py and
transfer_unroll.py
run the transfer tasks (CR, MR, MPQA, SUBJ, SST, TREC, MRPC,
SICKRelatedness, SICKEntailment and STSBenchmark) of SentEval.
Each script runs with 10-fold cross validation, and saves the dictionary of all results as a pickle to the desired location. This can then be used for easy generation of plots and other analysis.
The context scripts similarity_context.py and transfer_context.py work for all decoder types.
The parameters of the context scripts are:
--model_dirThe path to the saved model you want to evaluate. Specifically, this should include this should be a folder containined checkpoint and decoder configuration information produced bytrain.py.--output_results_pathThe full path to save the pickle file containing all of the results from this evaluation.
The unroll scripts similarity_unroll.py and transfer_unroll.py only work for RNN decoder types, and use the decoder unrolling mechanism discussed in the Decoding Decoders paper.
In addition to the parameters of the context scripts (above), the unroll scripts require the following parameters:
--unroll_lengthThis should be a positive integer, and corresponds to how many time steps each decoder will "unroll" to produce the sentence representation.--decoder_typeThis should be either'mean'or'concat'and corresponds to either taking the sentence representation as the mean or concatentation over the unrolled hidden states respectively.
The example below is for running similarity_context.py, the exact same process will work for the other evaluation scripts.
# Directory to load the model from
MODEL_DIR=<model directory>
# Which GPU(s) to use (choose from e.g. one of [0 1 0,1])
GPU_IDS=0
# Log file
LOG_FILE=<log file>
# Pickle save path
PICKLE_PATH=<pickle path>
# Run the evaluation script.
CUDA_VISIBLE_DEVICES=$GPU_IDS \
python -m sent_eval.evaluation.similarity_context \
--model_dir="${MODEL_DIR}" \
--output_results_path="${PICKLE_PATH}" \
> "${LOG_FILE}" 2>&1Vitalii Zhelezniak vitali.zhelezniak@babylonhealth.com