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

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

Overview

• Language model: ckpts/mt5-small-trimmed-en-5000
• 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-5000-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-5000-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	61.29	default	lmqg/qg_squad
AnswerF1Score	74.01	default	lmqg/qg_squad
BERTScore	92.31	default	lmqg/qg_squad
Bleu_1	59.87	default	lmqg/qg_squad
Bleu_2	54.66	default	lmqg/qg_squad
Bleu_3	49.7	default	lmqg/qg_squad
Bleu_4	45.47	default	lmqg/qg_squad
METEOR	43.2	default	lmqg/qg_squad
MoverScore	83.24	default	lmqg/qg_squad
ROUGE_L	72.24	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-5000
• max_length: 512
• max_length_output: 32
• epoch: 13
• 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",
}