NLP: Datasets
Overview of NLP Tasks
The ultimate goal of NLP is free communication with an intelligent system in natural language (dialogue, obtaining information, setting tasks, etc.). For testing various models, formal specific tasks are chosen, which can be divided into three large groups: classification tasks, labeling tasks, and generation tasks.Classification Tasks
- Sentiment Analysis (determining the sentiment of the text) - usually assigning a given text to one of two classes (positive or negative). It is practically used in the analysis of reviews. This type of task also includes spam detection, news fakeness, etc.
- Next Sentence Prediction (predicting the next sentence) - the system receives a pair of sentences and must output the degree of confidence that the second sentence naturally can follow the first (in some text).
- Natural Language Inference (NLI, inference in natural language) - consists of text and a sentence that either follows (entailment), or contradicts (contradiction), or is neutral (neutral) to the text, see here or here. Similar to the previous one, but usually implies a certain logical connection between premise and conclusion.
Labeling Tasks
- Part of Speech Recognizing (part-of-speech tagging) - each token of the text must be assigned to one or another part of speech (noun, adjective, verb, etc.).
- Named-Entity Recognizing (extracting entities from text) assigning to a group of tokens a label whether it is a person's name, organization, geographical name, etc. (see, for example, here).
- Question Answering (answer extraction) - a text and a question to it are given. It is necessary to indicate a group of tokens in the text that are the answer to this question.
Generation Tasks
- Automatic summarization (automatic abstracting) - creation of a short version of the given text. This also includes the task of simplifying the text (without shortening it).
- Machine translation (machine translation) - translation of text from one natural language to another.
1 Billion Word Language Model Benchmark
The English-language text corpus is described in the article Chelba C., et.al (2014). Its vocabulary consists of 793471 words and has a length of 0.8 billion tokens. Words outside the vocabulary are marked (?) as <UNK> (there are 0.28% of them). The texts are divided into sentences, in which tokenization has been performed. Sentences are shuffled:
The journalist 's job is to help inform the public on issues that matter . The golfer , still the highest-paid athlete in the world , was second last year . Anderson scored 12 points and Amanda Nisleit added 11 points and 12 rebounds .
For downloading, the corpus is available at this link or at this and occupies 1.7 Gb in the archive. It is divided into files.
ROC Story Cloze
ROCStories are coherent simple stories of four sentences. In the test set, two sentences are added to each story, one of which is a meaningful continuation of the story, while the second is not. For example:
Karen was assigned a roommate her first year of college.
Her roommate asked her to go to a nearby city for a concert.
Karen agreed happily. The show was absolutely exhilarating.
- Right Ending: Karen became good friends with her roommate.
- Wrong Ending: Karen hated her roommate.
There is a file
100KStories.csv
with 98'167 stories, totaling 4'859'629 tokens
and 36'566 unique word forms.
- NN_Embedding and NN_Embedding_ROCStories.
- Mostafazadeh N., et al. A Corpus and Evaluation Framework for Deeper Understanding of Commonsense Stories (2016) - announcement of the creation of the dataset using AMT workers.
- Schwartz R., et al. The Effect of Different Writing Tasks on Linguistic Style: A Case Study of the ROC Story Cloze Task (2017) They claim that since 1) the story, 2) its continuation, and 3) the wrong continuation were written by different people, a simple linear classifier can distinguish these three cases.
MultiNLI
MultiNLI - Natural Language Inference corpus with 433k sentence pairsStanford Natural Language Inference (SNLI)
SNLI Corpus consists of 570k pairs of English sentences in which the second sentence either follows from the first (entailment), or contradicts (contradiction) or is neutral (neutral).Text : A man inspects the uniform of a figure in some East Asian country. Judgments : contradiction CCCCC Hypothesis: The man is sleeping Text : A soccer game with multiple males playing. Judgments : entailment EEEEE Hypothesis: Some men are playing a sport. Text : An older and younger man smiling. Judgments : neutral NNENN Hypothesis: Two men are smiling and laughing at the cats playing on the floor.
Stanford Question Answering Dataset (SQuAD)
SQuAD - a dataset for reading comprehension, consisting of questions posed by people on a set of Wikipedia articles, where the answer to each question is a fragment of text or a span from the corresponding reading passage. The question may also be unanswerable.
"data": [
{"title": "Beyonc\u00e9",
"paragraphs": [{"qas": [
{"question": "When did Beyonce start becoming popular?",
"answers": [{"text": "in the late 1990s", "answer_start": 269}],
"is_impossible": false},
{"question": "What areas did Beyonce compete in when she was growing up?",
"answers": [{"text": "singing and dancing", "answer_start": 207}],
"is_impossible": false},
{"question": "In what city and state did Beyonce grow up? ",
"answers": [{"text": "Houston, Texas", "answer_start": 166}],
"is_impossible": false},
Stanford Sentiment Analysis
Sentiment Treebank for predicting the sentiment of movie reviews. Data can be downloaded here.hide new secretions from the parental units 0 contains no wit , only labored gags 0 that loves its characters and communicates something rather beautiful about human nature 1
General Language Understanding Evaluation (GLUE)
This benchmark is a collection of eleven language understanding tasks: The Corpus of Linguistic Acceptability, The Stanford Sentiment Treebank, Microsoft Research Paraphrase Corpus, Semantic Textual Similarity Benchmark, Quora Question Pairs, MultiNLI Matched, MultiNLI Mismatched, Question NLI, Recognizing Textual Entailment, Winograd NLI, Diagnostics Main.
The download page contains text data separately for each task.
CommonsenseQA
CommonsenseQA is a new multiple-choice question answering dataset that requires different types of commonsense knowledge to predict the correct answers . It contains 12,102 questions with one correct answer and four distractor answers. The dataset is provided in two major training/validation/testing set splits: "Random split" which is the main evaluation split, and "Question token split", see paper for details.
- Where would I not want a fox?
👍 hen house, 👎 england, 👎 mountains, 👎 english hunt, 👎 california - Why do people read gossip magazines?
👍 entertained, 👎 get information, 👎 learn, 👎 improve know how, 👎 lawyer told to - What do all humans want to experience in their own home?
👍 feel comfortable, 👎 work hard, 👎 fall in love, 👎 lay eggs, 👎 live forever
Literature
- Talmor A., et al. CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledge They claim that: "Our best baseline is based on BERT-large (Devlin et al., 2018) and obtains 56% accuracy, well below human performance, which is 89%."