EdgeMTSC
April 3, 2026 ยท View on GitHub
AAAI'26: EdgeMTSC: A Lightweight Large-Kernel ConvNet for Multivariate Time Series Classification
Project Structure
- appendix.pdf: the additional Appendix of our paper
- data/: the root dir of the datasets
- data//_TRAIN.ts: the samples for training
- data//_TEST.ts: the samples for evaluation
- conf.py: a script for stable configurations
- dataloaders.py: a script for data preprocessing and data loading
- model.py: a script holding the Pytorch models
- utils.py: a script holding project construction tools
Main Parameters
Input python main.py --help, then you will get the following output.
usage: main.py [-h] [--ablation ABLATION] [--dataset DATASET] [--if-valid IF_VALID] [--num-workers NUM_WORKERS] [--batch-size BATCH_SIZE] [--dropout DROPOUT] [--epochs EPOCHS] [--checkpoint-dir CHECKPOINT_DIR] [--device DEVICE]
optional arguments:
-h, --help show this help message and exit
--ablation ABLATION
--dataset DATASET
--if-valid IF_VALID
--num-workers NUM_WORKERS
--batch-size BATCH_SIZE
--dropout DROPOUT
--epochs EPOCHS
--checkpoint-dir CHECKPOINT_DIR
--device DEVICE
Run EdgeMTSC
python main.py --dataset ArticularyWordRecognition
Requirements
nvidia-cublas-cu12 12.1.3.1
nvidia-cuda-cupti-cu12 12.1.105
nvidia-cuda-nvrtc-cu12 12.1.105
nvidia-cuda-runtime-cu12 12.1.105
nvidia-cudnn-cu12 8.9.2.26
nvidia-cufft-cu12 11.0.2.54
nvidia-curand-cu12 10.3.2.106
nvidia-cusolver-cu12 11.4.5.107
nvidia-cusparse-cu12 12.1.0.106
nvidia-nccl-cu12 2.20.5
nvidia-nvjitlink-cu12 12.4.127
nvidia-nvtx-cu12 12.1.105
scikit-base 0.7.8
scikit-image 0.19.3
scikit-learn 1.3.0
scipy 1.10.1
numpy 1.23.4
pandas 2.0.3
timm 1.0.9
torch 2.3.0
torch_cluster 1.6.3+pt23cu121
torch_geometric 2.5.3
torch-geometric-temporal 0.54.0
torch_scatter 2.1.2+pt23cu121
torch_sparse 0.6.18+pt23cu121
torch_spline_conv 1.2.2+pt23cu121
If you find it is useful, please cite our paper:
@inproceedings{EdgeMTSC,
title={EdgeMTSC: A Lightweight Large-Kernel ConvNet for Multivariate Time Series Classification},
author={Zhou, Xueyi and Li, Zhenyu and Chae, Dong-Kyu},
booktitle={Proceedings of the AAAI Conference on Artificial Intelligence},
volume={40},
number={19},
pages={16531--16539},
year={2026}
}