This repo is the official Pytorch implementation of "Frequency-domain MLPs are More Effective Learners in Time Series Forecasting".
- forecasting
python run_longExp.py
- draw the visualization
python weight_plot.py
If you find this repo useful, please cite our paper.
@inproceedings{yi2023frequencydomain,
title={Frequency-domain {MLP}s are More Effective Learners in Time Series Forecasting},
author={Kun Yi and Qi Zhang and Wei Fan and Shoujin Wang and Pengyang Wang and Hui He and Ning An and Defu Lian and Longbing Cao and Zhendong Niu},
booktitle={Thirty-seventh Conference on Neural Information Processing Systems},
year={2023}
}
We appreciate the following github repos a lot for their valuable code base or datasets:
- Informer: https://github.com/zhouhaoyi/Informer2020
- Autoformer: https://github.com/thuml/Autoformer
- FEDformer: https://github.com/MAZiqing/FEDformer
- LTSF-Linear: https://github.com/cure-lab/LTSF-Linear
- PatchTST: https://github.com/PatchTST