Consistent Text Categorization using Data Augmentation in e-Commerce

Noa Avigdor, Guy Horowitz, Ariel Raviv, Stav Yanovsky Daye


Abstract
The categorization of massive e-Commerce data is a crucial, well-studied task, which is prevalent in industrial settings. In this work, we aim to improve an existing product categorization model that is already in use by a major web company, serving multiple applications. At its core, the product categorization model is a text classification model that takes a product title as an input and outputs the most suitable category out of thousands of available candidates. Upon a closer inspection, we found inconsistencies in the labeling of similar items. For example, minor modifications of the product title pertaining to colors or measurements majorly impacted the model’s output. This phenomenon can negatively affect downstream recommendation or search applications, leading to a sub-optimal user experience. To address this issue, we propose a new framework for consistent text categorization. Our goal is to improve the model’s consistency while maintaining its production-level performance. We use a semi-supervised approach for data augmentation and presents two different methods for utilizing unlabeled samples. One method relies directly on existing catalogs, while the other uses a generative model. We compare the pros and cons of each approach and present our experimental results.
Anthology ID:
2023.acl-industry.30
Volume:
Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics (Volume 5: Industry Track)
Month:
July
Year:
2023
Address:
Toronto, Canada
Editors:
Sunayana Sitaram, Beata Beigman Klebanov, Jason D Williams
Venue:
ACL
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
313–321
Language:
URL:
https://aclanthology.org/2023.acl-industry.30
DOI:
10.18653/v1/2023.acl-industry.30
Bibkey:
Cite (ACL):
Noa Avigdor, Guy Horowitz, Ariel Raviv, and Stav Yanovsky Daye. 2023. Consistent Text Categorization using Data Augmentation in e-Commerce. In Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics (Volume 5: Industry Track), pages 313–321, Toronto, Canada. Association for Computational Linguistics.
Cite (Informal):
Consistent Text Categorization using Data Augmentation in e-Commerce (Avigdor et al., ACL 2023)
Copy Citation:
PDF:
https://aclanthology.org/2023.acl-industry.30.pdf