Deep entity matching with pre-trained language models

By Yuliang Li, Yoshihiko Suhara, AnHai Doan, Wang Chiew Tan. Proceedings of the VLDB Endowment, 2020

516 citations recorded by the source corpus.

We present Ditto, a novel entity matching system based on pre-trained Transformer-based language models. We fine-tune and cast EM as a sequence-pair classification problem to leverage such models with a simple architecture. Our experiments show that a straight-forward application of language models such as BERT, DistilBERT, or RoBERTa pre-trained on large text corpora already significantly improves the matching quality and outperforms previous state-of-the-art (SOTA), by up to 29% of F1 score on benchmark datasets. We also developed three optimization techniques to further improve Ditto's matching capability. Ditto allows domain knowledge to be injected by highlighting important pieces of input information that may be of interest when making matching decisions. Ditto also summarizes strings that are too long so that only the essential information is retained and used for EM. Finally, Ditto adapts a SOTA technique on data augmentation for text to EM to augment the training data with (difficult) examples. This way, Ditto is forced to learn "harder" to improve the model's matching capability. The optimizations we developed further boost the performance of Ditto by up to 9.8%. Perhaps more surprisingly, we establish that Ditto can achieve the previous SOTA results with at most half the number of labeled data. Finally, we demonstrate Ditto's effectiveness on a real-world large-scale EM task. On matching two company datasets consisting of 789K and 412K records, Ditto achieves a high F1 score of 96.5%.

Subfields: Artificial Intelligence, Management Science and Operations Research

View Paper: https://www.semanticscholar.org/paper/512f34906ddaefe885af2e5eec9b2b3b50ffd377

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. Federico Bianchi
  2. Dirk Hovy
  3. David A. Sontag
  4. Monica Agrawal
  5. Hunter Lang
  6. Stefan Hegselmann
  7. Ines Chami
  8. Christopher R'e

In Management Science and Operations Research

  1. Avanika Narayan
  2. David A. Sontag
  3. Gerhard Weikum
  4. Dirk Hovy
  5. Monica Agrawal
  6. Christopher R'e
  7. Ines Chami
  8. Laurel J. Orr

Citations in the corpus