bjascob / amrlib

A python library that makes AMR parsing, generation and visualization simple.
MIT License
216 stars 33 forks source link

Consider Adding a New Parse Model #31

Closed bjascob closed 2 years ago

bjascob commented 2 years ago

The amrlib/parse_T5 model scores 81 on AMR-3. There are two publicly available models that have slightly better performance..

Both of these models are based on BART-large which has roughly 2X the parameters of the T5-base model used in amrlib. This may cause issues training on older/smaller GPUs and could be slower for inference.

bjascob commented 2 years ago

Added model parse_spring based on the SapienzaNLP/SPRING code. This model achieves an 83.5 smatch, pre-wikification.

Revised the parse_t5 code to train and infer significantly faster and retrained the model to 16 epochs. This pushes up the smatch score to 81.9 pre-wikification. The new model is model_parse_t5-v0_2_0. The parse_t5 code is compatible with both the new and old models.