MADLAD-400: A Multilingual And Document-Level Large Audited Dataset

TL;DR

MADLAD-400 is a manually audited multilingual monolingual dataset covering 419 languages, enabling high-quality pretraining for multilingual models.

cs.CL πŸ”΄ Advanced 2023-09-09 55 views
Sneha Kudugunta Isaac Caswell Biao Zhang Xavier Garcia Christopher A. Choquette-Choo Katherine Lee Derrick Xin Aditya Kusupati Romi Stella Ankur Bapna Orhan Firat
multilingual dataset data auditing machine translation pretraining cross-lingual learning

Key Findings

Methodology

This work collected web data from CommonCrawl, applied a multilingual LangID model for document-level annotation, and conducted multi-stage manual audits to filter noise and low-quality content. The resulting dataset, MADLAD-400, includes over 3 trillion tokens across 419 languages. The team trained a 10.7B-parameter multilingual Transformer model using 250 billion tokens, combining supervised and unsupervised objectives, which outperformed larger models on multiple benchmarks. They also trained an 8B decoder-only language model for few-shot translation evaluation. The process involved iterative filtering, language code correction, and content filtering for religious and inappropriate material, ensuring high data quality and broad language coverage.

Key Results

  • On WMT and Flores-200 benchmarks, the 10.7B model achieved an average BLEU of 32.7, surpassing larger models, especially in low-resource languages. Back-translation further improved BLEU by approximately 3 points, demonstrating the importance of data quality. The 8B language model showed promising few-shot translation capabilities, though still lagging behind supervised models, highlighting the significance of high-quality data. The manual audit identified and removed 79 languages with predominantly noisy or inappropriate content, enhancing model robustness and fairness.
  • The models exhibited strong performance across diverse language families and scripts, with notable improvements in low-resource settings. The filtering process effectively reduced noise, religious, and pornographic content, leading to more reliable multilingual training data. The experiments confirmed that data quality, combined with model size and training strategies, is crucial for advancing multilingual NLP. The results set new baselines for future research and applications in multilingual translation and understanding.
  • Additional experiments with back-translation and fine-tuning demonstrated consistent performance gains, validating the dataset's utility. The study also evaluated models' memorization risks, ensuring data privacy and copyright compliance. Overall, MADLAD-400 provides a comprehensive resource for developing fairer, more inclusive multilingual models, addressing key bottlenecks in low-resource language processing.

Significance

This research addresses critical challenges in multilingual NLP by providing a high-quality, broad-coverage dataset and scalable training strategies. It bridges the gap between resource-rich and resource-poor languages, enabling models to perform better across diverse linguistic contexts. The systematic filtering and manual auditing set new standards for data quality, fostering more equitable AI systems. The models trained on MADLAD-400 demonstrate improved translation accuracy and robustness, paving the way for real-world applications such as multilingual chatbots, cross-lingual information retrieval, and global communication tools. This work significantly advances the state-of-the-art, making multilingual AI more accessible and reliable worldwide.

Technical Contribution

The paper introduces a multi-stage filtering pipeline combining automated heuristics, a document-level LangID model, and manual review, to produce a high-quality multilingual dataset. It proposes a scalable training framework for large Transformer models, integrating supervised and unsupervised objectives, with a shared vocabulary and dynamic sampling strategies. The models leverage massive data (250 billion tokens) and parameter scaling (up to 10.7B) to improve low-resource language performance. The work also explores few-shot capabilities of decoder-only models, demonstrating transfer learning potential. These innovations collectively push the boundaries of multilingual pretraining, offering new avenues for research and deployment.

Novelty

This study is the first to systematically combine multi-stage manual auditing with automated filtering in constructing a multilingual monolingual dataset of this scale. It introduces a novel document-level LangID model for precise language annotation and applies rigorous content filtering, including religious and inappropriate content removal. The trained models outperform larger counterparts in low-resource settings, validating the effectiveness of high-quality data. The integration of semi-supervised learning, back-translation, and multi-task training in a unified framework represents a significant advancement over prior work, setting new benchmarks for multilingual NLP.

Limitations

  • Despite rigorous filtering, some noisy or biased content may remain, especially in underrepresented languages. Manual auditing is labor-intensive and subject to human bias, limiting scalability. The models require substantial computational resources, restricting deployment in resource-constrained environments. Low-resource languages still face data scarcity issues, impacting generalization. Future work should focus on automating quality control, reducing costs, and expanding coverage to include multimodal data for richer language understanding.

Future Work

Future directions include developing more automated and scalable auditing techniques, integrating multimodal data (images, speech) to enhance low-resource language modeling, and exploring model compression for deployment efficiency. The authors also plan to extend the dataset to cover more dialects and scripts, improve few-shot and zero-shot transfer capabilities, and incorporate community feedback to refine filtering strategies. These efforts aim to democratize multilingual AI, making it more accessible, fair, and effective across diverse linguistic and cultural contexts.

AI Executive Summary

MADLAD-400 marks a significant milestone in multilingual NLP, offering a meticulously curated, high-quality dataset covering 419 languages. By leveraging web-scale data from CommonCrawl, combined with sophisticated multi-stage filtering and manual review, the researchers created a resource that balances scale with quality. This dataset underpins the training of a 10.7B-parameter multilingual Transformer model, which demonstrates competitive performance on diverse benchmarks such as WMT, Flores-200, and NTREX. The model's success underscores the importance of data quality, as back-translation and filtering strategies substantially boost translation accuracy, especially for low-resource languages.

The study emphasizes that large-scale models trained on high-quality data can outperform bigger models trained on noisier datasets, highlighting the critical role of data curation. The introduction of a document-level LangID model and rigorous filtering processes effectively mitigated noise, religious, and inappropriate content, ensuring broad language coverage and fairness. The models trained exhibit promising few-shot transfer capabilities, although they still lag behind supervised counterparts, indicating room for further improvement.

Overall, MADLAD-400 provides a valuable resource for advancing multilingual NLP, fostering more inclusive and robust models. The authors advocate for future work on automating data filtering, reducing computational costs, and expanding multimodal integration, aiming to democratize access to high-quality multilingual AI. This work sets a new benchmark in the field, inspiring subsequent research and practical applications in global language understanding and translation.

Deep Dive

Abstract

We introduce MADLAD-400, a manually audited, general domain 3T token monolingual dataset based on CommonCrawl, spanning 419 languages. We discuss the limitations revealed by self-auditing MADLAD-400, and the role data auditing had in the dataset creation process. We then train and release a 10.7B-parameter multilingual machine translation model on 250 billion tokens covering over 450 languages using publicly available data, and find that it is competitive with models that are significantly larger, and report the results on different domains. In addition, we train a 8B-parameter language model, and assess the results on few-shot translation. We make the baseline models available to the research community.

cs.CL cs.LG