Rank-guided Diffusion for Noise Few-Shot Learning
Zelei Wu, Kun Zhou, xulun ye, Yifan Mei, Jie Hong, Jieyu Zhao
Abstract
In real-world Few-Shot Learning (FSL), support sets are quickly constructed and inevitably contain noisy samples. With limited examples per class, even a single noisy instance can distort class distributions, cause prototype drift, and reduce generalization. Existing methods mostly assume clean data or require large-scale statistics, which are impractical in FSL’s data-scarce setting. We find that clean samples in semantic feature space lie in low-rank subspaces, while noisy samples cause rank anomalies disrupting this structure. To address this, we propose a differentiable low-rank approximation that estimates the intrinsic rank of the support set and detects anomalous noisy samples. Building on this, a rank-guided diffusion process generates high-quality replacements under low-rank constraints, reconstructing a clean, consistent support set for improved robustness.This low-rank guided approach effectively mitigates prototype drift and significantly reduces errors under noise levels up to 40% across MiniImageNet, TieredImageNet, and other noisy benchmarks, demonstrating the power of low-rank geometry for noise detection and correction in FSL. Our source code is available at https://github.com/wuzelei123/CRDProto.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 21273ee2-af52-4538-93ab-65bd2e2f9e23Builds on14
- Language Models are Few-Shot LearnersTom B. Brown, Benjamin Mann, Nick Ryder, Melanie Subbiah et al.NeurIPS 2020 · 64,255 citations
- Learning Transferable Visual Models From Natural Language SupervisionAlec Radford, Jong Wook Kim, Chris Hallacy, Aditya Ramesh et al.ICML 2021 · 47,906 citations
- Zero-Shot Text-to-Image GenerationAditya Ramesh, Mikhail Pavlov, Gabriel Goh, Scott Gray et al.ICML 2021 · 6,356 citations
- DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object DetectionHao Zhang, Feng Li, Shilong Liu, Lei Zhang et al.ICLR 2023 · 753 citations
- Generating Representative Samples for Few-Shot ClassificationJingyi Xu, Hieu LeCVPR 2022 · 96 citations
Related papers
- Few-shot Learning with Noisy LabelsKevin J. Liang, Samrudhdhi B. Rangrej, Vladan Petrovic, Tal HassnerCVPR 2022 · 46 citations
- DDSF: Robust Few-Shot Learning via Disentangled Subspaces with Determinantal Point ProcessXulun Ye, Yifan Mei, Kun Zhou, Zelei Wu et al.CVPR 2026
- Dual Alignment Framework for Few-shot Learning with Inter-Set and Intra-Set ShiftsSiyang Jiang, Rui Fang, Hsi-Wen Chen, Wei Ding et al.NeurIPS 2025 · 2 citations
- FoPro: Few-Shot Guided Robust Webly-Supervised Prototypical LearningYulei Qin, Xingyu Chen, Chao Chen, Yunhang Shen et al.AAAI 2023 · 5 citations
- DeFSS: Image-to-Mask Denoising Learning for Few-Shot SegmentationZishu Qin, Junhao Xu, Weifeng GeICCV 2025 · 1 citation
