计算机应用 ›› 2018, Vol. 38 ›› Issue (8): 2211-2217.DOI: 10.11772/j.issn.1001-9081.2018010223

• 人工智能 • 上一篇    下一篇

基于自适应池化的神经网络的服装图像识别

胡聪, 屈瑾瑾, 许川佩, 朱爱军   

  1. 桂林电子科技大学 电子工程与自动化学院, 广西 桂林 541004
  • 收稿日期:2018-01-25 修回日期:2018-03-18 出版日期:2018-08-10 发布日期:2018-08-11
  • 通讯作者: 屈瑾瑾
  • 作者简介:胡聪(1981-),男,江西新余人,副教授,博士,主要研究方向:深度学习、模式识别;屈瑾瑾(1990-),女,河南周口人,硕士研究生,主要研究方向:图像检测与识别、机器学习;许川佩(1968-),女,广西合浦人,教授,博士,主要研究方向:自动测试系统、图像检测与识别;朱爱军(1978-),男,江西新干人,副教授,博士,主要研究方向:机器学习、计算机视觉。
  • 基金资助:
    国家自然科学基金资助项目(61561012);广西自然科学基金资助项目(2017GXNSFAA198021);广西自动检测技术与仪器重点实验室资助项目(YQ18109);桂林电子科技大学研究生教育创新计划资助项目(2017YJCX104)。

Garment image recognition based on adaptive pooling neural network

HU Cong, QU Jinjin, XU Chuanpei, ZHU Aijun   

  1. School of Electronic Engineering and Automation, Guilin University of Electronic Technology, Guilin Guangxi 541004, China
  • Received:2018-01-25 Revised:2018-03-18 Online:2018-08-10 Published:2018-08-11
  • Supported by:
    This work is partially supported by the National Natural Science Foundation of China (61561012), Guangxi Natural Science Foundation (2017GXNSFAA198021), the Guangxi Key Laboratory of Automatic Detection Technology and Equipment (YQ18109), the Innovation Project of GUET Graduate Education (2017YJCX104).

摘要: 针对传统池化方式不能提取有效特征值的问题,提出根据池化域的尺寸、池化域内的元素值和网络的训练轮数调整池化结果的自适应池化方法,该算法依据插值原理与最大值池化模型构建函数,以特定函数值作为池化结果,然后利用交叉验证进行模型对比实验。同时提出了小样本调优法以解决目前依靠经验值在全部数据集上验证选取超参数效率较低的问题。在原始数据集上,按照分层抽样的规则抽取小样本,并基于小样本数据集对已编码的超参数组合循环训练并测试,通过对识别率最高的组合解码确定最优超参数。选用DeepFashion数据库进行相关实验,结果显示自适应池化模型的识别率达到83%左右,与最大值池化模型相比提高约2.5%。通过小样本选定超参数,并与随机组合超参数在原始数据集上进行对比实验,结果显示小样本调优法选择的超参数在经验值范围内最优,识别结果为86.98%,与随机组合超参数的平均识别率相比提高了约41.4%。自适应池化方法可以扩展到其他的神经网络中,小样本调优法对高效选取神经网络的超参数提供了依据。

关键词: 卷积神经网络, 服装图像, 自适应池化, 小样本调优, 交叉验证

Abstract: Focusing on the issue that traditional pooling methods cannot extract valid eigenvalues, an adaptive pooling method was proposed to adjust the pooling results based on the size of the pooling domain, the element values in the pooling domain and the network training rounds. A function based on the interpolation principle and the maximum pooling model was constructed, and the specific function value was used as the pooling result. Then the cross-validation method was used for the comparative experiment of the model. Focusing on the issue that the hyperparameter selection based on empirical values to verify on all datasets is inefficient, a small sample tuning method was proposed. On the original data set, small samples were extracted according to the rule of stratified sampling, and the encoded hyperparameter combinations were cyclically trained and tested based on the small data set. The optimal hyperparameters were identified by decoding the combination with the highest recognition. Using DeepFashion dataset for the related experiments, the results show that the recognition rate of adaptive pooling model is about 83%, which is about 2.5% higher than that of the maximum pooling model. Hyperparameters selected by small samples were compared though experiments with random combinations of hyperparameters on the original dataset. The results show that the hyperparameters selected by the small sample tuning method are optimal within the empirical value range. The recognition result is 86.98%, which is about 41.4% higher than the average recognition rate of the hyperparameters random combinations. The adaptive pooling method can be extended to other neural networks, and the small sample tuning method provides a basis for efficient selection of hyperparameters of deep neural networks.

Key words: Convolutional Neural Network (CNN), garment image, adaptive pooling, small sample tuning, cross-validation

中图分类号: