huggingface / transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
https://huggingface.co/transformers
Apache License 2.0
135.45k stars 27.11k forks source link

BERT and SpanBERT for Coreference Resolution #6497

Open sayanb-7c6 opened 4 years ago

sayanb-7c6 commented 4 years ago

🌟 New model addition

Model description

This is a recent approach for co-reference resolution using BERT, implemented from the papers BERT for Coreference Resolution: Baselines and Analysis and SpanBERT: Improving Pre-training by Representing and Predicting Spans, which is the current state of the art on OntoNotes (79.6 F1). It uses tensorflow 1.14.0.

Reason why this is interesting is it achieves strong improvements on the OntoNotes (+3.9 F1) and GAP (+11.5 F1) benchmarks. Also, I think it would be a nice addition to huggingface library, as it has only the neuralcoref as the coreference resolution module.

Open source status

krrishdholakia commented 4 years ago

Commenting for visibility - is this available now ? can't seem to find it, would love to use this for a question-answering project i'm working on!

eugeneware commented 4 years ago

I'd also like to see this model incorporated into the core list of supported models. I did note that you can download it from the community models here though: https://huggingface.co/SpanBERT/spanbert-base-cased

iliemihai commented 4 years ago

Are there any translations of the above repository (https://github.com/mandarjoshi90/coref) into the awesome HuggingFace API ? That would be very cool to test :D !

hlahkar commented 3 years ago

I would like to work on this...but will need some guidance