Edit model card

Model Card of lmqg/mt5-base-dequad-qg-ae

This model is fine-tuned version of google/mt5-base for question generation and answer extraction jointly on the lmqg/qg_dequad (dataset_name: default) via lmqg.

Overview

Usage

from lmqg import TransformersQG

# initialize model
model = TransformersQG(language="de", model="lmqg/mt5-base-dequad-qg-ae")

# model prediction
question_answer_pairs = model.generate_qa("das erste weltweit errichtete Hermann Brehmer 1855 im niederschlesischen ''Görbersdorf'' (heute Sokołowsko, Polen).")
  • With transformers
from transformers import pipeline

pipe = pipeline("text2text-generation", "lmqg/mt5-base-dequad-qg-ae")

# answer extraction
answer = pipe("generate question: Empfangs- und Sendeantenne sollen in ihrer Polarisation übereinstimmen, andernfalls <hl> wird die Signalübertragung stark gedämpft. <hl>")

# question generation
question = pipe("extract answers: Sommerzeit <hl> Frühling <hl>: Umstellung von Normalzeit auf Sommerzeit – die Uhr wird um eine Stunde ''vor''gestellt. Herbst: Umstellung von Sommerzeit auf Normalzeit – die Uhr wird um eine Stunde ''zurück''gestellt. Als Sommerzeit wird die gegenüber der Zonenzeit meist um eine Stunde vorgestellte Uhrzeit bezeichnet, die während eines bestimmten Zeitraums im Sommerhalbjahr (und oft auch etwas darüber hinaus) als gesetzliche Zeit dient. Eine solche Regelung wird fast nur in Ländern der gemäßigten Zonen angewandt. Die mitteleuropäische Sommerzeit beginnt am letzten Sonntag im März um 2:00 Uhr MEZ, indem die Stundenzählung um eine Stunde von 2:00 Uhr auf 3:00 Uhr vorgestellt wird. Sie endet jeweils am letzten Sonntag im Oktober um 3:00 Uhr MESZ, indem die Stundenzählung um eine Stunde von 3:00 Uhr auf 2:00 Uhr zurückgestellt wird.")

Evaluation

Score Type Dataset
BERTScore 77.86 default lmqg/qg_dequad
Bleu_1 8.19 default lmqg/qg_dequad
Bleu_2 3.17 default lmqg/qg_dequad
Bleu_3 1.12 default lmqg/qg_dequad
Bleu_4 0.38 default lmqg/qg_dequad
METEOR 10.56 default lmqg/qg_dequad
MoverScore 53.77 default lmqg/qg_dequad
ROUGE_L 8.58 default lmqg/qg_dequad
Score Type Dataset
QAAlignedF1Score (BERTScore) 6.11 default lmqg/qg_dequad
QAAlignedF1Score (MoverScore) 4.24 default lmqg/qg_dequad
QAAlignedPrecision (BERTScore) 6.3 default lmqg/qg_dequad
QAAlignedPrecision (MoverScore) 4.34 default lmqg/qg_dequad
QAAlignedRecall (BERTScore) 5.95 default lmqg/qg_dequad
QAAlignedRecall (MoverScore) 4.15 default lmqg/qg_dequad
Score Type Dataset
AnswerExactMatch 0.32 default lmqg/qg_dequad
AnswerF1Score 8.63 default lmqg/qg_dequad
BERTScore 63.01 default lmqg/qg_dequad
Bleu_1 5.06 default lmqg/qg_dequad
Bleu_2 2.4 default lmqg/qg_dequad
Bleu_3 1.35 default lmqg/qg_dequad
Bleu_4 0.85 default lmqg/qg_dequad
METEOR 5.34 default lmqg/qg_dequad
MoverScore 47.54 default lmqg/qg_dequad
ROUGE_L 3.93 default lmqg/qg_dequad

Training hyperparameters

The following hyperparameters were used during fine-tuning:

  • dataset_path: lmqg/qg_dequad
  • dataset_name: default
  • input_types: ['paragraph_answer', 'paragraph_sentence']
  • output_types: ['question', 'answer']
  • prefix_types: ['qg', 'ae']
  • model: google/mt5-base
  • max_length: 512
  • max_length_output: 32
  • epoch: 8
  • batch: 32
  • lr: 0.0001
  • 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",
}
Downloads last month
3

Dataset used to train lmqg/mt5-base-dequad-qg-ae

Evaluation results

  • BLEU4 (Question Generation) on lmqg/qg_dequad
    self-reported
    0.380
  • ROUGE-L (Question Generation) on lmqg/qg_dequad
    self-reported
    8.580
  • METEOR (Question Generation) on lmqg/qg_dequad
    self-reported
    10.560
  • BERTScore (Question Generation) on lmqg/qg_dequad
    self-reported
    77.860
  • MoverScore (Question Generation) on lmqg/qg_dequad
    self-reported
    53.770
  • QAAlignedF1Score-BERTScore (Question & Answer Generation (with Gold Answer)) on lmqg/qg_dequad
    self-reported
    6.110
  • QAAlignedRecall-BERTScore (Question & Answer Generation (with Gold Answer)) on lmqg/qg_dequad
    self-reported
    5.950
  • QAAlignedPrecision-BERTScore (Question & Answer Generation (with Gold Answer)) on lmqg/qg_dequad
    self-reported
    6.300
  • QAAlignedF1Score-MoverScore (Question & Answer Generation (with Gold Answer)) on lmqg/qg_dequad
    self-reported
    4.240
  • QAAlignedRecall-MoverScore (Question & Answer Generation (with Gold Answer)) on lmqg/qg_dequad
    self-reported
    4.150
  • QAAlignedPrecision-MoverScore (Question & Answer Generation (with Gold Answer)) on lmqg/qg_dequad
    self-reported
    4.340
  • BLEU4 (Answer Extraction) on lmqg/qg_dequad
    self-reported
    0.850
  • ROUGE-L (Answer Extraction) on lmqg/qg_dequad
    self-reported
    3.930
  • METEOR (Answer Extraction) on lmqg/qg_dequad
    self-reported
    5.340
  • BERTScore (Answer Extraction) on lmqg/qg_dequad
    self-reported
    63.010
  • MoverScore (Answer Extraction) on lmqg/qg_dequad
    self-reported
    47.540
  • AnswerF1Score (Answer Extraction) on lmqg/qg_dequad
    self-reported
    8.630
  • AnswerExactMatch (Answer Extraction) on lmqg/qg_dequad
    self-reported
    0.320