Hybrid lightweight vision transformers with attention mechanism for feature extraction and classification of product designs

Read full text
Authors: Abdul Wahid, Hikmat Ullah Khan, Anam Naz, Fawaz Khaled Alarfaj
Cited by: 3
References: 53

Abstract

In modern consumer markets, product packaging strongly influences customer attention and buying decisions. Attractive and informative designs help brands stand out in competitive environments. Recently, Artificial Intelligence (AI) has been widely used to support packaging evaluation, especially for design analysis, personalized user experiences, and product recommendation systems. However, traditional deep learning models, such as CNN-based ResNet-50 architectures, often fail to capture long-range relationships and global visual context. These limitations reduce their effectiveness in complex visual tasks like packaging classification. To address this issue, this study investigates the use of vision transformer-based models for packaging design analysis. We propose LeViT, an efficient hybrid architecture that combines convolutional neural networks with vision transformers. This design enables the model to learn both local visual details and global contextual features. The proposed approach improves feature representation while maintaining computational efficiency. Experiments were conducted on an image dataset of packaging designs. The performance of LeViT was compared with state-of-the-art models, including CNN-ResNet-50, RegNet, and ConvNeXt. The results show that the proposed model achieves the highest classification accuracy of 95%, outperforming all comparison methods. These findings demonstrate the effectiveness of transformer-based architectures for packaging classification. The proposed approach offers practical benefits for retail analytics, brand assessment, and marketing decision-making.

Fields of Study & Subjects

Consumer Packaging Perceptions and Trends Color perception and design Aesthetic Perception and Analysis

Publication details

Journal / source
PLoS ONE
Publisher
Public Library of Science (PLoS)
Volume
21
Issue
3
Pages
e0343510–e0343510
Publication date
3 March 2026
Language
EN