From 0039b965db7e2d077363583ae95e1487a81ee0af Mon Sep 17 00:00:00 2001 From: Manuel Romero Date: Fri, 10 Jul 2020 17:38:11 +0200 Subject: [PATCH] Create model card (#5655) Create model card for T5-small fine-tuned on SQUAD v2 --- .../t5-small-finetuned-squadv2/README.md | 78 +++++++++++++++++++ 1 file changed, 78 insertions(+) create mode 100644 model_cards/mrm8488/t5-small-finetuned-squadv2/README.md diff --git a/model_cards/mrm8488/t5-small-finetuned-squadv2/README.md b/model_cards/mrm8488/t5-small-finetuned-squadv2/README.md new file mode 100644 index 0000000000..1866d1120f --- /dev/null +++ b/model_cards/mrm8488/t5-small-finetuned-squadv2/README.md @@ -0,0 +1,78 @@ +--- +language: english +datasets: +- squad_v2 +--- + +# T5-small fine-tuned on SQuAD v2 + +[Google's T5](https://ai.googleblog.com/2020/02/exploring-transfer-learning-with-t5.html) [(small)](https://huggingface.co/t5-small) fine-tuned on [SQuAD v2](https://rajpurkar.github.io/SQuAD-explorer/) for **Q&A** downstream task. + +## Details of T5 + +The **T5** model was presented in [Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer](https://arxiv.org/pdf/1910.10683.pdf) by *Colin Raffel, Noam Shazeer, Adam Roberts, Katherine Lee, Sharan Narang, Michael Matena, Yanqi Zhou, Wei Li, Peter J. Liu* in Here the abstract: + +Transfer learning, where a model is first pre-trained on a data-rich task before being fine-tuned on a downstream task, has emerged as a powerful technique in natural language processing (NLP). The effectiveness of transfer learning has given rise to a diversity of approaches, methodology, and practice. In this paper, we explore the landscape of transfer learning techniques for NLP by introducing a unified framework that converts every language problem into a text-to-text format. Our systematic study compares pre-training objectives, architectures, unlabeled datasets, transfer approaches, and other factors on dozens of language understanding tasks. By combining the insights from our exploration with scale and our new โ€œColossal Clean Crawled Corpusโ€, we achieve state-of-the-art results on many benchmarks covering summarization, question answering, text classification, and more. To facilitate future work on transfer learning for NLP, we release our dataset, pre-trained models, and code. + +![model image](https://i.imgur.com/jVFMMWR.png) + + +## Details of the downstream task (Q&A) - Dataset ๐Ÿ“š ๐Ÿง โ“ + +Dataset ID: ```squad_v2``` from [HugginFace/NLP](https://github.com/huggingface/nlp) + +| Dataset | Split | # samples | +| -------- | ----- | --------- | +| squad_v2 | train | 130319 | +| squad_v2 | valid | 11873 | + +How to load it from [nlp](https://github.com/huggingface/nlp) + +```python +train_dataset = nlp.load_dataset('squad_v2, split=nlp.Split.TRAIN) +valid_dataset = nlp.load_dataset('squad_v2', split=nlp.Split.VALIDATION) +``` +Check out more about this dataset and others in [NLP Viewer](https://huggingface.co/nlp/viewer/) + + +## Model fine-tuning ๐Ÿ‹๏ธโ€ + +The training script is a slightly modified version of [this awesome one](https://colab.research.google.com/github/patil-suraj/exploring-T5/blob/master/T5_on_TPU.ipynb) by [Suraj Patil](https://twitter.com/psuraj28) + +## Results ๐Ÿ“ + +| Metric | # Value | +| ------ | --------- | +| **EM** | **69.46** | +| **F1** | **73.01** | + + + +## Model in Action ๐Ÿš€ + +```python +from transformers import AutoModelWithLMHead, AutoTokenizer + +tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-small-finetuned-squadv2") +model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-small-finetuned-squadv2") + +def get_answer(question, context): + input_text = "question: %s context: %s " % (question, context) + features = tokenizer([input_text], return_tensors='pt') + + output = model.generate(input_ids=features['input_ids'], + attention_mask=features['attention_mask']) + + return tokenizer.decode(output[0]) + +context = "Manuel have created RuPERTa-base (a Spanish RoBERTa) with the support of HF-Transformers and Google" +question = "Who has supported Manuel?" + +get_answer(question, context) + +# output: 'HF-Transformers and Google' +``` + +> Created by [Manuel Romero/@mrm8488](https://twitter.com/mrm8488) | [LinkedIn](https://www.linkedin.com/in/manuel-romero-cs/) + +> Made with in Spain