Homework and bake-off: word-level entailment with neural networks


In [ ]:
__author__ = "Christopher Potts"
__version__ = "CS224u, Stanford, Spring 2020"

Overview

The general problem is word-level natural language inference.

Training examples are pairs of words $(w_{L}, w_{R}), y$ with $y = 1$ if $w_{L}$ entails $w_{R}$, otherwise $0$.

The homework questions below ask you to define baseline models for this and develop your own system for entry in the bake-off, which will take place on a held-out test-set distributed at the start of the bake-off. (Thus, all the data you have available for development is available for training your final system before the bake-off begins.)

Set-up

See the first notebook in this unit for set-up instructions.


In [ ]:
from collections import defaultdict
import json
import numpy as np
import os
import pandas as pd
from torch_shallow_neural_classifier import TorchShallowNeuralClassifier
import nli
import utils

In [ ]:
DATA_HOME = 'data'

NLIDATA_HOME = os.path.join(DATA_HOME, 'nlidata')

wordentail_filename = os.path.join(
    NLIDATA_HOME, 'nli_wordentail_bakeoff_data.json')

GLOVE_HOME = os.path.join(DATA_HOME, 'glove.6B')

Data

I've processed the data into two different train/test splits, in an effort to put some pressure on our models to actually learn these semantic relations, as opposed to exploiting regularities in the sample.

  • edge_disjoint: The train and dev edge sets are disjoint, but many words appear in both train and dev.
  • word_disjoint: The train and dev vocabularies are disjoint, and thus the edges are disjoint as well.

These are very different problems. For word_disjoint, there is real pressure on the model to learn abstract relationships, as opposed to memorizing properties of individual words.


In [ ]:
with open(wordentail_filename) as f:
    wordentail_data = json.load(f)

The outer keys are the splits plus a list giving the vocabulary for the entire dataset:


In [ ]:
wordentail_data.keys()

Edge disjoint


In [ ]:
wordentail_data['edge_disjoint'].keys()

This is what the split looks like; all three have this same format:


In [ ]:
wordentail_data['edge_disjoint']['dev'][: 5]

Let's test to make sure no edges are shared between train and dev:


In [ ]:
nli.get_edge_overlap_size(wordentail_data, 'edge_disjoint')

As we expect, a lot of vocabulary items are shared between train and dev:


In [ ]:
nli.get_vocab_overlap_size(wordentail_data, 'edge_disjoint')

This is a large percentage of the entire vocab:


In [ ]:
len(wordentail_data['vocab'])

Here's the distribution of labels in the train set. It's highly imbalanced, which will pose a challenge for learning. (I'll go ahead and reveal that the dev set is similarly distributed.)


In [ ]:
def label_distribution(split):
    return pd.DataFrame(wordentail_data[split]['train'])[1].value_counts()

In [ ]:
label_distribution('edge_disjoint')

Word disjoint


In [ ]:
wordentail_data['word_disjoint'].keys()

In the word_disjoint split, no words are shared between train and dev:


In [ ]:
nli.get_vocab_overlap_size(wordentail_data, 'word_disjoint')

Because no words are shared between train and dev, no edges are either:


In [ ]:
nli.get_edge_overlap_size(wordentail_data, 'word_disjoint')

The label distribution is similar to that of edge_disjoint, though the overall number of examples is a bit smaller:


In [ ]:
label_distribution('word_disjoint')

Baseline

Even in deep learning, feature representation is vital and requires care! For our task, feature representation has two parts: representing the individual words and combining those representations into a single network input.

Representing words: vector_func

Let's consider two baseline word representations methods:

  1. Random vectors (as returned by utils.randvec).
  2. 50-dimensional GloVe representations.

In [ ]:
def randvec(w, n=50, lower=-1.0, upper=1.0):
    """Returns a random vector of length `n`. `w` is ignored."""
    return utils.randvec(n=n, lower=lower, upper=upper)

In [ ]:
# Any of the files in glove.6B will work here:

glove_dim = 50

glove_src = os.path.join(GLOVE_HOME, 'glove.6B.{}d.txt'.format(glove_dim))

# Creates a dict mapping strings (words) to GloVe vectors:
GLOVE = utils.glove2dict(glove_src)

def glove_vec(w):    
    """Return `w`'s GloVe representation if available, else return 
    a random vector."""
    return GLOVE.get(w, randvec(w, n=glove_dim))

Combining words into inputs: vector_combo_func

Here we decide how to combine the two word vectors into a single representation. In more detail, where u is a vector representation of the left word and v is a vector representation of the right word, we need a function vector_combo_func such that vector_combo_func(u, v) returns a new input vector z of dimension m. A simple example is concatenation:


In [ ]:
def vec_concatenate(u, v):
    """Concatenate np.array instances `u` and `v` into a new np.array"""
    return np.concatenate((u, v))

vector_combo_func could instead be vector average, vector difference, etc. (even combinations of those) – there's lots of space for experimentation here; homework question 2 below pushes you to do some exploration.

Classifier model

For a baseline model, I chose TorchShallowNeuralClassifier:


In [ ]:
net = TorchShallowNeuralClassifier(hidden_dim=50, max_iter=100)

Baseline results

The following puts the above pieces together, using vector_func=glove_vec, since vector_func=randvec seems so hopelessly misguided for word_disjoint!


In [ ]:
word_disjoint_experiment = nli.wordentail_experiment(
    train_data=wordentail_data['word_disjoint']['train'],
    assess_data=wordentail_data['word_disjoint']['dev'], 
    model=net, 
    vector_func=glove_vec,
    vector_combo_func=vec_concatenate)

Homework questions

Please embed your homework responses in this notebook, and do not delete any cells from the notebook. (You are free to add as many cells as you like as part of your responses.)

Hypothesis-only baseline [2 points]

During our discussion of SNLI and MultiNLI, we noted that a number of research teams have shown that hypothesis-only baselines for NLI tasks can be remarkably robust. This question asks you to explore briefly how this baseline effects the 'edge_disjoint' and 'word_disjoint' versions of our task.

For this problem, submit two functions:

  1. A vector_combo_func function called hypothesis_only that simply throws away the premise, using the unmodified hypothesis (second) vector as its representation of the example.

  2. A function called run_hypothesis_only_evaluation that does the following:

    1. Loops over the two conditions 'word_disjoint' and 'edge_disjoint' and the two vector_combo_func values vec_concatenate and hypothesis_only, calling nli.wordentail_experiment to train on the conditions 'train' portion and assess on its 'dev' portion, with glove_vec as the vector_func. So that the results are consistent, use an sklearn.linear_model.LogisticRegression with default parameters as the model.
    2. Returns a dict mapping (condition_name, function_name) pairs to the 'macro-F1' score for that pair, as returned by the call to nli.wordentail_experiment. (Tip: you can get the str name of your function hypothesis_only with hypothesis_only.__name__.)

The test functions test_hypothesis_only and test_run_hypothesis_only_evaluation will help ensure that your functions have the desired logic.


In [ ]:
##### YOUR CODE HERE


    ##### YOUR CODE HERE




def run_hypothesis_only_evaluation():
    ##### YOUR CODE HERE

In [ ]:
def test_hypothesis_only(hypothesis_only):
    v = hypothesis_only(1, 2)
    assert v == 2

In [ ]:
test_hypothesis_only(hypothesis_only)

In [ ]:
def test_run_hypothesis_only_evaluation(run_hypothesis_only_evaluation):
    results = run_hypothesis_only_evaluation()
    assert ('word_disjoint', 'vec_concatenate') in results, \
        "The return value of `run_hypothesis_only_evaluation` does not have the intended kind of keys"
    assert isinstance(results[('word_disjoint', 'vec_concatenate')], float), \
        "The values of the `run_hypothesis_only_evaluation` result should be floats"

In [ ]:
test_run_hypothesis_only_evaluation(run_hypothesis_only_evaluation)

Alternatives to concatenation [2 points]

We've so far just used vector concatenation to represent the premise and hypothesis words. This question asks you to explore two simple alternative:

  1. Write a function vec_diff that, for a given pair of vector inputs u and v, returns the element-wise difference between u and v.

  2. Write a function vec_max that, for a given pair of vector inputs u and v, returns the element-wise max values between u and v.

You needn't include your uses of nli.wordentail_experiment with these functions, but we assume you'll be curious to see how they do!


In [ ]:
def vec_diff(u, v):
    ##### YOUR CODE HERE



    
def vec_max(u, v):
    ##### YOUR CODE HERE

In [ ]:
def test_vec_diff(vec_diff):
    u = np.array([10.2, 8.1])
    v = np.array([1.2, -7.1])
    result = vec_diff(u, v)
    expected = np.array([9.0, 15.2])
    assert np.array_equal(result, expected), \
        "Expected {}; got {}".format(expected, result)

In [ ]:
test_vec_diff(vec_diff)

In [ ]:
def test_vec_max(vec_max):
    u = np.array([1.2,  8.1])
    v = np.array([10.2, -7.1])
    result = vec_max(u, v)
    expected = np.array([10.2, 8.1])
    assert np.array_equal(result, expected), \
        "Expected {}; got {}".format(expected, result)

In [ ]:
test_vec_max(vec_max)

A deeper network [2 points]

It is very easy to subclass TorchShallowNeuralClassifier if all you want to do is change the network graph: all you have to do is write a new define_graph. If your graph has new arguments that the user might want to set, then you should also redefine __init__ so that these values are accepted and set as attributes.

For this question, please subclass TorchShallowNeuralClassifier so that it defines the following graph:

$$\begin{align} h_{1} &= xW_{1} + b_{1} \\ r_{1} &= \textbf{Bernoulli}(1 - \textbf{dropout\_prob}, n) \\ d_{1} &= r_1 * h_{1} \\ h_{2} &= f(d_{1}) \\ h_{3} &= h_{2}W_{2} + b_{2} \end{align}$$

Here, $r_{1}$ and $d_{1}$ define a dropout layer: $r_{1}$ is a random binary vector of dimension $n$, where the probability of a value being $1$ is given by $1 - \textbf{dropout_prob}$. $r_{1}$ is multiplied element-wise by our first hidden representation, thereby zeroing out some of the values. The result is fed to the user's activation function $f$, and the result of that is fed through another linear layer to produce $h_{3}$. (Inside TorchShallowNeuralClassifier, $h_{3}$ is the basis for a softmax classifier, so no activation function is applied to it.)

For your implementation, please use nn.Sequential, nn.Linear, and nn.Dropout to define the required layers.

For comparison, using this notation, TorchShallowNeuralClassifier defines the following graph:

$$\begin{align} h_{1} &= xW_{1} + b_{1} \\ h_{2} &= f(h_{1}) \\ h_{3} &= h_{2}W_{2} + b_{2} \end{align}$$

The following code starts this sub-class for you, so that you can concentrate on define_graph. Be sure to make use of self.dropout_prob

For this problem, submit just your completed TorchDeepNeuralClassifier. You needn't evaluate it, though we assume you will be keen to do that!

You can use test_TorchDeepNeuralClassifier to ensure that your network has the intended structure.


In [ ]:
import torch.nn as nn

class TorchDeepNeuralClassifier(TorchShallowNeuralClassifier):
    def __init__(self, dropout_prob=0.7, **kwargs):
        self.dropout_prob = dropout_prob
        super().__init__(**kwargs)
    
    def define_graph(self):
        """Complete this method!
        
        Returns
        -------
        an `nn.Module` instance, which can be a free-standing class you 
        write yourself, as in `torch_rnn_classifier`, or the outpiut of 
        `nn.Sequential`, as in `torch_shallow_neural_classifier`.
        
        """
        ##### YOUR CODE HERE


    

##### YOUR CODE HERE

In [ ]:
def test_TorchDeepNeuralClassifier(TorchDeepNeuralClassifier):
    dropout_prob = 0.55
    assert hasattr(TorchDeepNeuralClassifier(), "dropout_prob"), \
        "TorchDeepNeuralClassifier must have an attribute `dropout_prob`."
    try:
        inst = TorchDeepNeuralClassifier(dropout_prob=dropout_prob)
    except TypeError:
        raise TypeError("TorchDeepNeuralClassifier must allow the user "
                        "to set `dropout_prob` on initialization")
    inst.input_dim = 10
    inst.n_classes_ = 5
    graph = inst.define_graph()
    assert len(graph) == 4, \
        "The graph should have 4 layers; yours has {}".format(len(graph))    
    expected = {
        0: 'Linear',
        1: 'Dropout',
        2: 'Tanh',
        3: 'Linear'}
    for i, label in expected.items():
        name = graph[i].__class__.__name__
        assert label in name, \
            "The {} layer of the graph should be a {} layer; yours is {}".format(i, label, name)
    assert graph[1].p == dropout_prob, \
        "The user's value for `dropout_prob` should be the value of `p` for the Dropout layer."

In [ ]:
test_TorchDeepNeuralClassifier(TorchDeepNeuralClassifier)

Your original system [3 points]

This is a simple dataset, but our focus on the 'word_disjoint' condition ensures that it's a challenging one, and there are lots of modeling strategies one might adopt.

You are free to do whatever you like. We require only that your system differ in some way from those defined in the preceding questions. They don't have to be completely different, though. For example, you might want to stick with the model but represent examples differently, or the reverse.

Keep in mind that, for the bake-off evaluation, the 'edge_disjoint' portions of the data are off limits. You can, though, train on the combination of the 'word_disjoint' 'train' and 'dev' portions. You are free to use different pretrained word vectors and the like. Please do not introduce additional entailment datasets into your training data, though.

Please embed your code in this notebook so that we can rerun it.

In the cell below, please provide a brief technical description of your original system, so that the teaching team can gain an understanding of what it does. This will help us to understand your code and analyze all the submissions to identify patterns and strategies.


In [ ]:
# Enter your system description in this cell.
# Please do not remove this comment.

Bake-off [1 point]

The goal of the bake-off is to achieve the highest macro-average F1 score on word_disjoint, on a test set that we will make available at the start of the bake-off. The announcement will go out on the discussion forum. To enter, you'll be asked to run nli.bake_off_evaluation on the output of your chosen nli.wordentail_experiment run.

The cells below this one constitute your bake-off entry.

The rules described in the Your original system homework question are also in effect for the bake-off.

Systems that enter will receive the additional homework point, and systems that achieve the top score will receive an additional 0.5 points. We will test the top-performing systems ourselves, and only systems for which we can reproduce the reported results will win the extra 0.5 points.

Late entries will be accepted, but they cannot earn the extra 0.5 points. Similarly, you cannot win the bake-off unless your homework is submitted on time.

The announcement will include the details on where to submit your entry.


In [ ]:
# Enter your bake-off assessment code into this cell. 
# Please do not remove this comment.
##### YOUR CODE HERE

In [ ]:
# On an otherwise blank line in this cell, please enter
# your macro-avg f1 value as reported by the code above. 
# Please enter only a number between 0 and 1 inclusive.
# Please do not remove this comment.

##### YOUR CODE HERE