EfficientFSL: Enhancing Few-Shot Classification via Query-Only Tuning in Vision Transformers

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Liao, Wenwen, Ruan, Hang, Yu, Jianbo, Song, Bing, YuansongWang, Yang, Xiaofeng
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911376651321344
author Liao, Wenwen
Ruan, Hang
Yu, Jianbo
Song, Bing
YuansongWang
Yang, Xiaofeng
author_facet Liao, Wenwen
Ruan, Hang
Yu, Jianbo
Song, Bing
YuansongWang
Yang, Xiaofeng
contents Large models such as Vision Transformers (ViTs) have demonstrated remarkable superiority over smaller architectures like ResNet in few-shot classification, owing to their powerful representational capacity. However, fine-tuning such large models demands extensive GPU memory and prolonged training time, making them impractical for many real-world low-resource scenarios. To bridge this gap, we propose EfficientFSL, a query-only fine-tuning framework tailored specifically for few-shot classification with ViT, which achieves competitive performance while significantly reducing computational overhead. EfficientFSL fully leverages the knowledge embedded in the pre-trained model and its strong comprehension ability, achieving high classification accuracy with an extremely small number of tunable parameters. Specifically, we introduce a lightweight trainable Forward Block to synthesize task-specific queries that extract informative features from the intermediate representations of the pre-trained model in a query-only manner. We further propose a Combine Block to fuse multi-layer outputs, enhancing the depth and robustness of feature representations. Finally, a Support-Query Attention Block mitigates distribution shift by adjusting prototypes to align with the query set distribution. With minimal trainable parameters, EfficientFSL achieves state-of-the-art performance on four in-domain few-shot datasets and six cross-domain datasets, demonstrating its effectiveness in real-world applications.
format Preprint
id arxiv_https___arxiv_org_abs_2601_08499
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle EfficientFSL: Enhancing Few-Shot Classification via Query-Only Tuning in Vision Transformers
Liao, Wenwen
Ruan, Hang
Yu, Jianbo
Song, Bing
YuansongWang
Yang, Xiaofeng
Computer Vision and Pattern Recognition
Artificial Intelligence
Large models such as Vision Transformers (ViTs) have demonstrated remarkable superiority over smaller architectures like ResNet in few-shot classification, owing to their powerful representational capacity. However, fine-tuning such large models demands extensive GPU memory and prolonged training time, making them impractical for many real-world low-resource scenarios. To bridge this gap, we propose EfficientFSL, a query-only fine-tuning framework tailored specifically for few-shot classification with ViT, which achieves competitive performance while significantly reducing computational overhead. EfficientFSL fully leverages the knowledge embedded in the pre-trained model and its strong comprehension ability, achieving high classification accuracy with an extremely small number of tunable parameters. Specifically, we introduce a lightweight trainable Forward Block to synthesize task-specific queries that extract informative features from the intermediate representations of the pre-trained model in a query-only manner. We further propose a Combine Block to fuse multi-layer outputs, enhancing the depth and robustness of feature representations. Finally, a Support-Query Attention Block mitigates distribution shift by adjusting prototypes to align with the query set distribution. With minimal trainable parameters, EfficientFSL achieves state-of-the-art performance on four in-domain few-shot datasets and six cross-domain datasets, demonstrating its effectiveness in real-world applications.
title EfficientFSL: Enhancing Few-Shot Classification via Query-Only Tuning in Vision Transformers
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2601.08499