The PyTorch implementation for the paper titled "Neural Stochastic Differential Games for Time-series Analysis" by Sungwoo Park, Byoungwoo Park, Moontae Lee and Changhee Lee published at ICML 2023.
Park, S., Park, B., Lee, M., & Lee, C. (2023). Neural Stochastic Differential Games for Time-series Analysis. In International Conference on Machine Learning. PMLR.
Figure 1. Conceptual illustration: We utilizes game theory to model temporal dynamics of time-series data by extending conventional differential equation to the multi-agent counterpart for decomposing the time-series.
Figure 2. The highlighted region shows temporal aggregation level for each decision and the vertical axis in the past interval represents the average temporal aggregation of the agent's decision over the future interval.
You can find more information and details about the project on [Paper] & [Project page].
This code is developed with Python3 and Pytorch. To set up an environment with the required packages, run
conda create -n masde python=3.8
conda activate masde
pip install -r requirements.txt
python main.py -data_set air_quality -T_p 48 -T_o 36 -PI 12 -D 6 -ail
python main.py -data_set physionet -T_p 48 -T_o 36 -PI 12 -D 36 -ail
python main.py -data_set speech -T_p 54 -T_o 43 -PI 11 -D 65 -ail