【飞桨论文复现赛-小样本学习】FSL-Baseline
转自AI Studio,原文链接:【飞桨论文复现赛-小样本学习】FSL-Baseline - 飞桨AI Studio一、前言本项目为百度论文复现赛《A Closer Look at Few-shot Classification》论文复现代码。依赖环境:paddlepaddle-gpu2.2.2python3.7在miniImageNet数据集下训练和测试。5-way Acc:1-shot5-sh
转自AI Studio,原文链接:【飞桨论文复现赛-小样本学习】FSL-Baseline - 飞桨AI Studio
一、前言
本项目为百度论文复现赛《A Closer Look at Few-shot Classification》论文复现代码。
依赖环境:
- paddlepaddle-gpu2.2.2
- python3.7
在miniImageNet数据集下训练和测试。
5-way Acc:
1-shot | 5-shot | |
---|---|---|
论文 | 48.2% | 66.4% |
复现 | 48.3% | 66.6% |
二、模型背景及其介绍
参考论文:《A Closer Look at Few-shot Classification》论文链接
小样本分类旨在通过有限标记样例学习一个分类器来识别未知类,虽然近些年取得了一些重要的研究进展,但各方法网络结构、元学习算法的复杂性以及实现细节的差异为比较当前进展提出挑战。论文作者提出:
-
对几种代表性的小样本分类算法进行一致性比较分析,结果证明随着特征提取神经网络模型能力的提升,在给定领域差异的数据集上,各方法的性能差异显著缩小;
-
提出了一个基准模型,该模型在mini-ImageNet数据集和CUB数据集上的性能可以媲美几种SOTA方法;
-
提出了一种新的用于评估小样本分类算法跨领域泛化能力的实验设定,结果发现当特征提取神经网络能力较弱时,减少类内差异是提升模型性能的一个重要因素,当特征提取神经网络能力较强时,类内差异不再关键。
论文中提出的baseline模型遵循标准迁移学习的网络预训练以及微调。
-
训练阶段将图像输入到特征提取器中,再经过分类器进行分类。
-
微调阶段将训练好的模型参数固定在特征提取器中,然后重新训练一个分类器。
除Baseline模型外,论文还引入了Baseline模型的变体,Baseline++。Baseline++不同于前者的是在分类阶段采用了一个输入特征f(x)与权值矩阵的余弦相似度,然后根据这个相似度来进行softmax分类。 作者提出的Baseline和Baseline++模型如下图所示:
三、数据集
2016年google DeepMind团队从Imagnet数据集中抽取的一小部分(大小约3GB)制作了Mini-Imagenet数据集,共有100个类别,每个类别都有600张图片,共60000张(都是.jpg结尾的文件)。
Mini-Imagenet数据集中还包含了train.csv、val.csv以及test.csv三个文件。
- train.csv包含38400张图片,共64个类别。
- val.csv包含9600张图片,共16个类别。
- test.csv包含12000张图片,共20个类别。
每个csv文件之间的图像以及类别都是相互独立的,即共60000张图片,100个类。
四、运行
本项目5-way分类可设1-shot和5-shot。如果用5-shot可设置--n_shot 5,用1-shot可设置--n_shot 1。下面以5-shot为例。
解压miniImagenet数据集到./filelists目录下用于训练
In [1]
#加载miniImagenet数据集
%cd /home/aistudio/work/Paddle-CLFS/filelists/
!unzip -oq /home/aistudio/data/data138415/miniImagenet.zip
/home/aistudio/work/Paddle-CLFS/filelists
1、训练
训练的模型保存在./record目录下
训练的日志保存在./logs目录下
In [ ]
%cd /home/aistudio/work/Paddle-CLFS/
!python3 train.py --n_shot 5
2、保存特征
将提取的特征保存在分类层之前,以提高测试速度。
加载./record目录下的模型进行特征保存
In [ ]
# 可加载预先训练好的模型文件到./record目录下
%cd /home/aistudio/work/Paddle-CLFS/record/
!unzip -oq /home/aistudio/data/data140016/checkpoint_clfs.zip
In [ ]
%cd /home/aistudio/work/Paddle-CLFS/
!python3 save_features.py --n_shot 5
3、测试
测试之前执行!python3 save_features.py预先提取特征
这里展示5-shot测试结果
In [ ]
%cd /home/aistudio/work/Paddle-CLFS/
!python3 test.py --n_shot 5
五、代码结构
├─data # 数据处理包
├─filelists # 数据文件
├─methods # 模型方法
├─logs # 训练日志
├─record # 训练保存文件
│ backbone.py # 特征提取
│ configs.py # 配置文件
│ io_utils.py # 配置文件
│ README.md # readme
│ save_features.py # 保存特征
│ train.py # 训练
│ test.py # 测试
六、总结
本文其实更像是一篇研究论文,复现和实验现有算法过程中,总结出的一份实验报告。文中主要介绍了少样本学习的几种方法,增加网络深度可以弥补方法上的不足,减少类内的差异可以显著提升模型的效果,将传统的分类转化距离度量分类可以实现这种减少类内差异,正如baseline++所描述的那样。在基类和新类之间存在域转换的场景下,基于baseline的模型表现要好于元学习方法。
所有评论(0)