图像与视频处理中的深度学习算法解析
1. 小样本学习技术的应用与方法
小样本学习技术在生产生活中的潜在应用极为广泛。以电子商务平台为例,通过将多模式融合网络与元学习范式相结合,能够预测市场细分需求。目前,小样本学习(FSL)方法主要分为以下三类:
-
基于元学习的方法
:通过学习众多相关任务来优化模型空间的初始参数,总结这些任务的共性,构建学习参数的参数空间,从而获得强大的泛化能力。面对新任务时,能快速拟合和泛化新类别,经过几步梯度迭代,就能取得良好的分类效果。
-
基于数据增强的方法
:从额外的训练数据样本中获取可迁移知识,扩展训练数据,以拥有更大的数据集。通过减少目标数据集和辅助训练集样本数量的差异,避免模型过拟合,有效提高模型的泛化性能和鲁棒性。此类方法主要有两种:一是直接对原始训练集进行基于物理属性的数据操作,如旋转、缩放等;二是使用深度学习算法充分学习原始训练集,拟合新图像进行数据增强。
-
基于度量学习的方法
:“度量”一词源于数学,机器学习发展后,该概念被应用于机器学习领域。通过测量对象与种子簇中心的距离,实现对象聚类分类。度量学习通常由特征提取模块、嵌入空间模块、类别表示模块和相似度测量模块四部分组成。首先,利用神经网络模型构建嵌入空间,将样本特征非线性映射到高维嵌入空间;然后测量支持集样本和查询集样本的距离;最后使用最近邻算法完成分类。
2. 特征提取网络
在小样本图像分类任务中,特征提取网络的选择是提高分类效果的关键。尤其是在样本数量不足或样本数据不平衡的情况下,容易导致分类模型过拟合。以下介绍三种常用的特征提取网络:
-
Conv4
:早期小样本学习领域的浅层卷积神经网络,通过迭代学习自适应提取图像特征,结构简单,鲁棒性强。它由四层卷积神经网络组成,从卷积层提取图像信息并输出特征图。批量归一化层的所有通道具有一致的拉伸和偏移参数,激活函数层使用ReLU作为神经元激活函数,最大池化层用于压缩图像数据、减少模型参数以避免过拟合,最后使用全连接层进行图像分类。
-
ResNet12
:随着深度学习技术的发展,为加深网络深度,研究者提出了残差学习模块。给定输出图像特征为H(x),图像的残差为F(x),则原始图像x的特征可表示为F(x) + x。当F(x) = 0时,能保证网络性能不下降,因为学习残差比直接学习原始特征更简单。在小样本学习算法中,由于Conv4提取复杂特征的能力有限,研究者开始使用更深的ResNet。经过对比实验,发现12层的ResNet12效果良好,因此得到广泛应用。
-
WRN - 28–10
:在小样本学习中也有应用,不同的特征提取网络在不同的算法和数据集上表现各异。
3. DC - EPNet算法
为解决当前小样本学习算法中样本特征分布不平衡和样本不足导致的模型过拟合问题,提出了基于特征分布校准的小样本学习算法——分布校准嵌入传播网络(DC - EPNet)。其具体操作步骤如下:
1.
构建嵌入空间与分布校准
:利用特征提取网络构建嵌入空间,使用幂变换映射模块减少特征分布的偏度,多次进行嵌入空间中特征的分布校准,充分利用样本间的特征信息,减少类内距离,增加类间距离。
2.
构建图模型与训练分类器
:使用标签传播算法构建图模型,利用有标签的支持集样本推断无标签的查询集样本,同时训练两个具有线性激活函数的线性分类器。
3.
实验验证
:在Mini - Imagenet、TieredImageNet和CUB_200_2011基准数据集上与其他小样本学习算法进行对比实验。结果表明,DC - EPNet算法的分类准确率有显著提高,比其他算法高约1 - 10%。
以下是DC - EPNet在Mini - Imagenet数据集上使用不同特征提取网络的分类准确率对比表格:
| 特征提取网络 | 5 - Way 1 - Shot | 5 - Way 5 - Shot |
| — | — | — |
| Conv4 - MatchingNets | 43.56 | 55.31 |
| Conv4 - MAML | 48.70 | 63.11 |
| Conv4 - ProtoNets | 49.42 | 68.20 |
| Conv4 - ReNet | 50.44 | 65.32 |
| Conv4 - TPN | 53.75 | 69.43 |
| Conv4 - DC - EPNet(ours) | 59.86 | 73.14 |
| ResNet12 - ProtoNets++ | 56.52 | 74.28 |
| ResNet12 - TADAM | 58.50 | 76.70 |
| ResNet12 - MetaOpt - SVM | 62.64 | 78.60 |
| ResNet12 - TPN | 59.46 | 75.65 |
| ResNet12 - MTL | 61.20 | 75.50 |
| ResNet12 - DC - EPNet(ours) | 66.61 | 81.61 |
| WRN - 28–10 - LEO | 61.76 | 77.59 |
| WRN - 28–10 - Robust - 20++ | 62.80 | 80.85 |
| WRN - 28–10 - WDAE - GNN | 62.96 | 78.85 |
| WRN - 28–10 - CC + rot | 62.93 | 79.87 |
| WRN - 28–10 - Manifold mixup | 64.93 | 83.18 |
| WRN - 28–10 - DC - EPNet(ours) | 68.85 | 82.85 |
4. AM - EPNet算法
针对当前小样本学习算法中特征提取网络能力弱、样本相似度高导致分类准确率低的问题,提出了基于注意力机制的小样本学习算法——注意力机制嵌入传播网络(AM - EPNet)。其操作步骤如下:
1.
嵌入注意力模块
:在三个特征提取网络中嵌入卷积块注意力模块,从通道和空间两个不同维度计算图像特征,生成注意力图,使特征提取网络充分利用深度图像特征信息。
2.
匹配标签相关性与全局优化
:使用半全局匹配算法的互信息机制匹配嵌入空间中特征向量的标签相关性,使用二维平滑约束进行全局优化,利用大感受野建模解决局部和全局不一致的问题。
3.
实验验证
:在Mini - Imagenet和CUB_200_2011基准数据集上的实验结果表明,AM - EPNet算法的分类准确率显著提高。与其他小样本学习算法相比,在Mini - Imagenet数据集上提高约1 - 15%,比DC - EPNet高约1%;在CUB_200_2011数据集上提高约1 - 13%。
5. 相关模型的总结与挑战
- ConvUNeXt模型 :通过添加残差结构、丢弃池化和卷积下采样,改进了UNet的卷积块,大幅减少了总参数数量。同时提出门控机制增强特征融合,将相位计算比例从1:1:1修改为1:1:3:1。该模型在实现轻量化设计的同时提高了分割性能,但由于PyTorch缺乏对大核卷积的支持,导致显存占用增加,MLP层的显存开销也增大,未来仍有改进空间。
-
目标检测算法
:深度学习技术的应用是目标检测算法成功的关键。目前基于深度学习的目标检测算法发展迅速,但仍面临诸多困难和挑战:
- 获取高质量目标检测数据集 :基于深度学习的目标检测算法是数据驱动的,算法的准确性和鲁棒性依赖于数据集的规模和质量。目前主要有两种解决方案:一是提高现有数据集的利用效率,如同时使用多个数据集的数据训练算法,或采用数据增强、迁移学习等技术;二是借助半自动标注技术降低数据标注成本,但这些方法都无法根本解决大规模目标检测训练数据缺乏的问题。未来,半监督和无监督深度学习方法的应用有望大幅降低目标检测数据集的标注成本。
- 更好地应用于视频目标检测 :目前深度卷积网络在静态图像检测方面取得了很大进展,但视频流检测不仅要关注每帧的信息,还要考虑帧间关系,拥挤情况会降低检测效率。未来需要对视频目标检测算法进行更多总结,提高模型的泛化能力、提取高质量语义特征的能力以及建立相邻帧关系的能力。
- 快速设计易于实际项目实施的目标检测算法 :深度学习强大的特征提取能力是基于深度学习的目标检测算法成功的关键。骨干网络的速度决定了目标检测算法的检测速度和项目的成败。目前应用的轻量级骨干网络在项目时间和硬件条件的要求下难以取得理想效果,且手动调整的骨干网络设计过程耗时,设计结果并非全局最优。基于神经架构搜索(NAS)的自动网络设计和基于自动机器学习(AutoML)的自动网络压缩技术,可在减少人工干预的情况下自动求解最优网络结构,有助于构建性能更高的骨干网络。
- 视频显著性检测 :深度注意力网络用于视频显著性检测,通过利用潜在对象提议的空间位置信息有效过滤背景噪声,捕捉动态视频场景的时空特征并使用权重共享机制,可有效提高连续帧之间显著性图的一致性。但基于深度网络的模型通常缺乏知识可解释性,利用低水平图像线索(如对比度和纹理)可弥补神经网络的不足,提高视频显著性检测模型的可解释性,这将是未来研究的重点。
小样本学习算法旨在利用少量有标签样本学习优秀的分类器,具有重要意义。本文介绍了小样本学习算法的研究背景、意义和定义,总结了三种经典小样本学习方法,介绍了三种特征提取网络和三个公共标准数据集,并验证了DC - EPNet算法和AM - EPNet算法的有效性。不过,仍有一些不足之处值得进一步研究,如结合语义机制和注意力机制的网络,以及改进度量分类器。
图像与视频处理中的深度学习算法解析(续)
6. 小样本学习算法实验结果对比
为了更直观地展示DC - EPNet和AM - EPNet算法在不同数据集上的性能,下面将详细对比它们与其他小样本学习算法的实验结果。
6.1 Mini - Imagenet数据集
| 算法 | 特征提取网络 | 5 - Way 1 - Shot 准确率 | 5 - Way 5 - Shot 准确率 |
|---|---|---|---|
| MatchingNets | Conv4 | 43.56 | 55.31 |
| MAML | Conv4 | 48.70 | 63.11 |
| ProtoNets | Conv4 | 49.42 | 68.20 |
| ReNet | Conv4 | 50.44 | 65.32 |
| TPN | Conv4 | 53.75 | 69.43 |
| DC - EPNet(ours) | Conv4 | 59.86 | 73.14 |
| ProtoNets++ | ResNet12 | 56.52 | 74.28 |
| TADAM | ResNet12 | 58.50 | 76.70 |
| MetaOpt - SVM | ResNet12 | 62.64 | 78.60 |
| TPN | ResNet12 | 59.46 | 75.65 |
| MTL | ResNet12 | 61.20 | 75.50 |
| DC - EPNet(ours) | ResNet12 | 66.61 | 81.61 |
| LEO | WRN - 28–10 | 61.76 | 77.59 |
| Robust - 20++ | WRN - 28–10 | 62.80 | 80.85 |
| WDAE - GNN | WRN - 28–10 | 62.96 | 78.85 |
| CC + rot | WRN - 28–10 | 62.93 | 79.87 |
| Manifold mixup | WRN - 28–10 | 64.93 | 83.18 |
| DC - EPNet(ours) | WRN - 28–10 | 68.85 | 82.85 |
| AM - EPNet(ours) | Conv4 | 较其他算法提高 1 - 15%,比 DC - EPNet 高约 1% | - |
| AM - EPNet(ours) | ResNet12 | - | - |
| AM - EPNet(ours) | WRN - 28–10 | - | - |
从表格中可以看出,DC - EPNet在不同的特征提取网络下,分类准确率都比其他部分算法有一定程度的提高,而AM - EPNet在Mini - Imagenet数据集上,相比其他小样本学习算法也有明显的提升,并且在使用Conv4作为特征提取网络时,比DC - EPNet略高。
6.2 TieredImageNet数据集
DC - EPNet在TieredImageNet数据集上进行了实验,设置了5 - way 1 - shot和5 - way 5 - shot。训练后调整学习率为0.01和0.001,权重设置为0、0.1和0.5,迭代次数设置为100和600。使用Conv4、ResNet12和WRN - 28–10作为特征提取网络,实验结果数据可视化后发现,与其他小样本学习算法相比,DC - EPNet的分类准确率提高了约1 - 8%,其中WRN - 28–10特征提取网络的分类效果更好。
以下是DC - EPNet在TieredImageNet数据集上的实验流程mermaid图:
graph LR
A[设置实验参数] --> B[选择特征提取网络]
B --> C[训练模型]
C --> D[调整学习率、权重和迭代次数]
D --> E[进行分类实验]
E --> F[数据可视化]
F --> G[对比分类准确率]
6.3 CUB_200_2011数据集
| 算法 | 特征提取网络 | 分类准确率提升情况 |
|---|---|---|
| DC - EPNet(ours) | Conv4 | 约 1 - 12% |
| DC - EPNet(ours) | ResNet12 | 约 1 - 12% |
| DC - EPNet(ours) | WRN - 28–10 | 约 1 - 12% |
| AM - EPNet(ours) | Conv4 | 约 1 - 13% |
| AM - EPNet(ours) | ResNet12 | 约 1 - 13% |
| AM - EPNet(ours) | WRN - 28–10 | 约 1 - 13% |
在CUB_200_2011数据集上,DC - EPNet和AM - EPNet都展现出了比其他小样本学习算法更高的分类准确率,AM - EPNet的提升幅度相对更明显一些。
7. 深度学习算法的未来展望
虽然目前深度学习算法在图像与视频处理领域取得了显著的成果,但仍存在许多需要改进和探索的方向。
- 模型优化 :如ConvUNeXt模型由于PyTorch对大核卷积支持不足,导致显存占用和开销问题,未来需要开发更好的框架支持或者对模型进行进一步优化,以减少显存占用,提高运行效率。
- 数据处理 :目标检测算法的数据获取和标注问题是当前的一大挑战。未来应加强半监督和无监督深度学习方法的研究,减少对大量标注数据的依赖,同时提高现有数据的利用效率。
- 算法创新 :在小样本学习领域,虽然DC - EPNet和AM - EPNet取得了较好的效果,但仍有改进空间。可以进一步探索结合语义机制和注意力机制的网络结构,改进度量分类器,以提高小样本学习算法的性能。
- 可解释性增强 :基于深度网络的模型缺乏知识可解释性是一个普遍问题。在视频显著性检测等任务中,利用低水平图像线索提高模型的可解释性是未来研究的重点,这有助于更好地理解模型的决策过程,提高模型的可信度和实用性。
深度学习算法在图像与视频处理领域有着广阔的应用前景,但也面临着诸多挑战。通过不断的研究和创新,有望解决现有问题,推动该领域取得更大的突破。
超级会员免费看


被折叠的 条评论
为什么被折叠?



