Multi Head Self Attention Over Image Patches With 12 Layer Transformer Encoder

via “vision transformer patch-based feature extraction”

image-classification model by undefined. 71,05,775 downloads.

Unique: Uses google/vit-base-patch16-224-in21k as foundation, which was pre-trained on ImageNet-21k (14M images) before fine-tuning on FairFace, providing strong initialization for age-relevant features. The 16x16 patch size balances between capturing fine facial details and maintaining computational efficiency, with 197 total tokens (196 patches + 1 class token).

vs others: Captures long-range facial dependencies better than CNN-based age classifiers because self-attention can directly relate distant facial regions; more parameter-efficient than stacking deep CNN layers while maintaining or exceeding accuracy on age classification benchmarks.