TimeDistill: Efficient Long-Term Time Series Forecasting with MLP via Cross-Architecture Distillation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Ni, Juntong, Liu, Zewen, Wang, Shiyu, Jin, Ming, Jin, Wei
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911358225743872
author Ni, Juntong
Liu, Zewen
Wang, Shiyu
Jin, Ming
Jin, Wei
author_facet Ni, Juntong
Liu, Zewen
Wang, Shiyu
Jin, Ming
Jin, Wei
contents Transformer-based and CNN-based methods demonstrate strong performance in long-term time series forecasting. However, their high computational and storage requirements can hinder large-scale deployment. To address this limitation, we propose integrating lightweight MLP with advanced architectures using knowledge distillation (KD). Our preliminary study reveals different models can capture complementary patterns, particularly multi-scale and multi-period patterns in the temporal and frequency domains. Based on this observation, we introduce TimeDistill, a cross-architecture KD framework that transfers these patterns from teacher models (e.g., Transformers, CNNs) to MLP. Additionally, we provide a theoretical analysis, demonstrating that our KD approach can be interpreted as a specialized form of mixup data augmentation. TimeDistill improves MLP performance by up to 18.6%, surpassing teacher models on eight datasets. It also achieves up to 7X faster inference and requires 130X fewer parameters. Furthermore, we conduct extensive evaluations to highlight the versatility and effectiveness of TimeDistill.
format Preprint
id arxiv_https___arxiv_org_abs_2502_15016
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle TimeDistill: Efficient Long-Term Time Series Forecasting with MLP via Cross-Architecture Distillation
Ni, Juntong
Liu, Zewen
Wang, Shiyu
Jin, Ming
Jin, Wei
Machine Learning
Transformer-based and CNN-based methods demonstrate strong performance in long-term time series forecasting. However, their high computational and storage requirements can hinder large-scale deployment. To address this limitation, we propose integrating lightweight MLP with advanced architectures using knowledge distillation (KD). Our preliminary study reveals different models can capture complementary patterns, particularly multi-scale and multi-period patterns in the temporal and frequency domains. Based on this observation, we introduce TimeDistill, a cross-architecture KD framework that transfers these patterns from teacher models (e.g., Transformers, CNNs) to MLP. Additionally, we provide a theoretical analysis, demonstrating that our KD approach can be interpreted as a specialized form of mixup data augmentation. TimeDistill improves MLP performance by up to 18.6%, surpassing teacher models on eight datasets. It also achieves up to 7X faster inference and requires 130X fewer parameters. Furthermore, we conduct extensive evaluations to highlight the versatility and effectiveness of TimeDistill.
title TimeDistill: Efficient Long-Term Time Series Forecasting with MLP via Cross-Architecture Distillation
topic Machine Learning
url https://arxiv.org/abs/2502.15016