用自蒸馏提升可穿戴设备多任务识别精度与训练效率
Smooth-Distill: A Self-distillation Framework for Multitask Learning with Wearable Sensor Data
- 用自身历史版本做教师模型,省去额外训练开销
- 在3种佩戴位置上识别12种睡姿,准确率显著提升
- 适合资源受限设备频繁更新模型的场景
本文提出Smooth-Distill,一种基于自蒸馏的多任务学习框架,用于可穿戴传感器数据中的人体活动识别(HAR)与设备佩戴位置检测。采用统一的CNN架构MTL-net,处理加速度计数据并分支输出两个任务结果。不同于传统需独立教师-学生模型的方法,该框架以自身平滑的历史版本作为教师模型,大幅降低训练计算开销,同时保持性能优势。为此构建了一个包含12种睡眠姿势、3种佩戴位置的加速度计数据集,补充了MHealth和WISDM两个公开数据集。实验表明,Smooth-Distill在多种评估场景下均优于对比方法,在人体活动识别和设备定位任务上均有显著提升,训练收敛更稳定,过拟合更少。该框架为可穿戴系统中的知识蒸馏提供了高效实用方案,尤其适用于需要频繁更新或在资源受限平台部署的场景。代码与模型已开源。
原文摘要 · Abstract (English)
This paper introduces Smooth-Distill, a novel self-distillation framework designed to simultaneously perform human activity recognition (HAR) and sensor placement detection using wearable sensor data. The proposed approach utilizes a unified CNN-based architecture, MTL-net, which processes accelerometer data and branches into two outputs for each respective task. Unlike conventional distillation methods that require separate teacher and student models, the proposed framework utilizes a smoothed, historical version of the model itself as the teacher, significantly reducing training computational overhead while maintaining performance benefits. To support this research, we developed a comprehensive accelerometer-based dataset capturing 12 distinct sleep postures across three different wearing positions, complementing two existing public datasets (MHealth and WISDM). Experimental results show that Smooth-Distill consistently outperforms alternative approaches across different evaluation scenarios, achieving notable improvements in both human activity recognition and device placement detection tasks. This method demonstrates enhanced stability in convergence patterns during training and exhibits reduced overfitting compared to traditional multitask learning baselines. This framework contributes to the practical implementation of knowledge distillation in human activity recognition systems, offering an effective solution for multitask learning with accelerometer data that balances accuracy and training efficiency. More broadly, it reduces the computational cost of model training, which is critical for scenarios requiring frequent model updates or training on resource-constrained platforms. The code and model are available at https://github.com/Kuan2vn/smooth\_distill.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。