RTIInternational / gobbli

Deep learning with text doesn't have to be scary.
Apache License 2.0
275 stars 23 forks source link

adds ncbi bert models #12

Closed pmbaumgartner closed 5 years ago

pmbaumgartner commented 5 years ago

Description of Changes

Adds the NCBI BERT models from https://github.com/ncbi-nlp/NCBI_BERT.

Paper: https://arxiv.org/abs/1906.05474

Some important notes for application: there was wide variability around which model performed best depending on the task. Counterintuitively, the base variants worked better for some tasks over the large variants. Additionally, sometimes the model that was also trained on the MIMIC dataset performed worse where they expected better performance.

Summary of performance: image

image

Related Issue(s), if any