Reliable Few-shot Learning under Dual Noises

TL;DR

This paper introduces DETA++, a novel method for reliable few-shot learning that effectively handles dual in-distribution and out-of-distribution noises during task adaptation, improving robustness and prediction accuracy.

Contribution

DETA++ presents a comprehensive noise-robust framework with novel modules like CoRA, LocalNCC, and IntraSwap, advancing few-shot learning under noisy conditions.

Findings

DETA++ outperforms existing methods on noisy FSL benchmarks.

The proposed modules significantly improve noise robustness.

Extensive experiments validate the effectiveness of DETA++.

Abstract

Recent advances in model pre-training give rise to task adaptation-based few-shot learning (FSL), where the goal is to adapt a pre-trained task-agnostic model for capturing task-specific knowledge with a few-labeled support samples of the target task.Nevertheless, existing approaches may still fail in the open world due to the inevitable in-distribution (ID) and out-of-distribution (OOD) noise from both support and query samples of the target task. With limited support samples available, i) the adverse effect of the dual noises can be severely amplified during task adaptation, and ii) the adapted model can produce unreliable predictions on query samples in the presence of the dual noises. In this work, we propose DEnoised Task Adaptation (DETA++) for reliable FSL. DETA++ uses a Contrastive Relevance Aggregation (CoRA) module to calculate image and region weights for support samples, based on which a clean prototype loss and a noise entropy maximization loss are proposed to achieve noise-robust task adaptation. Additionally,DETA++ employs a memory bank to store and refine clean regions for each inner-task class, based on which a Local Nearest Centroid Classifier (LocalNCC) is devised to yield noise-robust predictions on query samples. Moreover, DETA++ utilizes an Intra-class Region Swapping (IntraSwap) strategy to rectify ID class prototypes during task adaptation, enhancing the model's robustness to the dual noises. Extensive experiments demonstrate the effectiveness and flexibility of DETA++.