A Lightweight Graph Neural Network Algorithm for Action Recognition Based on Self-Distillation

被引:1
作者
Feng, Miao [1 ]
Meunier, Jean [1 ]
机构
[1] Univ Montreal, Dept Comp Sci & Operat Res, Montreal, PQ H3C 3J7, Canada
关键词
human action recognition; graph neural networks; skeleton graph; self-distillation; model compression;
D O I
10.3390/a16120552
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Recognizing human actions can help in numerous ways, such as health monitoring, intelligent surveillance, virtual reality and human-computer interaction. A quick and accurate detection algorithm is required for daily real-time detection. This paper first proposes to generate a lightweight graph neural network by self-distillation for human action recognition tasks. The lightweight graph neural network was evaluated on the NTU-RGB+D dataset. The results demonstrate that, with competitive accuracy, the heavyweight graph neural network can be compressed by up to 80%. Furthermore, the learned representations have denser clusters, estimated by the Davies-Bouldin index, the Dunn index and silhouette coefficients. The ideal input data and algorithm capacity are also discussed.
引用
收藏
页数:15
相关论文
共 28 条
[1]  
Bruna J, 2014, Arxiv, DOI arXiv:1312.6203
[2]   Carrot grading system using computer vision feature parameters and a cascaded graph convolutional neural network [J].
Bukumira, Milos ;
Antonijevic, Milos ;
Jovanovic, Dijana ;
Zivkovic, Miodrag ;
Mladenovic, Djordje ;
Kunjadic, Goran .
JOURNAL OF ELECTRONIC IMAGING, 2022, 31 (06)
[3]  
Cheng Y, 2020, Arxiv, DOI arXiv:1710.09282
[4]   CLUSTER SEPARATION MEASURE [J].
DAVIES, DL ;
BOULDIN, DW .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 1979, 1 (02) :224-227
[5]  
Dunn J. C., 1974, Journal of Cybernetics, V4, P95, DOI 10.1080/01969727408546059
[6]  
Dunn J. C., 1973, Journal of Cybernetics, V3, P32, DOI 10.1080/01969727308546046
[7]   Skeleton Graph-Neural-Network-Based Human Action Recognition: A Survey [J].
Feng, Miao ;
Meunier, Jean .
SENSORS, 2022, 22 (06)
[8]  
Gori M, 2005, IEEE IJCNN, P729
[9]   Knowledge Distillation: A Survey [J].
Gou, Jianping ;
Yu, Baosheng ;
Maybank, Stephen J. ;
Tao, Dacheng .
INTERNATIONAL JOURNAL OF COMPUTER VISION, 2021, 129 (06) :1789-1819
[10]   Graph neural networks-based scheduler for production planning problems using reinforcement learning [J].
Hameed, Mohammed Sharafath Abdul ;
Schwung, Andreas .
JOURNAL OF MANUFACTURING SYSTEMS, 2023, 69 :91-102