Supervised Contrastive Learning For Product Matching

Abstract

Contrastive learning has moved the state of the art for many tasks in computer vision and information retrieval in recent years. This poster is the first work that applies supervised contrastive learning to the task of product matching in e-commerce using product offers from different e-shops. More specifically, we employ a supervised contrastive learning technique to pre-train a Transformer encoder which is afterward fine-tuned for the matching task using pair-wise training data. We further propose a source-aware sampling strategy that enables contrastive learning to be applied for use cases in which the training data does not contain product identifiers. We show that applying supervised contrastive pre-training in combination with source-aware sampling significantly improves the state-of-the-art performance on several widely used benchmarks: For Abt-Buy, we reach an F1-score of 94.29 (+3.24 compared to the previous state-of-the-art), for Amazon-Google 79.28 (+ 3.7). For WDC Computers

Stats

• citations29
• S2 citations—
• github stars0
• HF likes0
• heat score11.08
• arxiv keypeeters2022supervised

Related papers

• Product1m: Towards Weakly Supervised Instance-level Product Retrieval Via Cross-modal Pretraining (2021)12.61
• Bringing Multimodality To Amazon Visual Search System (2024)6.34
• Large-scale Product Retrieval With Weakly Supervised Representation Learning (2022)0.00
• Evaluating Contrastive Models For Instance-based Image Retrieval (2021)5.24
• Unified Vision-language Representation Modeling For E-commerce Same-style Products Retrieval (2023)6.34
• TPDR: A Novel Two-step Transformer-based Product And Class Description Match And Retrieval Method (2023)0.00
• Graph Contrastive Learning With Multi-objective For Personalized Product Retrieval In Taobao Search (2023)0.00
• Training Vision Transformers For Image Retrieval (2021)0.00