Reservoir computing based network for few-shot image classification

被引:4
作者
Wang Bin [1 ]
Lan Hai [2 ]
Yu Hui [2 ,3 ]
Guo Jie-long [2 ,3 ]
Wei Xian [2 ,3 ]
机构
[1] Fuzhou Univ, Sch Adv Mfg, Quanzhou 362200, Peoples R China
[2] Chinese Acad Sci, Fujian Inst Res Struct Matter, Fuzhou 350002, Peoples R China
[3] Fujian Sci & Technol Innovat Lab Optoelect Inform, Mindu Innovat Lab, Fuzhou 350108, Peoples R China
关键词
few-shot learning; reservoir computing; attention mechanism; feature enhancement; image classification;
D O I
10.37188/CJLCD.2022-0407
中图分类号
O7 [晶体学];
学科分类号
0702 ; 070205 ; 0703 ; 080501 ;
摘要
Aiming at the problems that current few-shot learning algorithms are prone to overfitting and insufficient generalization ability for cross-domain cases,and inspired by the property that reservoir computing (RC) does not depend on training to alleviate overfitting,a few-shot image classification method based on reservoir computing(RCFIC) is proposed. The whole method consists of a feature extraction module,a feature enhancement module and a classifier module. The feature enhancement module consists of a RC module and an attention mechanism based on the RC,which performs channel-level enhancement and pixel-level enhancement of the features of the feature extraction module,respectively. Meanwhile,the joint cosine classifier drives the network to learn feature distributions with high inter-class variance and low intra-class variance properties. Experimental results indicate that the algorithm achieves at least 1. 07% higher classification accuracy than the existing methods in Cifar- FS,FC100 and Mini-ImageNet datasets, and outperforms the second-best method in cross-domain scenes from Mini-ImageNet to CUB-200 by at least 1. 77%. Meanwhile,the ablation experiments verify the effectiveness of RCFIC. The proposed method has great generalization ability and can effectively alleviate the overfitting problem in few-shot image classification and solve the cross- domain problem to a certain extent.
引用
收藏
页码:1399 / 1408
页数:10
相关论文
共 38 条
[1]   Associative Alignment for Few-Shot Image Classification [J].
Afrasiyabi, Arman ;
Lalonde, Jean-Francois ;
Gagne, Christian .
COMPUTER VISION - ECCV 2020, PT V, 2020, 12350 :18-35
[2]  
Bertinetto L., 2019, INT C LEARNING REPRE
[3]   CrossViT: Cross-Attention Multi-Scale Vision Transformer for Image Classification [J].
Chen, Chun-Fu ;
Fan, Quanfu ;
Panda, Rameswar .
2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, :347-356
[4]  
Chen W., 2019, 7 INT C LEARNING REP
[5]   Fine-grained Categorization and Dataset Bootstrapping using Deep Metric Learning with Humans in the Loop [J].
Cui, Yin ;
Zhou, Feng ;
Lin, Yuanqing ;
Belongie, Serge .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :1153-1162
[6]  
Deng J, 2009, PROC CVPR IEEE, P248, DOI 10.1109/CVPRW.2009.5206848
[7]   A- LinkNet:semantic segmentation network based on attention and spatial information fusion [J].
Du Min-min ;
Sima Hai-feng .
CHINESE JOURNAL OF LIQUID CRYSTALS AND DISPLAYS, 2022, 37 (09) :1199-1208
[8]  
Finn C, 2017, PR MACH LEARN RES, V70
[9]   A Broader Study of Cross-Domain Few-Shot Learning [J].
Guo, Yunhui ;
Codella, Noel C. ;
Karlinsky, Leonid ;
Codella, James V. ;
Smith, John R. ;
Saenko, Kate ;
Rosing, Tajana ;
Feris, Rogerio .
COMPUTER VISION - ECCV 2020, PT XXVII, 2020, 12372 :124-141
[10]  
HOU R B, 2019, P 33 INT C NEURAL IN