PDNet: Exploring Pruning and Distillation for Real-Time Worker Activity Recognition in Industrial Assembly
Organizations: Stuttgart University, Stuttgart, Germany · Institute for Artificial Intelligence (KI), Stuttgart University, Stuttgart, Germany · Intelligent Assistive Systems (IAS) group, University of Hildesheim, Hildesheim, Germany
Abstract
In this work, we address real-time fine-grained human activity recognition (HAR) in industrial assembly and introduce PDNet, a new framework that jointly exploits structured pruning and knowledge distillation for 3D convolutional networks. We start by training a 3D-CNN backbone in a supervised manner. Then, PDNet applies channel-wise structured pruning to obtain a more compact network. In the third step, our method is optimized using knowledge distillation from the large unpruned teacher to recover accuracy lost due to pruning. On the challenging MECCANO dataset with an X3D-M backbone, the pruned model without distillation reaches a Top-1 accuracy of 25.54 %, while PDNet raises this to 39.46 %, with only 6.90 MB model size and 3.78 GFLOPs. These results show that distillation within PDNet is highly effective in restoring pruned model performance and highlight its high potential for resource-constrained HAR in industrial assembly.
Figures & tables
| Model | Top-1 test (%) | Top-5 test (%) |
|---|---|---|
| SlowFast | 45.16 | 73.75 |
| 3D MobileNet | 27.31 | 50.51 |
| X3D-L | 34.50 | 63.09 |
| X3D-M | 36.49 | 65.64 |
| Ratio (%) | Size (MB) | Time (ms) | GFLOPs |
|---|---|---|---|
| 0.0 | 21.70 | 107.72 | 6.20 |
| 5.0 | 10.75 | 98.32 | 5.65 |
| 10.0 | 9.71 | 83.75 | 5.14 |
| 15.0 | 8.73 | 83.33 | 4.62 |
| 20.0 | 7.78 | 81.17 | 4.18 |
| 25.0 | 6.90 | 80.77 | 3.78 |
| Accuracy | Computational Efficiency | ||||
|---|---|---|---|---|---|
| Model Variant | Top-1 (%) | Top-5 (%) | Size (MB) | Time (ms) | GFLOPs |
| Original X3D-M | 36.49 | 65.64 | 21.70 | 107.72 | 6.20 |
| Pruned Only | 25.54 | 58.64 | 6.90 | 80.77 | 3.78 |
| PDNet (Ours) | 39.46 | 68.72 | 6.90 | 80.77 | 3.78 |
Human Activity Recognition
Human activity recognition classifies actions and behaviors from wearable sensors, video, radio signals, and other measurements.
See all 228 papers