• Stars
    star
    180
  • Rank 213,097 (Top 5 %)
  • Language
  • Created about 5 years ago
  • Updated about 3 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

XQuAD

XQuAD (Cross-lingual Question Answering Dataset) is a benchmark dataset for evaluating cross-lingual question answering performance. The dataset consists of a subset of 240 paragraphs and 1190 question-answer pairs from the development set of SQuAD v1.1 (Rajpurkar et al., 2016) together with their professional translations into ten languages: Spanish, German, Greek, Russian, Turkish, Arabic, Vietnamese, Thai, Chinese, and Hindi. Consequently, the dataset is entirely parallel across 11 languages.

For more information on how the dataset was created, refer to our paper, On the Cross-lingual Transferability of Monolingual Representations.

All files are in json format following the SQuAD dataset format. A parallel example in XQuAD in English, Spanish, and Chinese can be seen in the image below. The full dataset consists of 240 such parallel instances in 11 languages.

Update: Added SQuAD v1.1 professionally translated to Romanian.

An example from XQuAD

Data

This directory contains files in the following languages:

  • Arabic: xquad.ar.json
  • German: xquad.de.json
  • Greek: xquad.el.json
  • English: xquad.en.json
  • Spanish: xquad.es.json
  • Hindi: xquad.hi.json
  • Russian: xquad.ru.json
  • Thai: xquad.th.json
  • Turkish: xquad.tr.json
  • Vietnamese: xquad.vi.json
  • Chinese: xquad.zh.json
  • Romanian: xquad.ro.json (newly added; not included in the original XQuAD)

As the dataset is based on SQuAD v1.1, there are no unanswerable questions in the data. We chose this setting so that models can focus on cross-lingual transfer.

We show the average number of tokens per paragraph, question, and answer for each language in the table below. The statistics were obtained using Jieba for Chinese and the Moses tokenizer for the other languages.

en es de el ru tr ar vi th zh hi
Paragraph 142.4 160.7 139.5 149.6 133.9 126.5 128.2 191.2 158.7 147.6 232.4
Question 11.5 13.4 11.0 11.7 10.0 9.8 10.7 14.8 11.5 10.5 18.7
Answer 3.1 3.6 3.0 3.3 3.1 3.1 3.1 4.5 4.1 3.5 5.6

Training and evaluation

In order to evaluate on XQuAD, models should be trained on the SQuAD v1.1 training file. which can be downloaded from here. Model validation similarly should be conducted on the SQuAD v1.1 validation file.

For evaluation, we use the official SQuAD evaluate-v1.1.py script, which can be obtained from here. Note that the SQuAD evaluation script normalises the answer based on heuristics that are specific to English. We have observed language-specific normalisation heuristics only to have a marginal impact on performance, which is why we use the English SQuAD v1.1 evaluation script for convenience.

Baselines

We show results using baseline methods in the tables below. We directly fine-tune mBERT and XLM-R Large on the English SQuAD v1.1 training data and evaluate them via zero-shot transfer on the XQuAD test datasets. For translate-train, we fine-tune mBERT on the SQuAD v1.1 training data, which we automatically translate to the target language. For translate-test, we fine-tune BERT-Large on the SQuAD v1.1 training set and evaluate it on the XQuAD test set of the target language, which we automatically translate to English. Note that results with translate-test are not directly comparable as we drop a small number (less than 3%) of the test examples.

F1 scores:

Model en ar de el es hi ru th tr vi zh ro avg
mBERT 83.5 61.5 70.6 62.6 75.5 59.2 71.3 42.7 55.4 69.5 58.0 72.7 65.2
XLM-R Large 86.5 68.6 80.4 79.8 82.0 76.7 80.1 74.2 75.9 79.1 59.3 83.6 77.2
Translate-train mBERT 83.5 68.0 75.6 70.0 80.2 69.6 75.0 36.9 68.9 75.6 66.2 - 70.0
Translate-test BERT-L 87.9 73.7 79.8 79.4 82.0 74.9 79.9 64.6 67.4 76.3 73.7 - 76.3

EM scores:

Model en ar de el es hi ru th tr vi zh ro avg
mBERT 72.2 45.1 54.0 44.9 56.9 46.0 53.3 33.5 40.1 49.6 48.3 59.9 50.3
XLM-R Large 75.7 49.0 63.4 61.7 63.9 59.7 64.3 62.8 59.3 59.0 50.0 69.7 61.5
Translate-train mBERT 72.2 51.1 60.7 53.0 63.1 55.4 59.7 33.5 54.8 56.2 56.6 - 56.0
Translate-test BERT-L 77.1 58.8 66.7 65.5 68.4 60.1 66.7 50.0 49.6 61.5 59.1 - 62.1

Best practices

XQuAD is intended as an evaluation corpus for zero-shot cross-lingual transfer. Evaluation on the test data should ideally only be conducted at the very end of the experimentation in order to avoid overfitting to the data.

If you are evaluating on XQuAD in the zero-shot setting, please state explicitly your experimental settings, particularly what monolingual and cross-lingual data you used for pre-training and fine-tuning your model.

Reference

If you use this dataset, please cite [1]:

[1] Artetxe, M., Ruder, S., & Yogatama, D. (2019). On the cross-lingual transferability of monolingual representations. arXiv preprint arXiv:1910.11856.

@article{Artetxe:etal:2019,
      author    = {Mikel Artetxe and Sebastian Ruder and Dani Yogatama},
      title     = {On the cross-lingual transferability of monolingual representations},
      journal   = {CoRR},
      volume    = {abs/1910.11856},
      year      = {2019},
      archivePrefix = {arXiv},
      eprint    = {1910.11856}
}

The Romanian version of this data is part of LiRo, a benchmark for Romanian natural language understanding tasks:

@inproceedings{
      dumitrescu2021liro,
      title={LiRo: Benchmark and leaderboard for Romanian language tasks},
      author={Stefan Daniel Dumitrescu and Petru Rebeja and Beata Lorincz and Mihaela Gaman and Andrei Avram and Mihai Ilie and Andrei Pruteanu and Adriana Stan and Lorena Rosia and Cristina Iacobescu and Luciana Morogan and George Dima and Gabriel Marchidan and Traian Rebedea and Madalina Chitez and Dani Yogatama and Sebastian Ruder and Radu Tudor Ionescu and Razvan Pascanu and Viorica Patraucean},
      booktitle={Thirty-fifth Conference on Neural Information Processing Systems Datasets and Benchmarks Track (Round 1)},
      year={2021},
      url={https://openreview.net/forum?id=JH61CD7afTv}
}

License

This dataset is distributed under the CC BY-SA 4.0 license.

This is not an officially supported Google product.

More Repositories

1

deepmind-research

This repository contains implementations and illustrative code to accompany DeepMind publications
Jupyter Notebook
13,132
star
2

alphafold

Open source code for AlphaFold.
Python
12,602
star
3

sonnet

TensorFlow-based neural network library
Python
9,769
star
4

mujoco

Multi-Joint dynamics with Contact. A general purpose physics simulator.
Jupyter Notebook
8,113
star
5

pysc2

StarCraft II Learning Environment
Python
8,001
star
6

lab

A customisable 3D platform for agent-based AI research
C
7,101
star
7

graph_nets

Build Graph Nets in Tensorflow
Python
5,352
star
8

graphcast

Python
4,517
star
9

open_spiel

OpenSpiel is a collection of environments and algorithms for research in general reinforcement learning and search/planning in games.
C++
4,231
star
10

alphageometry

Python
4,079
star
11

learning-to-learn

Learning to Learn in TensorFlow
Python
4,064
star
12

dm_control

Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.
Python
3,793
star
13

acme

A library of reinforcement learning components and agents
Python
3,466
star
14

trfl

TensorFlow Reinforcement Learning
Python
3,136
star
15

dm-haiku

JAX-based neural network library
Python
2,848
star
16

alphatensor

Python
2,670
star
17

dnc

A TensorFlow implementation of the Differentiable Neural Computer.
Python
2,478
star
18

gemma

Open weights LLM from Google DeepMind.
Python
2,421
star
19

mctx

Monte Carlo tree search in JAX
Python
2,313
star
20

code_contests

C++
2,064
star
21

optax

Optax is a gradient processing and optimization library for JAX.
Python
1,670
star
22

kinetics-i3d

Convolutional neural network model for video classification trained on the Kinetics dataset.
Python
1,639
star
23

penzai

A JAX research toolkit for building, editing, and visualizing neural networks.
Python
1,639
star
24

mathematics_dataset

This dataset code generates mathematical question and answer pairs, from a range of question types at roughly school-level difficulty.
Python
1,621
star
25

bsuite

bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agent
Python
1,497
star
26

educational

Jupyter Notebook
1,398
star
27

jraph

A Graph Neural Network Library in Jax
Python
1,349
star
28

rc-data

Question answering dataset featured in "Teaching Machines to Read and Comprehend
Python
1,285
star
29

mujoco_menagerie

A collection of high-quality models for the MuJoCo physics engine, curated by Google DeepMind.
Jupyter Notebook
1,278
star
30

tapnet

Tracking Any Point (TAP)
Jupyter Notebook
1,266
star
31

rlax

Python
1,223
star
32

scalable_agent

A TensorFlow implementation of Scalable Distributed Deep-RL with Importance Weighted Actor-Learner Architectures.
Python
981
star
33

android_env

RL research on Android devices.
Python
977
star
34

neural-processes

This repository contains notebook implementations of the following Neural Process variants: Conditional Neural Processes (CNPs), Neural Processes (NPs), Attentive Neural Processes (ANPs).
Jupyter Notebook
969
star
35

mujoco_mpc

Real-time behaviour synthesis with MuJoCo, using Predictive Control
C++
959
star
36

dramatron

Dramatron uses large language models to generate coherent scripts and screenplays.
Jupyter Notebook
947
star
37

tree

tree is a library for working with nested data structures
Python
925
star
38

materials_discovery

Jupyter Notebook
866
star
39

xmanager

A platform for managing machine learning experiments
Python
815
star
40

open_x_embodiment

Jupyter Notebook
785
star
41

chex

Python
751
star
42

ferminet

An implementation of the Fermionic Neural Network for ab-initio electronic structure calculations
Python
707
star
43

reverb

Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research
C++
700
star
44

funsearch

Jupyter Notebook
699
star
45

alphadev

Python
688
star
46

pycolab

A highly-customisable gridworld game engine with some batteries included. Make your own gridworld games to test reinforcement learning agents!
Python
659
star
47

concordia

A library for generative social simulation
Python
634
star
48

hanabi-learning-environment

hanabi_learning_environment is a research platform for Hanabi experiments.
Python
614
star
49

recurrentgemma

Open weights language model from Google DeepMind, based on Griffin.
Python
603
star
50

ai-safety-gridworlds

This is a suite of reinforcement learning environments illustrating various safety properties of intelligent agents.
Python
577
star
51

meltingpot

A suite of test scenarios for multi-agent reinforcement learning.
Python
576
star
52

ithaca

Restoring and attributing ancient texts using deep neural networks
Jupyter Notebook
547
star
53

dqn

Lua/Torch implementation of DQN (Nature, 2015)
Lua
546
star
54

uncertain_ground_truth

Dermatology ddx dataset, Jax implementations of Monte Carlo conformal prediction, plausibility regions and statistical annotation aggregation from our recent work on uncertain ground truth (TMLR'23 and ArXiv pre-print).
Python
534
star
55

distrax

Python
527
star
56

long-form-factuality

Benchmarking long-form factuality in large language models. Original code for our paper "Long-form factuality in large language models".
Python
526
star
57

surface-distance

Library to compute surface distance based performance metrics for segmentation tasks.
Python
526
star
58

tracr

Python
496
star
59

alphamissense

Python
494
star
60

dsprites-dataset

Dataset to assess the disentanglement properties of unsupervised learning methods
Jupyter Notebook
476
star
61

narrativeqa

This repository contains the NarrativeQA dataset. It includes the list of documents with Wikipedia summaries, links to full stories, and questions and answers.
Shell
452
star
62

clrs

Jupyter Notebook
444
star
63

lab2d

A customisable 2D platform for agent-based AI research
C++
420
star
64

dqn_zoo

DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (DQN) agent.
Python
406
star
65

alphastar

Python
403
star
66

dm_pix

PIX is an image processing library in JAX, for JAX.
Python
386
star
67

opro

official code for "Large Language Models as Optimizers"
Python
383
star
68

mathematics_conjectures

Jupyter Notebook
367
star
69

spriteworld

Spriteworld: a flexible, configurable python-based reinforcement learning environment
Python
367
star
70

torax

TORAX: Tokamak transport simulation in JAX
Python
361
star
71

dm_env

A Python interface for reinforcement learning environments
Python
343
star
72

dm_robotics

Libraries, tools and tasks created and used at DeepMind Robotics.
Python
341
star
73

spiral

We provide a pre-trained model for unconditional 19-step generation of CelebA-HQ images
C++
327
star
74

launchpad

Python
310
star
75

leo

Implementation of Meta-Learning with Latent Embedding Optimization
Python
302
star
76

enn

Python
291
star
77

streetlearn

A C++/Python implementation of the StreetLearn environment based on images from Street View, as well as a TensorFlow implementation of goal-driven navigation agents solving the task published in “Learning to Navigate in Cities Without a Map”, NeurIPS 2018
C++
285
star
78

gqn-datasets

Datasets used to train Generative Query Networks (GQNs) in the ‘Neural Scene Representation and Rendering’ paper.
Python
269
star
79

treescope

An interactive HTML pretty-printer for machine learning research in IPython notebooks.
Python
256
star
80

multi_object_datasets

Multi-object image datasets with ground-truth segmentation masks and generative factors.
Python
254
star
81

AQuA

A algebraic word problem dataset, with multiple choice questions annotated with rationales.
238
star
82

synjax

Python
238
star
83

grid-cells

Implementation of the supervised learning experiments in Vector-based navigation using grid-like representations in artificial agents, as published at https://www.nature.com/articles/s41586-018-0102-6
Python
236
star
84

card2code

A code generation dataset for generating the code that implements Hearthstone and Magic The Gathering card effects.
236
star
85

arnheim

Jupyter Notebook
235
star
86

torch-hdf5

Torch interface to HDF5 library
Lua
234
star
87

kfac-jax

Second Order Optimization and Curvature Estimation with K-FAC in JAX.
Python
231
star
88

dm_memorytasks

A set of 13 diverse machine-learning tasks that require memory to solve.
Python
221
star
89

Temporal-3D-Pose-Kinetics

Exploiting temporal context for 3D human pose estimation in the wild: 3D poses for the Kinetics dataset
Python
218
star
90

dm_alchemy

DeepMind Alchemy task environment: a meta-reinforcement learning benchmark
Python
197
star
91

neural_testbed

Jupyter Notebook
191
star
92

perception_test

Jupyter Notebook
184
star
93

jmp

JMP is a Mixed Precision library for JAX.
Python
183
star
94

neural_networks_chomsky_hierarchy

Neural Networks and the Chomsky Hierarchy
Python
183
star
95

nanodo

Python
180
star
96

pg19

179
star
97

spectral_inference_networks

Implementation of Spectral Inference Networks, ICLR 2019
Python
165
star
98

barkour_robot

Barkour Robot: Agile Quadruped Robots by Google DeepMind
C++
165
star
99

onetwo

Python
164
star
100

abstract-reasoning-matrices

Progressive matrices dataset, as described in: Measuring abstract reasoning in neural networks (Barrett*, Hill*, Santoro*, Morcos, Lillicrap), ICML2018
162
star