SemDeDup: Data-efficient learning at web-scale through semantic deduplication

By Amro Abbas, Kushal Tirumala, Daniel Simig, Surya Ganguli, Ari S. Morcos. 2023

307 citations recorded by the source corpus.

Progress in machine learning has been driven in large part by massive increases in data. However, large web-scale datasets such as LAION are largely uncurated beyond searches for exact duplicates, potentially leaving much redundancy. Here, we introduce SemDeDup, a method which leverages embeddings from pre-trained models to identify and remove semantic duplicates: data pairs which are semantically similar, but not exactly identical. Removing semantic duplicates preserves performance and speeds up learning. Analyzing a subset of LAION, we show that SemDeDup can remove 50% of the data with minimal performance loss, effectively halving training time. Moreover, performance increases out of distribution. Also, analyzing language models trained on C4, a partially curated dataset, we show that SemDeDup improves over prior approaches while providing efficiency gains. SemDeDup provides an example of how simple ways of leveraging quality embeddings can be used to make models learn faster with less data.

Subfields: Artificial Intelligence, Management Science and Operations Research, Natural Language Processing

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Hannaneh Hajishirzi
  2. Mohit Bansal
  3. Percy Liang
  4. Luke S. Zettlemoyer
  5. Yejin Choi
  6. Noah A. Smith
  7. Jonathan Berant
  8. Matt Gardner

In Management Science and Operations Research

  1. Hannaneh Hajishirzi
  2. Siddharth Karamcheti
  3. S. Gadre
  4. Giannis Daras
  5. Alexander J. Ratner
  6. Ali Farhadi
  7. Ranjay Krishna
  8. Ludwig Schmidt

In Natural Language Processing

  1. Hannaneh Hajishirzi
  2. Mohit Bansal
  3. Luke S. Zettlemoyer
  4. Yejin Choi
  5. Matt Gardner
  6. Tushar Khot
  7. Ashish Sabharwal
  8. Jonathan Berant

Citations in the corpus