Enhancing deep learning models for time series classification via knowledge distillation
Summary
Knowledge Distillation (KD) is investigated for enhancing deep learning models in Time Series Classification (TSC), addressing the high computational and memory demands of state-of-the-art architectures. This work evaluates KD's effectiveness across three architectures: the Fully Convolutional Network (FCN), the convolutional Inception model, and the transformer-based ConvTran model. The approach was tested on the UCR Archive, the largest benchmark repository for time series datasets, by modifying architectural components like convolutional filters, Inception modules, and attention heads. Results consistently demonstrate that KD most effectively benefits student models of intermediate complexity across all three architectures. Specifically, a distilled FCN student achieved a 38x parameter reduction, a distilled Inception student maintained nearly teacher performance with 42% fewer parameters, and a distilled ConvTran student with 2 attention heads showed the most significant improvement. An implementation is available at https://github.com/MSD-IRIMAS/KD-4-TSC.
Key takeaway
For AI Engineers deploying deep learning models for Time Series Classification in resource-limited settings, you should integrate Knowledge Distillation. This enables significant parameter reduction, like a 38x cut for FCNs or 42% for Inception models. You can maintain competitive performance. Focus on applying KD to student models of intermediate complexity for the most impactful improvements, especially for transformer-based architectures like ConvTran.
Key insights
Knowledge Distillation effectively optimizes deep learning models for Time Series Classification, especially for intermediate-complexity student architectures.
Principles
- KD transfers knowledge from large to small models.
- Intermediate student complexity benefits most from KD.
- Architectural components can be modified for KD.
Method
Apply Knowledge Distillation to FCN, Inception, or ConvTran models for Time Series Classification, evaluating on benchmarks like UCR Archive by modifying filters, modules, or attention heads.
In practice
- Reduce FCN parameters by 38x using KD.
- Achieve 42% parameter reduction for Inception with KD.
- Improve ConvTran performance with 2 attention heads via KD.
Topics
- Knowledge Distillation
- Time Series Classification
- Deep Learning Models
- Model Compression
- FCN
- ConvTran
Code references
Best for: Research Scientist, Machine Learning Engineer, AI Scientist, AI Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Machine Learning.