Compact Yet Powerful: Group Query Attention in TinyViT Student Models for Efficient Classifications
摘要
Vision Transformers (ViTs) perform well in computer vision but need high computational power. This paper explores Grouped Query Attention (GQA) to reduce parameters in TinyViT models with competitive accuracy. We compare GQA settings with query-to-key ratios of 2:1 to 10:1 on CIFAR-10 and CIFAR-100 benchmarks. Findings show that GQA decreases attention parameters by a maximum of 42.8% with no loss of accuracy. The GQA 5:1 setup reaches 89.95% accuracy in CIFAR-10, surpassing the usual MHA’s 89.37% while shrinking attention parameters by 39.9%. These results prove GQA’s viability for deploying effective vision transformers within resource-limited environments.