Enhanced Product Classification Using Learned Prompt Ensembling and Dual Interpolation with CLIP-Based Model
Retail product classification is a crucial technology due to its market size and potential. Several algorithms have been proposed; however, these approaches are not suitable due to their inability to accommodate new data without retraining or their insufficient performance caused by class names reflecting product-specific names. In this study, we adopt a CLIP-based model for retail product classification to overcome the challenges associated with model retraining for new products and the issue of unique class names. Our approach, learned prompt ensembling and dual interpolation (LPEDI), combines prompt learning and its ensembling with encoder fine-tuning, and employs dual interpolation for coefficient adjustment. The method outperforms existing solutions on two retail product datasets, achieving a 5.9% improvement for in-distribution data and a 4.5% gain for out-of-distribution data. These results establish LPEDI as a practical and effective solution for retail product classification.
Authors
- Koichiro Niinuma (ORCID: https://orcid.org/0000-0001-8367-3988)
- Takahisa Yamamoto (ORCID: https://orcid.org/0000-0001-9929-2805)
- László A. Jeni (ORCID: https://orcid.org/0000-0002-2830-700X)
Institutions
- Fujitsu (United States) (US)
- Carnegie Mellon University (US)
Publication Details
- Published
- 2024-10-02
- DOI
- https://doi.org/10.1109/mmsp61759.2024.10743547
- Primary Topic
- Machine Learning and Data Classification
- Type
- article
- Field-Weighted Citation Impact
- 0.00