英文

lmqg/mt5-small-frquad-qg模型卡片

该模型是 google/mt5-small 的问题生成任务的微调版本,使用 lmqg lmqg/qg_frquad (dataset_name: default)进行训练。

概述

用法

from lmqg import TransformersQG

# initialize model
model = TransformersQG(language="fr", model="lmqg/mt5-small-frquad-qg")

# model prediction
questions = model.generate_q(list_context="Créateur » (Maker), lui aussi au singulier, « le Suprême Berger » (The Great Shepherd) ; de l'autre, des réminiscences de la théologie de l'Antiquité : le tonnerre, voix de Jupiter, « Et souvent ta voix gronde en un tonnerre terrifiant », etc.", list_answer="le Suprême Berger")
  • 使用 transformers
from transformers import pipeline

pipe = pipeline("text2text-generation", "lmqg/mt5-small-frquad-qg")
output = pipe("Créateur » (Maker), lui aussi au singulier, « <hl> le Suprême Berger <hl> » (The Great Shepherd) ; de l'autre, des réminiscences de la théologie de l'Antiquité : le tonnerre, voix de Jupiter, « Et souvent ta voix gronde en un tonnerre terrifiant », etc.")

评估

Score Type Dataset
BERTScore 80.71 default 12313321
Bleu_1 29.26 default 12313321
Bleu_2 17.56 default 12313321
Bleu_3 12.03 default 12313321
Bleu_4 8.55 default 12313321
METEOR 17.51 default 12313321
MoverScore 56.5 default 12313321
ROUGE_L 28.56 default 12313321
  • 指标(问题和答案生成,参考答案):每个问题都是基于黄金答案生成的。 raw metric file
Score Type Dataset
QAAlignedF1Score (BERTScore) 88.52 default 12313321
QAAlignedF1Score (MoverScore) 62.46 default 12313321
QAAlignedPrecision (BERTScore) 88.53 default 12313321
QAAlignedPrecision (MoverScore) 62.46 default 12313321
QAAlignedRecall (BERTScore) 88.51 default 12313321
QAAlignedRecall (MoverScore) 62.45 default 12313321
Score Type Dataset
QAAlignedF1Score (BERTScore) 79.72 default 12313321
QAAlignedF1Score (MoverScore) 53.94 default 12313321
QAAlignedPrecision (BERTScore) 77.58 default 12313321
QAAlignedPrecision (MoverScore) 52.7 default 12313321
QAAlignedRecall (BERTScore) 82.06 default 12313321
QAAlignedRecall (MoverScore) 55.32 default 12313321

训练超参数

微调过程中使用了以下超参数:

  • dataset_path:lmqg/qg_frquad
  • dataset_name:默认
  • input_types: ['paragraph_answer']
  • output_types: ['question']
  • prefix_types: 无
  • model:google/mt5-small
  • max_length:512
  • max_length_output:32
  • epoch:14
  • batch:64
  • lr:0.001
  • fp16:False
  • random_seed:1
  • gradient_accumulation_steps:1
  • label_smoothing: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",
}