View on GitHub

Detecting spouse mentions in sentences

In this tutorial, we will see how Snorkel can be used for Information Extraction. We will walk through an example text classification task for information extraction, where we use labeling functions involving keywords and distant supervision.

Classification Task

We want to classify each candidate or pair of people mentioned in a sentence, as being married at some point or not.

In the above example, our candidate represents the possible relation (Barack Obama, Michelle Obama). As readers, we know this mention is true due to external knowledge and the keyword of wedding occuring later in the sentence. We begin with some basic setup and data downloading.

from utils import load_data

((df_dev, Y_dev), df_train, (df_test, Y_test)) = load_data()

Input Data: df_dev, df_train, and df_test are Pandas DataFrame objects, where each row represents a particular candidate. For our problem, a candidate consists of a sentence, and two people mentioned in the sentence. The DataFrames contain the fields sentence, which refers to the sentence of the candidate, tokens, the tokenized form of the sentence, and person1_word_idx and person2_word_idx, which represent [start, end] indices in the tokens at which the first and second person’s name appear, respectively.

We also have certain preprocessed fields, that we discuss a few cells below.

Let’s look at a candidate in the development set:

from preprocessors import get_person_text

candidate = df_dev.loc[2]
person_names = get_person_text(candidate).person_names

print("Sentence: ", candidate["sentence"])
print("Person 1: ", person_names[0])
print("Person 2: ", person_names[1])
Sentence:  The Richards are half-sisters to Kathy Hilton, the mother of socialite Paris Hilton and spouse of luxury hotel magnate Richard Howard Hilton.   
Person 1:  Kathy Hilton
Person 2:  Richard Howard Hilton

Preprocessing the Data

In a real application, there is a lot of data preparation, parsing, and database loading that needs to be completed before we generate candidates and dive into writing labeling functions. Here we’ve pre-generated candidates in a pandas DataFrame object per split (train,dev,test).

Labeling Function Helpers

When writing labeling functions, there are several functions you will use over and over again. In the case of text relation extraction as with this task, common functions include those for fetching text between mentions of the two people in a candidate, examing word windows around person mentions, and so on. We will wrap these functions as preprocessors.

from snorkel.preprocess import preprocessor


@preprocessor()
def get_text_between(cand):
    """
    Returns the text between the two person mentions in the sentence for a candidate
    """
    start = cand.person1_word_idx[1] + 1
    end = cand.person2_word_idx[0]
    cand.text_between = " ".join(cand.tokens[start:end])
    return cand

Candidate PreProcessors

For the purposes of the tutorial, we have three fields (between_tokens, person1_right_tokens, person2_right_tokens) preprocessed in the data, which can be used when creating labeling functions. We also provide the following set of preprocessors for this task in preprocessors.py, along with the fields these populate.

  • get_person_text(cand): person_names
  • get_person_lastnames(cand): person_lastnames
  • get_left_tokens(cand): person1_left_tokens, person2_left_tokens
from preprocessors import get_left_tokens, get_person_last_names

POSITIVE = 1
NEGATIVE = 0
ABSTAIN = -1
from snorkel.labeling import labeling_function

# Check for the `spouse` words appearing between the person mentions
spouses = {"spouse", "wife", "husband", "ex-wife", "ex-husband"}


@labeling_function(resources=dict(spouses=spouses))
def lf_husband_wife(x, spouses):
    return POSITIVE if len(spouses.intersection(set(x.between_tokens))) > 0 else ABSTAIN
# Check for the `spouse` words appearing to the left of the person mentions
@labeling_function(resources=dict(spouses=spouses), pre=[get_left_tokens])
def lf_husband_wife_left_window(x, spouses):
    if len(set(spouses).intersection(set(x.person1_left_tokens))) > 0:
        return POSITIVE
    elif len(set(spouses).intersection(set(x.person2_left_tokens))) > 0:
        return POSITIVE
    else:
        return ABSTAIN
# Check for the person mentions having the same last name
@labeling_function(pre=[get_person_last_names])
def lf_same_last_name(x):
    p1_ln, p2_ln = x.person_lastnames

    if p1_ln and p2_ln and p1_ln == p2_ln:
        return POSITIVE
    return ABSTAIN
# Check for the word `married` between person mentions
@labeling_function()
def lf_married(x):
    return POSITIVE if "married" in x.between_tokens else ABSTAIN
# Check for words that refer to `family` relationships between and to the left of the person mentions
family = {
    "father",
    "mother",
    "sister",
    "brother",
    "son",
    "daughter",
    "grandfather",
    "grandmother",
    "uncle",
    "aunt",
    "cousin",
}
family = family.union({f + "-in-law" for f in family})


@labeling_function(resources=dict(family=family))
def lf_familial_relationship(x, family):
    return NEGATIVE if len(family.intersection(set(x.between_tokens))) > 0 else ABSTAIN


@labeling_function(resources=dict(family=family), pre=[get_left_tokens])
def lf_family_left_window(x, family):
    if len(set(family).intersection(set(x.person1_left_tokens))) > 0:
        return NEGATIVE
    elif len(set(family).intersection(set(x.person2_left_tokens))) > 0:
        return NEGATIVE
    else:
        return ABSTAIN
# Check for `other` relationship words between person mentions
other = {"boyfriend", "girlfriend", "boss", "employee", "secretary", "co-worker"}


@labeling_function(resources=dict(other=other))
def lf_other_relationship(x, other):
    return NEGATIVE if len(other.intersection(set(x.between_tokens))) > 0 else ABSTAIN

Distant Supervision Labeling Functions

In addition to using factories that encode pattern matching heuristics, we can also write labeling functions that distantly supervise data points. Here, we’ll load in a list of known spouse pairs and check to see if the pair of persons in a candidate matches one of these.

DBpedia: Our database of known spouses comes from DBpedia, which is a community-driven resource similar to Wikipedia but for curating structured data. We’ll use a preprocessed snapshot as our knowledge base for all labeling function development.

We can look at some of the example entries from DBPedia and use them in a simple distant supervision labeling function.

Make sure dbpedia.pkl is in the spouse/data directory.

with open("data/dbpedia.pkl", "rb") as f:
    known_spouses = pickle.load(f)

list(known_spouses)[0:5]
[('Gustave Choquet', 'Yvonne Choquet-Bruhat'),
 ('Richard Karn', 'Tudi Roche'),
 ('Kaarin Fairfax', 'Paul Kelly'),
 ('Ana de Armas', 'Marc Clotet'),
 ('Mami Tai', 'Yasushi Akimoto')]
@labeling_function(resources=dict(known_spouses=known_spouses), pre=[get_person_text])
def lf_distant_supervision(x, known_spouses):
    p1, p2 = x.person_names
    if (p1, p2) in known_spouses or (p2, p1) in known_spouses:
        return POSITIVE
    else:
        return ABSTAIN
from preprocessors import last_name

# Last name pairs for known spouses
last_names = set(
    [
        (last_name(x), last_name(y))
        for x, y in known_spouses
        if last_name(x) and last_name(y)
    ]
)


@labeling_function(resources=dict(last_names=last_names), pre=[get_person_last_names])
def lf_distant_supervision_last_names(x, last_names):
    p1_ln, p2_ln = x.person_lastnames

    return (
        POSITIVE
        if (p1_ln != p2_ln)
        and ((p1_ln, p2_ln) in last_names or (p2_ln, p1_ln) in last_names)
        else ABSTAIN
    )

Apply Labeling Functions to the Data

We create a list of labeling functions and apply them to the data

from snorkel.labeling import PandasLFApplier

lfs = [
    lf_husband_wife,
    lf_husband_wife_left_window,
    lf_same_last_name,
    lf_married,
    lf_familial_relationship,
    lf_family_left_window,
    lf_other_relationship,
    lf_distant_supervision,
    lf_distant_supervision_last_names,
]
applier = PandasLFApplier(lfs)
from snorkel.labeling import LFAnalysis

L_dev = applier.apply(df_dev)
L_train = applier.apply(df_train)
LFAnalysis(L_dev, lfs).lf_summary(Y_dev)
j Polarity Coverage Overlaps Conflicts Correct Incorrect Emp. Acc.
lf_husband_wife 0 [1] 0.089648 0.036642 0.017432 93 159 0.369048
lf_husband_wife_left_window 1 [1] 0.025258 0.021345 0.003557 30 41 0.422535
lf_same_last_name 2 [1] 0.040555 0.016009 0.008538 19 95 0.166667
lf_married 3 [1] 0.019210 0.006759 0.002490 22 32 0.407407
lf_familial_relationship 4 [0] 0.115617 0.051939 0.026325 310 15 0.953846
lf_family_left_window 5 [0] 0.041266 0.033440 0.007826 114 2 0.982759
lf_other_relationship 6 [0] 0.013874 0.002846 0.002846 33 6 0.846154
lf_distant_supervision 7 [1] 0.001067 0.001067 0.000000 2 1 0.666667
lf_distant_supervision_last_names 8 [1] 0.001067 0.000711 0.000356 0 3 0.000000

Training the Label Model

Now, we’ll train a model of the LFs to estimate their weights and combine their outputs. Once the model is trained, we can combine the outputs of the LFs into a single, noise-aware training label set for our extractor.

from snorkel.labeling import LabelModel

label_model = LabelModel(cardinality=2, verbose=True)
label_model.fit(L_train, Y_dev, n_epochs=5000, log_freq=500, seed=12345)

Label Model Metrics

Since our dataset is highly unbalanced (91% of the labels are negative), even a trivial baseline that always outputs negative can get a high accuracy. So we evaluate the label model using the F1 score and ROC-AUC rather than accuracy.

from snorkel.analysis import metric_score
from snorkel.utils import probs_to_preds

probs_dev = label_model.predict_proba(L_dev)
preds_dev = probs_to_preds(probs_dev)
print(
    f"Label model f1 score: {metric_score(Y_dev, preds_dev, probs=probs_dev, metric='f1')}"
)
print(
    f"Label model roc-auc: {metric_score(Y_dev, preds_dev, probs=probs_dev, metric='roc_auc')}"
)
Label model f1 score: 0.4199134199134199
Label model roc-auc: 0.7421454246069199

Part 4: Training our End Extraction Model

In this final section of the tutorial, we’ll use our noisy training labels to train our end machine learning model. We start by filtering out training data points which did not recieve a label from any LF, as these data points contain no signal.

from snorkel.labeling import filter_unlabeled_dataframe

probs_train = label_model.predict_proba(L_train)
df_train_filtered, probs_train_filtered = filter_unlabeled_dataframe(
    X=df_train, y=probs_train, L=L_train
)

Next, we train a simple LSTM network for classifying candidates. tf_model contains functions for processing features and building the keras model for training and evaluation.

from tf_model import get_model, get_feature_arrays
from utils import get_n_epochs

X_train = get_feature_arrays(df_train_filtered)
model = get_model()
batch_size = 64
model.fit(X_train, probs_train_filtered, batch_size=batch_size, epochs=get_n_epochs())

Finally, we evaluate the trained model by measuring its F1 score and ROC_AUC.

X_test = get_feature_arrays(df_test)
probs_test = model.predict(X_test)
preds_test = probs_to_preds(probs_test)
print(
    f"Test F1 when trained with soft labels: {metric_score(Y_test, preds=preds_test, metric='f1')}"
)
print(
    f"Test ROC-AUC when trained with soft labels: {metric_score(Y_test, probs=probs_test, metric='roc_auc')}"
)
Test F1 when trained with soft labels: 0.4705882352941177
Test ROC-AUC when trained with soft labels: 0.7386078545115963

Summary

In this tutorial, we showed how Snorkel can be used for Information Extraction. We demonstrated how to create LFs that leverage keywords and external knowledge bases (distant supervision). Finally, we showed how a model trained using the probabilistic outputs of the Label Model can achieve comparable performance while generalizing to all data points.