Skip to content

SC-Block is a supervised contrastive blocking method which combines supervised contrastive learning for positioning records in an embedding space and nearest neighbour search for candidate set building.

License

Notifications You must be signed in to change notification settings

wbsg-uni-mannheim/SC-Block

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

18 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SC-Block

SC-Block is a blocking method that utilizes supervised contrastive learning for positioning records in the embedding space, and nearest neighbour search for candidate set building. In this repository we share the code for SC-Block to reproduce the results of the paper "SC-Block: Supervised Contrastive Blocking within Entity Resolution Pipelines" and for benchmarking SC-Block against eight state-of-the-art blocking methods. In order to relate the training time of SC-Block to the reduction of the overall runtime of the entity resolution pipeline, we combine SC-Block with four state-of-the-art matching methods into complete pipelines.

SC-Block Framework

Requirements

Install conda environment with required packages

conda env create -f environment.yml

Result reproduction

To reproduce the results run the following scripts consecutively from the root directory:

  • src/scripts/01_prepare_datasets.sh prepares the datasets (query table and index table)
  • src/scripts/02_load_data_into_es.sh loads the contents of the index table into an elastic search index
  • src/scripts/03_process_training_data.sh prepares the contrastive training
  • src/scripts/04_load_data_into_faiss.sh embeds records and loads the embeddings into faiss. Please be aware that you have to train the respective models first.
  • src/scripts/05_run_strategy.sh runs selected blocking and matching strategies.

Train contrastive models

Navigate to the folder src/finetuning/open_book/contrastive_pretraining/src/contrastive

Dataset WDC Block

The datasets WDC Block small and WDC Block medium are found in the folder data/deepmatcher. The dataset WDC Block large is too large for this repository and can be downloaded via the following link.

About

SC-Block is a supervised contrastive blocking method which combines supervised contrastive learning for positioning records in an embedding space and nearest neighbour search for candidate set building.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published