关键词:
深度学习
卷积神经网络
注意力机制
海洋生物分类
图像识别
摘要:
海洋生物分类是海洋开发工作的基石,随着海洋开发的不断推进,其重要性也日益凸显。海洋生物种类繁多,且同门类生物具有较强的类间相似性,而不同门类生物具有较大差异。海洋生物分类方法总与一般分类任务保持一致,通过一次分类完成任务,忽视了海洋生物中隐含的层次关系。卷积具有优秀的局部特征提取能力,而缺乏了全局视野,注意力机制擅长处理全局特征,但缺乏局部视野。将二者正确结合,才能充分利用其优势,使海洋生物分类任务更高效。
本文针对上述问题,设计了多层次的海洋生物分类方法,并且利用卷积与注意力机制的结合进一步改进方法。通过南麂列岛潮间带海洋生物数据集、CIFAR-100、Wildfish野生鱼类数据集进行实验验证。主要研究内容及工作如下:
(1)设计了多层次海洋生物分类方法CM-EfficientNetV2,引入生物先验知识,辅助学习海洋生物数据集的多阶层关系。构建了南麂列岛潮间带海洋生物数据集,并构建多阶层数据集,设计了平衡参数保证划分数据集的样本均衡及类别均衡。提出了C-MBConv模块同时提取空间特征及通道特征,并利用其构建了具有六个主要阶段的C-EfficientNetV2,网络性能相较于原网络有不小提升。设计了精细分类模块匹配上下文特征,融合多层次分类知识使网络更适合多层次分类任务,并且输出最终目标类别。提出了风险最小化策略,来避免多层次分类过程中产生的错误传递,即在一阶类别时分类失误导致后续连锁失败,建模网络输出,构建类别间的特征分布,计算差异并且优化。实验结果显示了为网络引入多层次分类知识可以有效提升海洋生物分类的精度,成功解决了一般海洋生物分类方法忽视物种阶层关系的问题,在CIFAR-100中也有着不错表现。
(2)设计了融合卷积与注意力机制的多层次分类方法MCT-EfficientNetV2,通过先利用卷积提取局部特征,再利用注意力机制获取全局关系的策略,重新设计了网络架构CT-EfficientNetV2。同时利用注意力机制重新设计了精细分类模块,使得整体网络架构符合浅层卷积、深层注意力的模式。提出了高斯全连接层,利用高斯分布的思想,辅助网络学习二阶类别间特征平均值可能出现的正态分布,重新规划了风险最小化策略,帮助学习类别间的分布特征,降低错误传播概率。设计了多层次损失,更高效的辅助网络学习多层次类别特征。实验结果显示本方法的计算量以及参数量都获得了大幅度的降低,同时精度也获得了提升,实现了卷积与注意力机制的高效结合。
(3)使用本文的研究方法开发了南麂列岛潮间带海洋生物识别小程序,体现了其实际应用价值。实现了通过相册上传或拍照的方式,返回所识别的物种的种名,分布地区,趣味百科,分类地位等详细信息。