《计算机应用》唯一官方网站 ›› 2022, Vol. 42 ›› Issue (7): 2058-2064.DOI: 10.11772/j.issn.1001-9081.2021050798

• 人工智能 • 上一篇    

基于渐进式神经网络架构搜索的人体运动识别

王震宇, 张雷(), 高文彬, 权威铭   

  1. 南京师范大学 电气与自动化工程学院,南京 210023
  • 收稿日期:2021-05-17 修回日期:2021-09-13 接受日期:2021-09-22 发布日期:2021-09-13 出版日期:2022-07-10
  • 通讯作者: 张雷
  • 作者简介:王震宇(1996—),男,江苏扬州人,硕士研究生,主要研究方向:深度学习、模式识别、自然语言处理
    高文彬(1996—),男,江苏盐城人,硕士研究生,主要研究方向:计算机视觉、信号处理、目标检测
    权威铭(1996—),男,安徽宿州人,硕士研究生,主要研究方向:人工智能。
  • 基金资助:
    国家自然科学基金资助项目(61971228);江苏省自然科学基金资助项目(BK20191371)

Human activity recognition based on progressive neural architecture search

Zhenyu WANG, Lei ZHANG(), Wenbin GAO, Weiming QUAN   

  1. School of Electrical and Automation Engineering,Nanjing Normal University,Nanjing Jiangsu 210023,China
  • Received:2021-05-17 Revised:2021-09-13 Accepted:2021-09-22 Online:2021-09-13 Published:2022-07-10
  • Contact: Lei ZHANG
  • About author:WANG Zhenyu, born in 1996, M. S. candidate. His research interests include deep learning, pattern recognition, natural language processing.
    GAO Wenbin, born in 1996, M. S. candidate. His research interests include computer vision, signal processing, object detection.
    QUAN Weiming, born in 1996, M. S. candidate. His research interests include artificial intelligence.
  • Supported by:
    National Natural Science Foundation of China(61971228);Natural Science Foundation of Jiangsu Province(BK20191371)

摘要:

为了解决基于传感器数据的运动识别问题,利用深度卷积神经网络(CNN)在公开的OPPORTUNITY传感器数据集上进行运动识别,提出了一种改进的渐进式神经网络架构搜索(PNAS)算法。首先,神经网络模型设计过程中不再依赖于合适拓扑结构的手动选择,而是通过PNAS算法来设计最优拓扑结构以最大化F1分数;其次,使用基于序列模型的优化(SMBO)策略,在该策略中将按照复杂度从低到高的顺序搜索结构空间,同时学习一个代理函数以引导对结构空间的搜索;最后,将搜索过程中表现最好的20个模型在OPPORTUNIT数据集上进行完全训练,并从中选出表现最好的模型作为搜索到的最优架构。通过这种方式搜索到的最优架构在OPPORTUNITY数据集上的F1分数达到了93.08%,与进化算法搜索到的最优架构及DeepConvLSTM相比分别提升了1.34%和1.73%,证明该方法能够改进以前手工设计的模型结构,且是可行有效的。

关键词: 人体运动识别, 深度学习, 神经网络架构搜索, 卷积神经网络, 基于序列模型的优化

Abstract:

Concerning the sensor data based activity recognition problem, deep Convolutional Neural Network (CNN) was used to perform activity recognition on public OPPORTUNITY sensor dataset, and an improved Progressive Neural Architecture Search (PNAS) algorithm was proposed. Firstly, in the process of neural network model design, without manual selection of suitable topology, PNAS algorithm was used to design the optimal topology in order to maximize the F1 score. Secondly, a Sequential Model-Based Optimization (SMBO) strategy was used, in which the structure space was searched in the order of low complexity to high complexity, while a surrogate function was learned to guide the search of the structure space. Finally, the top 20 models with the best performance in the search process were fully trained on OPPORTUNIT dataset, and the best performing model was selected as the optimal architecture searched. The F1 score of the optimal architecture searched in this way reaches 93.08% on OPPORTUNITY dataset, which is increased by 1.34% and 1.73% respectively compared with those of the optimal architecture searched by evolutionary algorithm and DeepConvlSTM, which indicates that the proposed method can improve previously manually-designed architectures and is feasible and effective.

Key words: Human Activity Recognition (HAR), deep learning, Neural Architecture Search (NAS), Convolutional Neural Network (CNN), Sequential Model-Based Optimization (SMBO)

中图分类号: