Google AI Weblog: Saying WIT: A Wikipedia-Based mostly Picture-Textual content Dataset


Multimodal visio-linguistic fashions depend on wealthy datasets so as to mannequin the relationship between photos and textual content. Historically, these datasets have been created by both manually captioning photos, or crawling the net and extracting the alt-text because the caption. Whereas the previous method tends to end in larger high quality information, the intensive handbook annotation course of limits the quantity of knowledge that may be created. Then again, the automated extraction method can result in greater datasets, however these require both heuristics and cautious filtering to guarantee information high quality or scaling-up fashions to attain sturdy efficiency. An extra shortcoming of present datasets is the dearth of protection in non-English languages. This naturally led us to ask: Can one overcome these limitations and create a high-quality, large-sized, multilingual dataset with quite a lot of content material?

As we speak we introduce the Wikipedia-Based mostly Picture Textual content (WIT) Dataset, a big multimodal dataset, created by extracting a number of totally different textual content picks related to a picture from Wikipedia articles and Wikimedia picture hyperlinks. This was accompanied by rigorous filtering to solely retain top quality image-text units. As detailed in “WIT: Wikipedia-based Picture Textual content Dataset for Multimodal Multilingual Machine Studying”, offered at SIGIR ‘21, this resulted in a curated set of 37.5 million entity-rich image-text examples with 11.5 million distinctive photos throughout 108 languages. The WIT dataset is on the market for obtain and use below the Artistic Commons license. We’re additionally excited to announce that we’re internet hosting a competitors with the WIT dataset in Kaggle in collaboration with Wikimedia Analysis and different exterior collaborators.

Dataset  Pictures    Textual content    Contextual Textual content    Languages  
Flickr30K32K158K< 8
SBU Captions    1M1M1
MS-COCO330K1.5M< 4; 7 (check solely)

WIT’s elevated language protection and bigger dimension relative to earlier datasets.

The distinctive benefits of the WIT dataset are:

  1. Dimension: WIT is the most important multimodal dataset of image-text examples that’s publicly out there.
  2. Multilingual: With 108 languages, WIT has 10x or extra languages than some other dataset.
  3. Contextual info: In contrast to typical multimodal datasets, which have just one caption per picture, WIT contains many page-level and section-level contextual info.
  4. Actual world entities: Wikipedia, being a broad knowledge-base, is wealthy with actual world entities which might be represented in WIT.
  5. Difficult check set: In our latest work accepted at EMNLP, all state-of-the-art fashions demonstrated considerably decrease efficiency on WIT vs. conventional analysis units (e.g., ~30 level drop in recall).

Producing the Dataset
The primary purpose of WIT was to create a big dataset with out sacrificing on high quality or protection of ideas. Thus, we began by leveraging the most important on-line encyclopedia out there at present: Wikipedia.

For an instance of the depth of data out there, contemplate the Wikipedia web page for Half Dome (Yosemite Nationwide Park, CA). As proven under, the article has quite a few fascinating textual content captions and related contextual info for the picture, such because the web page title, foremost web page description, and different contextual info and metadata.

Instance wikipedia web page with numerous image-associated textual content picks and contexts we will extract. From the Wikipedia web page for Half Dome : Photograph by DAVID ILIFF. License: CC BY-SA 3.0.

We began by deciding on Wikipedia pages which have photos, then extracted numerous image-text associations and surrounding contexts. To additional refine the info, we carried out a rigorous filtering course of to make sure information high quality. This included text-based filtering to make sure caption availability, size and high quality (e.g., by eradicating generic default filler textual content); image-based filtering to make sure every picture is a sure dimension with permissible licensing; and eventually, image-and-text-entity–based mostly filtering to make sure suitability for analysis (e.g., excluding these categorized as hate speech). We additional randomly sampled image-caption units for analysis by human editors, who overwhelmingly agreed that 98% of the samples had good image-caption alignment.

Extremely Multilingual
With information in 108 languages, WIT is the primary large-scale, multilingual, multimodal dataset.

# of Picture-Textual content Units  Distinctive Languages  # of Pictures  Distinctive Languages  
> 1M9> 1M6
500K – 1M10500K – 1M12
  100K – 500K  36  100K – 500K  35
50K – 100K1550K – 100K17
14K – 50K3813K – 50K38
WIT: protection statistics throughout languages.
Instance of a picture that’s current in additional than a dozen Wikipedia pages throughout >12 languages. From the Wikipedia web page for Wolfgang Amadeus Mozart.

The First Contextual Picture-Textual content Dataset
Most multimodal datasets solely supply a single textual content caption (or a number of variations of an analogous caption) for the given picture. WIT is the primary dataset to offer contextual info, which can assist researchers mannequin the impact of context on picture captions in addition to the selection of photos.

WIT dataset instance exhibiting image-text information and extra contextual info.

Particularly, key textual fields of WIT which may be helpful for analysis embody:

  • Textual content captions: WIT provides three totally different sorts of picture captions. This contains the (doubtlessly context influenced) “Reference description”, the (seemingly context unbiased) “Attribution description” and “Alt-text description”.
  • Contextual info: This contains the web page title, web page description, URL and native context in regards to the Wikipedia part together with the part title and textual content.

WIT has broad protection throughout these totally different fields, as proven under.

Picture-Textual content Fields of WIT    PrepareValTake a look atWhole / Distinctive
Rows / Tuples  37.1M    261.8K    210.7K  37.6M
Distinctive Pictures11.4M58K57K11.5M
Reference Descriptions16.9M150K104K  17.2M / 16.7M  
Attribution Descriptions34.8M193K200K35.2M / 10.9M
Alt-Textual content5.3M29K29K5.4M / 5.3M
Context Texts119.8M
Key fields of WIT embody each textual content captions and contextual info.

A Excessive-High quality Coaching Set and a Difficult Analysis Benchmark
The broad protection of various ideas in Wikipedia implies that the WIT analysis units function a difficult benchmark, even for state-of-the-art fashions. We discovered that for image-text retrieval, the imply recall scores for conventional datasets had been within the 80s, whereas for the WIT check set, it was within the 40s for well-resourced languages and within the 30s for the under-resourced languages. We hope this in flip can assist researchers to construct stronger, extra sturdy fashions.

WIT Dataset and Competitors with Wikimedia and Kaggle
Moreover, we’re comfortable to announce that we’re partnering with Wikimedia Analysis and some exterior collaborators to prepare a competitors with the WIT check set. We’re internet hosting this competitors in Kaggle. The competitors is an image-text retrieval activity. Given a set of photos and textual content captions, the duty is to retrieve the suitable caption(s) for every picture.

To allow analysis on this space, Wikipedia has kindly made out there photos at 300-pixel decision and a Resnet-50–based mostly picture embeddings for a lot of the coaching and the check dataset. Kaggle might be internet hosting all this picture information along with the WIT dataset itself and can present colab notebooks. Additional, the opponents may have entry to a dialogue discussion board in Kaggle so as to share code and collaborate. This allows anybody thinking about multimodality to get began and run experiments simply. We’re excited and searching ahead to what is going to consequence from the WIT dataset and the Wikipedia photos within the Kaggle platform.

We consider that the WIT dataset will assist researchers in constructing higher multimodal multilingual fashions and in figuring out higher studying and illustration methods, finally resulting in improved Machine Studying fashions in real-world duties over visio-linguistic information. For any questions, please contact We might love to listen to about how you’re utilizing the WIT dataset.

We want to thank our co-authors in Google Analysis: Jiecao Chen, Michael Bendersky and Marc Najork. We thank Beer Changpinyo, Corinna Cortes, Joshua Gang, Chao Jia, Ashwin Kakarla, Mike Lee, Zhen Li, Piyush Sharma, Radu Soricut, Ashish Vaswani, Yinfei Yang, and our reviewers for his or her insightful suggestions and feedback.

We thank Miriam Redi and Leila Zia from Wikimedia Analysis for collaborating with us on the competitors and offering picture pixels and picture embedding information. We thank Addison Howard and Walter Reade for serving to us host this competitors in Kaggle. We additionally thank Diane Larlus (Naver Labs Europe (NLE)), Yannis Kalantidis (NLE), Stéphane Clinchant (NLE), Tiziano Piccardi Ph.D. scholar at EPFL, Lucie-Aimée Kaffee PhD scholar at College of Southampton and Yacine Jernite (Hugging Face) for his or her helpful contribution in the direction of the competitors.


Please enter your comment!
Please enter your name here