利索能及
我要发布
收藏
专利号: 2018115727642
申请人: 电子科技大学
专利类型:发明专利
专利状态:已下证
更新日期:2025-12-01
缴费截止日期: 暂无
联系人

摘要:

权利要求书:

1.一种基于改进型度量学习的少样本图像分类方法,其特征在于,包括以下步骤:(1)、少样本数据集预处理

获取少样本数据集,设少样本数据集中共有N个类别的样本图片,每个类别包含K张样本图片;

将少样本数据集划分为三个不相交的子集,分别记为训练集Dtrain、验证集Dval和测试集Dtest;

(2)、按照“N-way k-shot”原则制作少样本任务集(2.1)、获取支持集Dsupport和目标集Dtarget随机从训练集Dtrain中选取n个类别的样本图片,每个类别选取k张,将这n×k张样本图片作为支持集Dsupport;再从每个类别中剩余的样本图片中选取b张,将这n×b张样本图片作为支持集Dsupport,其中,n<N,k<K,b≤K-k;

然后将支持集Dsupport和目标集Dtarget采取{0°,90°,180°,270°}四种随机旋转数据增强方式进行数据增强;

(2.2)、在验证集Dval中,按照步骤(2.1)所述方法,获取支持集 和目标集(3)、搭建基于卷积神经网络的多尺度特征提取器去掉卷积神经网络的全连接层,按照“卷积层-池化层-卷积层-池化层-卷积层-卷积层”搭建基于卷积神经网络的多尺度特征提取器;

(4)、多尺度特征提取

(4.1)、提取支持集的多尺度特征

将支持集Dsupport输入至多尺度特征提取器器,在提取过程中,将多尺度特征提取器的第四个卷积层提取到的高层次特征和第三个卷积层提取到的中层次特征进行平均池化,得到这两个卷积层的输出特征,记为F3和F4;

将支持集的输出特征F3和F4按深度方向合并,得到支持集的n×k个多尺度特征Fsupport;

(4.2)、提取目标集的多尺度特征

将目标集Dtarget输入至多尺度特征提取器,在提取过程中,将多尺度特征提取器的第四个卷积层提取到的高层次特征和第三个卷积层提取到的中层次特征进行平均池化,得到这两个卷积层的输出特征,记为 和将目标集的输出特征 和 按深度方向合并,得到目标集的n×b个多尺度特征Ftarget;

(5)、多尺度特征拼接

(5.1)、将支持集的多尺度特征Fsupport拉直为一维长向量,再对每个类别求均值,得到n个多尺度特征(5.2)、将目标集的多尺度特征Ftarget拉直为一维长向量,得到n×b个多尺度特征(5.3)、从 和 中分别选择一个多尺度特征,按两特征间的元素相减后取绝对值的方式进行拼接;

(5.4)、按上述步骤拼接完成后共计得到n2×b个拼接特征;

(6)、搭建基于全连接神经网络的改进型度量学习器

搭建三层全连接神经网络的改进型度量学习器,改进型度量学习器的前两层引入Relu非线性激活函数,第三三层引用Sigmiod非线性激活函数,三层输出分类结果;

(7)、训练基于改进型度量学习器的少样本分类器

(7.1)、设少样本分类器的分类正确率阈值为H%;

(7.2)、训练少样本分类器:少样本分类器以交叉熵损失函数作为优化目标;

在前向传播过程中,将步骤(5)得到的拼接特征输入至改进型度量学习器,输出分类结果,然后利用分类结果计算出交叉熵损失;

在后向传播过程中,通过Adam优化算法计算出交叉熵损失函数的参数梯度,再根据参数梯度更新少样本分类器的参数;然后返回步骤(2),按照上述方法进行下一轮训练,总共训练T轮;

(7.2)、验证少样本分类器:当T轮训练结束后,在验证集上,按步骤(2)所述方法制作支持集 和目标集 再将支持集 和目标集 进行多尺度特征提取及拼接,得到拼接特征;

将拼接特征输入至少样本分类器,输出分类结果及对应的分类正确率,分类正确率与分类正确率阈值H%比较,如果分类正确率大于或等于分类正确率阈值,则本轮验证的分类正确率通过,得到训练好的少样本分类器,否则,返回步骤(2),进行下一轮训练;

(8)、利用训练好的少样本分类器对待测试的少样本图像进行分类(8.1)、在测试集Dtest中,按照步骤(2.1)所述方法,获取支持集D′support和目标集D′target;

(8.2)、将支持集D′support和目标集D′target进行多尺度特征提取及拼接,得到拼接特征;

(8.3)、将拼接特征输入至训练好的少样本分类器,得到少样本图像分类结果。