Official implementation of BALM-TSF (CIKM 2025).
BALM-TSF is a lightweight framework that alleviates modality imbalance between text and time series via statistical + learnable prompts and a balanced alignment module (scaling + contrastive learning).
-
Python 3.11 (recommended via Miniconda)
-
Install dependencies:
pip install -r requirements.txt
Download the pre-processed datasets from Google Drive, and place the extracted contents under ./dataset. This is a public dataset sharelink from Time-LLM.
- Download and unzip datasets into
./dataset. - Tune the model using the provided scripts.
- The training and test results will be visualized and recorded via Wandb.
bash ./scripts_long_term/BALM_ETTh1_GPT2.shbash ./scripts_few_shot/BALM_ETTh1_GPT2.shWe appreciate Time-Series-Library and Time-LLM for code references and datasets.
If you find this repository useful, please consider citing our work:
@article{zhou2025balmtsf,
title={BALM-TSF: Balanced Multimodal Alignment for LLM-Based Time Series Forecasting},
author={Zhou, Shiqiao and Schöner, Holger and Lyu, Huanbo and Fouché, Edouard and Wang, Shuo},
journal={arXiv preprint arXiv:2509.00622},
year={2025}
}


