arrow
Return

Semi-Supervised Few-Shot Classification With Multitask Learning and Iterative Label Correction

delete2024-01-01
delete0
PRE
AI
H
Hong Ji
Z
Zhi Gao *
Y
Yao Lu
Z
Ziyao Li
B
Boan Chen
Y
Yanzhang Li
J
Jun Zhu
王超 (Chao Wang)
石智成 (Zhi‐Cheng Shi)
DOI:10.1109/TGRS.2024.3401071delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Few-shot learning enables rapid generalization from extremely limited training examples. While previous efforts have utilized meta-learning or data augmentation methods to mitigate the problem of data scarcity, such approaches may struggle to maintain robustness and generalize effectively due to overfitting and noise sensitivity. In this article, we propose a novel approach, the semi-supervised label correction method for few-shot learning (SSLC-FSL), which leverages the data distribution of readily available and easily obtainable unlabeled data. SSLC-FSL iteratively corrects the labels of testing samples with alternating steps of pseudo-labeling and sample selection. The objective of pseudo-labeling is to repurpose graph-based semi-supervised learning for joint prediction of the entire testing set. We then introduce a modulation selection network (MSN) to rank testing samples by learning with noisy labels. The training set is expanded by selecting confident pseudo-labeled samples. In the MSN, a modulation aggregation layer is designed to encode support class information into each testing sample, thereby highlighting target category features and mitigating the negative impact of incorrect labels. The iterative label correction process is repeated until all testing samples are recalled to the expanded support set. To boost the SSLC-FSL algorithm, we pretrain a feature extractor to produce general-purpose representations. Particularly, we investigate two types of auxiliary tasks and their collaborative learning to acquire transferable visual information via an end-to-end multitask learning model. Our SSLC-FSL outperforms current state-of-the-art methods in any shot and all data settings, with up to +27.74% on standard remote sensing benchmarks and +5.70% on standard natural scene benchmarks.
Keywords:
Task analysis
Training
Remote sensing
Metalearning
Modulation
Feature extraction
Multitasking
Few-shot learning
modulation aggregation layer (MAL)
modulation selection network (MSN)
multitask learning
semi-supervised learning

Journal

IEEE Transactions on Geoscience and Remote Sensing cover
IEEE Transactions on Geoscience and Remote Sensing
IF:
8.6
Papers:
2.1W
Citations:
10.7W

Organization

W
wuhan university
Scholars:
8.1W
Papers: 5.8W
Citations: 70