compnet-renard/bert-base-cased-literary-coref-v2

This model should be used with Tibert, a BERT-based end-to-end coreference solver.

Example code:

from tibert import BertForCoreferenceResolution, predict_coref_simple
from tibert.utils import pprint_coreference_document
from transformers import BertTokenizerFast

model = BertForCoreferenceResolution.from_pretrained(
    "compnet-renard/bert-base-cased-literary-coref-v2"
)
tokenizer = BertTokenizerFast.from_pretrained("bert-base-cased")

coref_out = predict_coref_simple(
    "Sli did not want the earpods. He didn't like them.", model, tokenizer
)

pprint_coreference_document(coref_out)

>>> (0 Sli ) did not want the earpods. (0 He ) didn't like them.

Performance

The following table presents the results we obtained on Litbank by training this model. We evaluate on 10% of Litbank documents, each of which consists of ~2000 tokens. The split column indicate whether documents were split in blocks of 512 tokens. The HM coumns indicates whether we use hierarchical merging. Time is measured on cpu inference.

compnet-renard/bert-base-cased-literary-coref-v2