Model Card of vocabtrimmer/mt5-small-trimmed-en-90000-squad-qa

This model is fine-tuned version of ckpts/mt5-small-trimmed-en-90000 for question answering task on the lmqg/qg_squad (dataset_name: default) via lmqg.

Overview

• Language model: ckpts/mt5-small-trimmed-en-90000
• Language: en
• Training data: lmqg/qg_squad (default)
• Online Demo: https://autoqg.net/
• Repository: https://github.com/asahi417/lm-question-generation
• Paper: https://arxiv.org/abs/2210.03992

Usage

• With lmqg

from lmqg import TransformersQG

# initialize model
model = TransformersQG(language="en", model="vocabtrimmer/mt5-small-trimmed-en-90000-squad-qa")

# model prediction
answers = model.answer_q(list_question="What is a person called is practicing heresy?", list_context=" Heresy is any provocative belief or theory that is strongly at variance with established beliefs or customs. A heretic is a proponent of such claims or beliefs. Heresy is distinct from both apostasy, which is the explicit renunciation of one's religion, principles or cause, and blasphemy, which is an impious utterance or action concerning God or sacred things.")

• With transformers

from transformers import pipeline

pipe = pipeline("text2text-generation", "vocabtrimmer/mt5-small-trimmed-en-90000-squad-qa")
output = pipe("question: What is a person called is practicing heresy?, context: Heresy is any provocative belief or theory that is strongly at variance with established beliefs or customs. A heretic is a proponent of such claims or beliefs. Heresy is distinct from both apostasy, which is the explicit renunciation of one's religion, principles or cause, and blasphemy, which is an impious utterance or action concerning God or sacred things.")

Evaluation

• Metric (Question Answering): raw metric file

	Score	Type	Dataset
AnswerExactMatch	54.26	default	lmqg/qg_squad
AnswerF1Score	68.65	default	lmqg/qg_squad
BERTScore	91.86	default	lmqg/qg_squad
Bleu_1	49.27	default	lmqg/qg_squad
Bleu_2	43.25	default	lmqg/qg_squad
Bleu_3	37.89	default	lmqg/qg_squad
Bleu_4	33.47	default	lmqg/qg_squad
METEOR	39.13	default	lmqg/qg_squad
MoverScore	81.36	default	lmqg/qg_squad
ROUGE_L	67.38	default	lmqg/qg_squad

Training hyperparameters

The following hyperparameters were used during fine-tuning:

• dataset_path: lmqg/qg_squad
• dataset_name: default
• input_types: ['paragraph_question']
• output_types: ['answer']
• prefix_types: None
• model: ckpts/mt5-small-trimmed-en-90000
• max_length: 512
• max_length_output: 32
• epoch: 10
• batch: 32
• lr: 0.0005
• fp16: False
• random_seed: 1
• gradient_accumulation_steps: 2
• label_smoothing: 0.15

The full configuration can be found at fine-tuning config file.

Citation

@inproceedings{ushio-etal-2022-generative,
    title = "{G}enerative {L}anguage {M}odels for {P}aragraph-{L}evel {Q}uestion {G}eneration",
    author = "Ushio, Asahi  and
        Alva-Manchego, Fernando  and
        Camacho-Collados, Jose",
    booktitle = "Proceedings of the 2022 Conference on Empirical Methods in Natural Language Processing",
    month = dec,
    year = "2022",
    address = "Abu Dhabi, U.A.E.",
    publisher = "Association for Computational Linguistics",
}