Noise-robust few-shot classification via variational adversarial data augmentation  

作　　者：Renjie Xu Baodi Liu Kai Zhang Honglong Chen Dapeng Tao Weifeng Liu 

机构地区：[1]College of Oceanography and Space Informatics,China University of Petroleum(East China),Qingdao 266580,China [2]College of Control Science and Engineering,China University of Petroleum(East China),Qingdao 266580,China [3]School of Petroleum Engineering,China University of Petroleum(East China),Qingdao 266580,China [4]School of Information Science and Engineering,Yunnan University,Yunnan 650504,China

出　　处：《Computational Visual Media》2025年第1期227-239,共13页计算可视媒体(英文版)

基　　金：supported by Yunnan Provincial Major Science and Technology Special Plan Projects(202202AD080003);the Natural Science Foundation of Shandong Province(ZR2022MD090).

摘　　要：Few-shot classification models trained with clean samples poorly classify samples from the real world with various scales of noise.To enhance the model for recognizing noisy samples,researchers usually utilize data augmentation or use noisy samples generated by adversarial training for model training.However,existing methods still have problems:(i)The effects of data augmentation on the robustness of the model are limited.(ii)The noise generated by adversarial training usually causes overfitting and reduces the generalization ability of the model,which is very significant for few-shot classification.(iii)Most existing methods cannot adaptively generate appropriate noise.Given the above three points,this paper proposes a noise-robust few-shot classification algorithm,VADA—Variational Adversarial Data Augmentation.Unlike existing methods,VADA utilizes a variational noise generator to generate an adaptive noise distribution according to different samples based on adversarial learning,and optimizes the generator by minimizing the expectation of the empirical risk.Applying VADA during training can make few-shot classification more robust against noisy data,while retaining generalization ability.In this paper,we utilize FEAT and ProtoNet as baseline models,and accuracy is verified on several common few-shot classification datasets,including MiniImageNet,TieredImageNet,and CUB.After training with VADA,the classification accuracy of the models increases for samples with various scales of noise.

关 键 词：few-shot learning adversarial learning ROBUSTNESS variational method 

分 类 号：TP181[自动化与计算机技术—控制理论与控制工程]