mandarjoshi90/coref
An implementation of BERT and SpanBERT transformer models for coreference resolution, achieving up to 79.6 F1 on OntoNotes.

Velocity · 7d
+0.2
★ / day
Trend
→steady
star history
This repository provides code and pretrained models for coreference resolution using BERT and SpanBERT. It extends the e2e-coref architecture with transformer-based encoders, enabling span-based mention detection and clustering for identifying all expressions referring to the same entity in text. The models are fine-tuned on the OntoNotes dataset and can be used via pretrained downloads or trained from scratch with GPU resources.