基于长短期记忆网络与MobileNet V2的图像分类研究OA
Research on image classification based on long and short-term memory network and MobileNet V2
为了获得更加高效、精度更加准确的图像分类结果,文中提出一种基于MobileNet V2和长短期记忆(LSTM)网络的图像分类框架.使用MobileNet V2对图像进行分类;通过长短期记忆网络维护在前一次图像分类过程中遇到的特征状态信息来增强模型的性能.由于注意力机制可以很容易地集成到标准的卷积神经网络(CNN)架构中,能够以最小的计算负载为代价来提高模型精度和预测精度,因此,文中采用注意力机制改进整个MobileNet V2模型,使其达到预期效果.利用CIFAR-10和CIFAR-100数据集进行分类任务测试,验证模型性能.实验结果表明,在CIFAR-10和CIFAR-100数据集上,所提模型与其他CNN架构相比,有效减少了计算复杂度和参数量,能够推动相关技术在多个领域的应用和发展.
An image classification framework based on MobileNet V2 and long and short-term memory(LSTM)network is proposed,which aims to obtain more efficient and accurate image classification results.MobileNet V2 is used to classify images,and the LSTM network is used to enhance the performance of the model by maintaining the feature state information encountered in the previous process of image classification.The attention mechanism can be easily integrated into the standard convolutional neural network(CNN)architecture to improve the model accuracy and prediction accuracy at the expense of the minimum computing load.Therefore,this attention mechanism is adopted to improve the whole MobileNet V2 model to achieve the desired effect.The CIFAR-10 and CIFAR-100 datasets are used for classification tasks in order to verify the performance of the model.Experiments show that the proposed model effectively reduces the computational complexity and required parameters compared with other CNN architectures on CIFAR-10 and CIFAR-100 datasets.To sum up,the proposed model can promote the application and development of related technologies in multiple fields.
李航;陈思行;殷守林;滕琳;王加驰
沈阳师范大学 人工智能学院,辽宁 沈阳 110034沈阳师范大学 人工智能学院,辽宁 沈阳 110034沈阳师范大学 人工智能学院,辽宁 沈阳 110034||哈尔滨工程大学 信息与通信工程学院,黑龙江 哈尔滨 150001沈阳师范大学 人工智能学院,辽宁 沈阳 110034||哈尔滨工程大学 信息与通信工程学院,黑龙江 哈尔滨 150001沈阳师范大学 人工智能学院,辽宁 沈阳 110034||灵山大学 计算机与信息工程系,韩国 釜山 50510
信息技术与安全科学
图像分类MobileNet V2长短期记忆网络注意力机制卷积神经网络损失函数
image classificationMobileNet V2LSTM networkattention mechanismCNNloss function
《现代电子技术》 2026 (19)
31-36,6
辽宁省科技联合计划项目(2024JH2/102600106)辽宁省教育厅项目(LJ232510166008)
评论