英文

lmqg/mt5-small-jaquad-qg的模型卡片

这个模型是基于 google/mt5-small 在问答生成任务上经过微调的版本,使用 lmqg lmqg/qg_jaquad (数据集名称:default)进行训练。

概述

用法

from lmqg import TransformersQG

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

# model prediction
questions = model.generate_q(list_context="フェルメールの作品では、17世紀のオランダの画家、ヨハネス・フェルメールの作品について記述する。フェルメールの作品は、疑問作も含め30数点しか現存しない。現存作品はすべて油彩画で、版画、下絵、素描などは残っていない。", list_answer="30数点")
  • 与transformers一起使用
from transformers import pipeline

pipe = pipeline("text2text-generation", "lmqg/mt5-small-jaquad-qg")
output = pipe("ゾフィーは貴族出身ではあったが王族出身ではなく、ハプスブルク家の皇位継承者であるフランツ・フェルディナントとの結婚は貴賤結婚となった。皇帝フランツ・ヨーゼフは、2人の間に生まれた子孫が皇位を継がないことを条件として結婚を承認していた。視察が予定されている<hl>6月28日<hl>は2人の14回目の結婚記念日であった。")

评估

Score Type Dataset
BERTScore 80.87 default 12313321
Bleu_1 56.34 default 12313321
Bleu_2 44.28 default 12313321
Bleu_3 36.31 default 12313321
Bleu_4 30.49 default 12313321
METEOR 29.03 default 12313321
MoverScore 58.67 default 12313321
ROUGE_L 50.88 default 12313321
  • 指标(问题和答案生成,参考答案):每个问题是从金标准答案生成的
Score Type Dataset
QAAlignedF1Score (BERTScore) 86.07 default 12313321
QAAlignedF1Score (MoverScore) 61.83 default 12313321
QAAlignedPrecision (BERTScore) 86.08 default 12313321
QAAlignedPrecision (MoverScore) 61.85 default 12313321
QAAlignedRecall (BERTScore) 86.06 default 12313321
QAAlignedRecall (MoverScore) 61.81 default 12313321
Score Type Dataset
QAAlignedF1Score (BERTScore) 79.78 default 12313321
QAAlignedF1Score (MoverScore) 55.85 default 12313321
QAAlignedPrecision (BERTScore) 76.84 default 12313321
QAAlignedPrecision (MoverScore) 53.8 default 12313321
QAAlignedRecall (BERTScore) 83.06 default 12313321
QAAlignedRecall (MoverScore) 58.22 default 12313321

训练超参数

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

  • dataset_path: lmqg/qg_jaquad
  • dataset_name: default
  • input_types: ['paragraph_answer']
  • output_types: ['question']
  • prefix_types: None
  • model: google/mt5-small
  • max_length: 512
  • max_length_output: 32
  • epoch: 21
  • batch: 64
  • lr: 0.0005
  • fp16: False
  • random_seed: 1
  • gradient_accumulation_steps: 1
  • label_smoothing: 0.0

完整的配置可以在 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",
}