英文

lmqg/mt5-small-dequad-qg-ae模型卡片

此模型是 google/mt5-small 的精细调整版本,通过 lmqg lmqg/qg_dequad (数据集名称:default)上同时进行问题生成和答案提取。

概述

用法

from lmqg import TransformersQG

# initialize model
model = TransformersQG(language="de", model="lmqg/mt5-small-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).")
  • 使用transformers
from transformers import pipeline

pipe = pipeline("text2text-generation", "lmqg/mt5-small-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.")

评估

Score Type Dataset
BERTScore 80.39 default 12313321
Bleu_1 10.13 default 12313321
Bleu_2 4.24 default 12313321
Bleu_3 1.89 default 12313321
Bleu_4 0.82 default 12313321
METEOR 12.18 default 12313321
MoverScore 55.1 default 12313321
ROUGE_L 10.15 default 12313321
Score Type Dataset
QAAlignedF1Score (BERTScore) 80.02 default 12313321
QAAlignedF1Score (MoverScore) 53.99 default 12313321
QAAlignedPrecision (BERTScore) 78.91 default 12313321
QAAlignedPrecision (MoverScore) 53.77 default 12313321
QAAlignedRecall (BERTScore) 81.23 default 12313321
QAAlignedRecall (MoverScore) 54.27 default 12313321
Score Type Dataset
AnswerExactMatch 9.94 default 12313321
AnswerF1Score 37.66 default 12313321
BERTScore 75.15 default 12313321
Bleu_1 19.77 default 12313321
Bleu_2 13.28 default 12313321
Bleu_3 8.84 default 12313321
Bleu_4 5.83 default 12313321
METEOR 22.47 default 12313321
MoverScore 57.76 default 12313321
ROUGE_L 18.26 default 12313321

训练超参数

在精细调整期间使用了以下超参数:

  • 数据集路径:lmqg/qg_dequad
  • 数据集名称:default
  • 输入类型:['paragraph_answer','paragraph_sentence']
  • 输出类型:['question','answer']
  • 前缀类型:['qg','ae']
  • 模型:google/mt5-small
  • 最大长度:512
  • 最大输出长度:32
  • 时代:15
  • 批次:16
  • lr:0.001
  • fp16:False
  • 随机种子:1
  • 梯度累积步数:4
  • 标签平滑:0.15

完整的配置可在 fine-tuning config file 找到。

引用

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