arXiv · 2511.02992
Hybrid Convolution and Vision Transformer NAS Search Space for TinyML Image Classification
Abstract
Hybrids of Convolutional Neural Network (CNN) and Vision Transformer (ViT) have outperformed pure CNN or ViT architecture. However, since these architectures require large parameters and incur large computational costs, they are unsuitable for tinyML deployment. This paper introduces a new hybrid CNN-ViT search space for Neural Architecture Search (NAS) to find efficient hybrid architectures for image classification. The search space covers hybrid CNN and ViT blocks to learn local and global information, as well as the novel Pooling block of searchable pooling layers for efficient feature map reduction. Experimental results on the CIFAR10 dataset show that our proposed search space can produce hybrid CNN-ViT architectures with superior accuracy and inference speed to ResNet-based tinyML models under tight model size constraints.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Mikhael Djajapermana, Moritz Reiber, Daniel Mueller-Gritschneder, Ulf Schlichtmann. 2025-11-04. Hybrid Convolution and Vision Transformer NAS Search Space for TinyML Image Classification. https://arxiv.org/abs/2511.02992
Cite the original work for its findings. Save a collection to share your selection of sources.