SparseSC is a package that implements an ML-enhanced version of Synthetic Control Methodologies, which introduces penalties on the both feature and unit weights. Specifically it optimizes these two equations:
by optimizing
-
$X_T$ and$X_C$ are matrices of features (covariates and/or pre-treatement outcomes) for the treated and control units, respectively -
$Y_T$ and$Y_C$ are matrices of post-treatement outcomes on the treated and control units, respectively -
$W$ is a matrix of weights such that$W \cdot (X_C \left |Y_C \right)$ forms a matrix of synthetic controls for all units -
$V$ is a diagnoal matrix of weights applied to the covariates / pre-treatment outcomes.
Note that all matrices are formated with one row per unit and one column per feature / outcome. Breaking down the two main equations, we have:
-
$\left\Vert Y_T - W\cdot Y_C \right\Vert_F^2$ is the out-of-sample squared prediction error (i.e. the squared Frobenius Norm), measured within the control units under cross validation -
$\left\Vert V \right\Vert_1$ represents how much the model depends on the features -
$\left\Vert X_T - W\cdot X_C \right\Vert_V$ is the difference between synthetic and observed units in the the feature space weighted by$V$ . Specifically,$\left\Vert A \right\Vert_V = AVA^T$ -
$\left\Vert W - J/c \right\Vert_F^2$ is the difference between optimistic weights and simple ($1/N_c$ ) weighted averages of the control units.
Typically this is used to estimate causal effects from binary treatments on observational panel (longitudinal) data. The functions fit()
and fit_fast()
provide basic fitting of the model. If you are estimating treatment effects, fitting and diagnostic information can be done via estimate_effects()
.
Though the fitting methods do not require such structure, the typical setup is where we have panel data of an outcome variable Y
for T
time periods for N
observation units (customer, computers, etc.). We may additionally have some baseline characteristics X
about the units. In the treatment effect setting, we will also have a discrete change in treatment status (e.g. some policy change) at time, T0
, for a select group of units. When there is treatment, we can think of the pre-treatment data as [X
, Y_pre
] and post-treatment data as [Y_post
].
import SparseSC
# Fit the model:
treatment_unit_size = np.full((N), np.NaN)
treatment_unit_size[treated_unit_idx] = T0
fitted_estimates = SparseSC.estimate_effects(Y,unit_treatment_periods,...)
# Print summary of the model including effect size estimates,
# p-values, and confidendence intervals:
print(fitted_estimates)
# Extract model attributes:
fitted_estimates.pl_res_post.avg_joint_effect.p_value
fitted_estimates.pl_res_post.avg_joint_effect.CI
# access the fitted Synthetic Controls model:
fitted_model = fitted_estimates.fit
See the docs for more details
See here for more info on Synthetic Controls. In essence, it is a type of matching estimator. For each unit it will find a weighted average of untreated units that is similar on key pre-treatment data. The goal of Synthetic controls is find out which variables are important to match on (the V
matrix) and then, given those, to find a vector of per-unit weights that combine the control units into its synthetic control. The synthetic control acts as the counterfactual for a unit, and the estimate of a treatment effect is the difference between the observed outcome in the post-treatment period and the synthetic control's outcome.
SparseSC makes a number of changes to Synthetic Controls. It uses regularization and feature learning to avoid overfitting, ensure uniqueness of the solution, automate researcher decisions, and allow for estimation on large datasets. See the docs for more details.
The main choices to make are:
- The solution structure
- The model-type
The first choice is whether to calculate all of the high-level parameters (V
, its regularization parameter, and the regularization parameters for the weights) on the main matching objective or whether to get approximate/fast estimates of them using non-matching formulations. The options are:
- Full joint (done by
fit()
): We optimize overv_pen
,w_pen
andV
, so that the resulting SC for controls have smallest squared prediction error onY_post
. - Separate (done by
fit_fast()
): We note that we can efficiently estimatew_pen
on main matching objective, since, givenV
, we can reformulate the finding problem into a Ridge Regression and use efficient LOO cross-validation (e.g.RidgeCV
) to estimatew_pen
. We will estimateV
using an alternative, non-matching objective (such as aMultiTaskLasso
of usingX,Y_pre
to predictY_post
). This setup also allows for feature generation to select the match space. There are two variants depending on how we handlev_pen
:- Mixed. Choose
v_pen
based on the resulting down-stream main matching objective. - Full separate: Choose
v_pen
base on approximate objective (e.g.,MultiTaskLassoCV
).
- Mixed. Choose
The Fully Separate solution is fast and often quite good so we recommend starting there, and if need be, advancing to the Mixed and then Fully Joint optimizations.
There are two main model-types (corresponding to different cuts of the data) that can be used to estimate treatment effects.
- Retrospective: The goal is to minimize squared prediction error of the control units on
Y_post
and the full-pre history of the outcome is used as features in fitting. This is the default and was used in the descriptive elements above. - Prospective: We make an artificial split in time before any treatment actually happens (
Y_pre=[Y_train,Y_test]
). The goal is to minimize squared prediction error of all units onY_test
andY_train
for all units is used as features in fitting.
Given the same amount of features, the two will only differ when there are a non-trivial number of treated units. In this case the prospective model may provide lower prediction error for the treated units, though at the cost of less pre-history data used for fitting. When there are a trivial number of units, the retrospective design will be the most efficient.
See more details about these and two additional model types (Prospective-restrictive, and full) at the docs.
You can read these online at Read the Docs. See there for:
- Custom Donor Pools Parallelization
- Constraining the
V
matrix to be in the unit simplex - Performance Notes for
fit()
- Additional Performance Considerations for
fit()
- Full parameter listings
To build the documentation see docs/dev_notes.md
.
Brian Quistorff, Matt Goldman, and Jason Thorpe (2020) Sparse Synthetic Controls: Unit-Level Counterfactuals from High-Dimensional Data, Microsoft Journal of Applied Research, 14, pp.155-170.
The easiest way to install SparseSC
is with:
pip install git+https://github.com/microsoft/SparseSC.git
Additional commands to run tests and examples are in the makefile.
This project welcomes contributions and suggestions. Most contributions require you to agree to a Contributor License Agreement (CLA) declaring that you have the right to, and actually do, grant us the rights to use your contribution. For details, visit https://cla.microsoft.com.
When you submit a pull request, a CLA-bot will automatically determine whether you need to provide a CLA and decorate the PR appropriately (e.g., label, comment). Simply follow the instructions provided by the bot. You will only need to do this once across all repos using our CLA.
This project has adopted the Microsoft Open Source Code of Conduct. For more information see the Code of Conduct FAQ or contact opencode@microsoft.com with any additional questions or comments.