Cs224n Squad2.0 Distilbert Base Uncased

🚀 CS224n SQuAD2.0項目數據集

本模型旨在為CS224n的學生在為默認最終項目建立待超越的基線時節省GPU時間。用於微調此模型的訓練集與官方訓練集相同；不過，評估和模型選擇使用的是官方開發集的大約一半，即隨機挑選的6078個示例。數據文件可在https://github.com/elgeish/squad/tree/master/data找到 —— 這是2020年冬季版本。鑑於官方SQuAD2.0開發集包含該項目的測試集，學生必須確保不以任何方式使用官方SQuAD2.0開發集，包括不使用在官方SQuAD2.0上微調的模型，因為這些模型在選擇時使用了官方SQuAD2.0開發集。

📊 結果

{
  "exact": 65.16946363935504,
  "f1": 67.87348075352251,
  "total": 6078,
  "HasAns_exact": 69.51890034364261,
  "HasAns_f1": 75.16667217179045,
  "HasAns_total": 2910,
  "NoAns_exact": 61.17424242424242,
  "NoAns_f1": 61.17424242424242,
  "NoAns_total": 3168,
  "best_exact": 65.16946363935504,
  "best_exact_thresh": 0.0,
  "best_f1": 67.87348075352243,
  "best_f1_thresh": 0.0
}

⚙️ 重要參數

{
  "do_lower_case": true,
  "doc_stride": 128,
  "fp16": false,
  "fp16_opt_level": "O1",
  "gradient_accumulation_steps": 24,
  "learning_rate": 3e-05,
  "max_answer_length": 30,
  "max_grad_norm": 1,
  "max_query_length": 64,
  "max_seq_length": 384,
  "model_name_or_path": "distilbert-base-uncased-distilled-squad",
  "model_type": "distilbert",
  "num_train_epochs": 4,
  "per_gpu_train_batch_size": 32,
  "save_steps": 5000,
  "seed": 42,
  "train_batch_size": 32,
  "version_2_with_negative": true,
  "warmup_steps": 0,
  "weight_decay": 0
}

🛠️ 環境設置

{
  "transformers": "2.5.1",
  "pytorch": "1.4.0=py3.6_cuda10.1.243_cudnn7.6.3_0",
  "python": "3.6.5=hc3d631a_2",
  "os": "Linux 4.15.0-1060-aws #62-Ubuntu SMP Tue Feb 11 21:23:22 UTC 2020 x86_64 x86_64 x86_64 GNU/Linux",
  "gpu": "Tesla V100-SXM2-16GB"
}

📚 如何引用

@misc{elgeish2020gestalt,
  title={Gestalt: a Stacking Ensemble for SQuAD2.0},
  author={Mohamed El-Geish},
  journal={arXiv e-prints},
  archivePrefix={arXiv},
  eprint={2004.07067},
  year={2020},
}

🧩 相關模型

• elgeish/cs224n-squad2.0-albert-base-v2
• elgeish/cs224n-squad2.0-albert-large-v2
• elgeish/cs224n-squad2.0-albert-xxlarge-v1
• elgeish/cs224n-squad2.0-roberta-base