Spectral refinement improves few-shot learning across domain shifts

Training-Free Spectral Transductive Refinement for Cross-Domain Few-Shot Classification

Computer Vision and Pattern Recognition

Summary

Few-shot learning means teaching a computer to recognize objects with very few examples, but it usually struggles when the new images are quite different from the ones it learned before. The authors found a way to help the computer figure out these differences better without retraining it, by looking at how all the known and unknown images relate to each other in a special math space. Their method, called Spectral Transductive Refinement, improves recognition especially when there is only one example for each object, and works well on many different image categories. Because it only works during testing and does not need extra training, it can be added to existing systems easily.

What this means in practice

  • For machine learning engineers: Improve few-shot image classification performance on new domains without retraining by applying spectral refinement at inference time.
  • For computer vision developers: Enhance recognition systems for rare or unseen object categories by refining class prototypes using label information from a single example and related unlabeled data.

Authors

Fahim Rahman, S. M. Tanjeeb Meheran Rohan, Md. Taimum Ibne Sayed, Asaduzzaman Herok, Md. Bakhtiar Hasan

Abstract

Few-shot recognition with frozen visual features is especially fragile under domain shift and one-shot supervision, where a single labelled image is an unreliable estimate of its class. We ask how far this fragility can be reduced purely at test time, without retraining the encoder or augmenting the source domain. We present Spectral Transductive Refinement (STR), a training-free transductive inference rule that exploits the geometry of the complete support-query episode. Given frozen embeddings, STR builds a joint k-nearest-neighbour graph, maps the episode into a normalized-Laplacian spectral coordinate system, initializes class representatives from the labelled support, and iteratively refines them using pseudo-labelled queries. We evaluate STR under two protocols. A controlled component study with frozen ResNet-18 features shows that spectral refinement consistently improves over single-prototype spectral initialization across five shifted domains, with the largest gains in the one-shot regime where support estimates are weakest. We then benchmark STR against recent Cross-Domain Few-Shot Learning (CD-FSL) methods using the standard miniImageNet-pretrained ResNet-10 backbone over eight established target domains. Operating entirely at inference time, STR attains the highest 1-shot average among compared methods and remains competitive at 5-shot, rivalling approaches relying on heavy source-domain meta-training augmentations. Because STR is transductive, we report its setting explicitly. Diagnostics attribute its gains to iterative refinement in spectral coordinates rather than added prototype capacity, which remains inactive in our configuration.