Skip to content

legalstark/RRKD

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

RRKD

Human Activity Recognition (HAR) has increasingly attracted interest within the ubiquitous computing community, focusing on improving model effectiveness and precision for deployment on devices with limited processing capabilities, essential for real-time applications. However, current deep learning approaches in HAR encounter challenges such as high computational demand, overfitting due to complex model architectures, and the difficulty of transferring deep model insights to simpler, deployable models. To address these issues, this paper introduces a pioneering Relaxed Relational Knowledge Distillation approach (RRKD) that captures relational dynamics in sensor data using a robust, well-optimized, and computationally modest fully convolutional network. Extensive experiments confirm that RRKD surpasses existing state-of-the-art algorithms in achieving superior accuracy and operational efficiency without requiring extensive hardware resources, bridging a critical gap between knowledge distillation and efficient activity recognition. Visualization analysis demonstrates that RRKD effectively identifies and interprets underlying patterns and connections in time series sensor data, highlighting its potential to capture the nuances of sensor data. The implementation has been successfully evaluated on an embedded platform. 图片描述

Training Script for HAR Datasets

This Python script (train solo.py) is designed for training and evaluating deep learning models on Human Activity Recognition (HAR) datasets. It uses PyTorch as the deep learning framework and supports multiple datasets and model architectures.

Features

  • Dataset Support: Supports multiple HAR datasets including UCI, UniMiB, WISDM, OPPO, and PAMAP.(Except for UCI, you need to download the dataset yourself)
  • Model Architectures: Supports CNN8_S, ResNet models.
  • Knowledge Distillation: Implements both RRKD (Relational Ranking Knowledge Distillation) and Vanilla Knowledge Distillation methods.
  • Mixed Precision Training: Utilizes PyTorch's automatic mixed precision training to speed up the training process.

Requirements

  • Python 3.8
  • PyTorch
  • NumPy
  • Pandas
  • Scikit-learn
  • Seaborn
  • Matplotlib
  • tqdm

About

This is the code for the RRKD project.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages