Logo image
FS-BAN: Born-Again Networks for Domain Generalization Few-Shot Classification
Journal article   Peer reviewed

FS-BAN: Born-Again Networks for Domain Generalization Few-Shot Classification

Yunqing Zhao and Ngai-Man Cheung
IEEE transactions on image processing, Vol.32, pp.2252-2266
01/01/2023
PMID: 37058382

Abstract

Computer Science Computer Science, Artificial Intelligence Engineering Engineering, Electrical & Electronic Science & Technology Technology
Conventional Few-shot classification (FSC) aims to recognize samples from novel classes given limited labeled data. Recently, domain generalization FSC (DG-FSC) has been proposed with the goal to recognize novel class samples from unseen domains. DG-FSC poses considerable challenges to many models due to the domain shift between base classes (used in training) and novel classes (encountered in evaluation). In this work, we make two novel contributions to tackle DG-FSC. Our first contribution is to propose Born-Again Network (BAN) episodic training and comprehensively investigate its effectiveness for DG-FSC. As a specific form of knowledge distillation, BAN has been shown to achieve improved generalization in conventional supervised classification with a closed-set setup. This improved generalization motivates us to study BAN for DG-FSC, and we show that BAN is promising to address the domain shift encountered in DG-FSC. Building on the encouraging findings, our second (major) contribution is to propose Few-Shot BAN (FS-BAN), a novel BAN approach for DG-FSC. Our proposed FS-BAN includes novel multi-task learning objectives: Mutual Regularization, Mismatched Teacher, and Meta-Control Temperature, each of these is specifically designed to overcome central and unique challenges in DG-FSC, namely overfitting and domain discrepancy. We analyze different design choices of these techniques. We conduct comprehensive quantitative and qualitative analysis and evaluation over six datasets and three baseline models. The results suggest that our proposed FS-BAN consistently improves the generalization performance of baseline models and achieves state-of-the-art accuracy for DG-FSC. Project Page: yunqing-me.github.io/Born-Again-FS/.
url
https://doi.org/10.1109/TIP.2023.3266172View
Published (Version of record) Open

Metrics

1 Record Views

Details

Logo image