Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

DoFlow: Flow-based Generative Models for Interventional and Counterfactual Forecasting on Time Series

About

Time-series forecasting increasingly demands not only accurate observational predictions but also causal forecasting under interventional and counterfactual queries in multivariate systems. We present DoFlow, a flow-based generative model defined over a causal Directed Acyclic Graph (DAG) that delivers coherent observational and interventional predictions, as well as counterfactuals through the natural encoding-decoding mechanism of continuous normalizing flows (CNFs). We also provide a supporting counterfactual recovery theory under certain assumptions. Beyond forecasting, DoFlow provides explicit likelihoods of future trajectories, enabling principled anomaly detection. Experiments on synthetic datasets with various causal DAG structures and real-world hydropower and cancer-treatment time series show that DoFlow achieves accurate system-wide observational forecasting, enables causal forecasting over interventional and counterfactual queries, and effectively detects anomalies. This work contributes to the broader goal of unifying causal reasoning and generative modeling for complex dynamical systems.

Dongze Wu, Feng Qiu, Yao Xie• 2025

Related benchmarks

TaskDatasetResultRank
Causal Treatment Effect EstimationCancer Treatment Simulation (gamma_c=5, gamma_r=5) (test)
Normalized RMSE1.25
20
Causal Treatment Effect EstimationCancer Treatment Simulation (gamma_c=5, gamma_r=0) (test)
Normalized RMSE0.49
20
Causal Treatment Effect EstimationCancer Treatment Simulation (gamma_c=0, gamma_r=5) (test)
Normalized RMSE0.94
20
Interventional Time Series ForecastingTree Additive (test)
RMSE0.54
7
Interventional Time Series ForecastingTree NLNA (test)
RMSE0.62
7
Interventional Time Series ForecastingDiamond Additive (test)
RMSE0.57
7
Interventional Time Series ForecastingDiamond NLNA (test)
RMSE0.38
7
Interventional Time Series ForecastingFC-Layer Additive (test)
RMSE0.41
7
Interventional Time Series ForecastingFC-Layer NLNA (test)
RMSE0.36
7
Observational Time Series ForecastingTree Additive (test)
RMSE0.57
7
Showing 10 of 33 rows

Other info

Follow for update