mt5-small-trimmed-en-squad-qa

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

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

Overview

• Language model: ckpts/mt5-small-trimmed-en
• 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-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-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

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
• max_length: 512
• max_length_output: 32
• epoch: 8
• batch: 32
• lr: 0.001
• fp16: False
• random_seed: 1
• gradient_accumulation_steps: 2
• label_smoothing: 0.0

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",
}