diff --git a/README.md b/README.md index 7e4b9d0..107f665 100644 --- a/README.md +++ b/README.md @@ -53,29 +53,29 @@ code. In some cases we would like to know the causal effect of some intervention but we do not have the counterfactual, making conventional methods of statistical analysis infeasible. However, it may still be possible to get an unbiased estimate of the causal effect (ATE, -ATE, or ITT) by predicting the counterfactual and comparing it to the observed outcomes. -This is the approach CausalELM takes to conduct interrupted time series analysis, -G-Computation, double machine learning, and metalearning via S-Learners, T-Learners, -X-Learners, R-learners, and doubly robust estimation. In interrupted time series analysis, -we want to estimate the effect of some intervention on the outcome of a single unit that we -observe during multiple time periods. For example, we might want to know how the -announcement of a merger affected the price of Stock A. To do this, we need to know what the -price of stock A would have been if the merger had not been announced, which we can predict -with machine learning methods. Then, we can compare this predicted counterfactual to the -observed price data to estimate the effect of the merger announcement. In another case, we -might want to know the effect of medicine X on disease Y but the administration of X was not -random and it might have also been administered at mulitiple time periods, which would -produce biased estimates. To overcome this, G-computation models the observed data, uses the -model to predict the outcomes if all patients recieved the treatment, and compares it to the -predictions of the outcomes if none of the patients recieved the treatment. Double machine -learning (DML) takes a similar approach but also models the treatment mechanism and uses it -to adjust the initial estimates. This approach has three advantages. First, it is more -efficient with high dimensional data than conventional methods. Metalearners take a similar -approach to estimate the CATE. While all of these models are different, they have one thing -in common: how well they perform depends on the underlying model they fit to the data. To -that end, CausalELMs use bagged ensembles of extreme learning machines because they are -simple yet flexible enough to be universal function approximators with lower varaince than -single extreme learning machines. +ATE, or ITT) by predicting the counterfactual and comparing it to the observed outcomes if +we are able to observe all potential confounders. This is the approach CausalELM takes to +conduct interrupted time series analysis, G-Computation, double machine learning, and +metalearning via S-Learners, T-Learners, X-Learners, R-learners, and doubly robust +estimation. In interrupted time series analysis, we want to estimate the effect of some +intervention on the outcome of a single unit that we observe during multiple time periods. +For example, we might want to know how the announcement of a merger affected the price of +Stock A. To do this, we need to know what the price of stock A would have been if the merger +had not been announced, which we can predict with machine learning methods. Then, we can +compare this predicted counterfactual to the observed price data to estimate the effect of +the merger announcement. In another case, we might want to know the effect of medicine X on +disease Y but the administration of X was not random and it might have also been +administered at mulitiple time periods, which would produce biased estimates. To overcome +this, G-computation models the observed data, uses the model to predict the outcomes if all +patients recieved the treatment, and compares it to the predictions of the outcomes if none +of the patients recieved the treatment. Double machine learning (DML) takes a similar +approach but also models the treatment mechanism and uses it to adjust the initial estimates. +This approach has three advantages. First, it is more efficient with high dimensional data +than conventional methods. Metalearners take a similar approach to estimate the CATE. While +all of these models are different, they have one thing in common: how well they perform +depends on the underlying model they fit to the data. To that end, CausalELMs use bagged +ensembles of extreme learning machines because they are simple yet flexible enough to be +universal function approximators with lower varaince than single extreme learning machines.

CausalELM Features

@@ -92,20 +92,20 @@ single extreme learning machines.

What's New?

What's Next?

Newer versions of CausalELM will hopefully support using GPUs and provide interpretations of -the results of calling validate on a model that has been estimated. In addition, some -estimators will also support using instrumental variables. However, these priorities could -also change depending on feedback recieved at JuliaCon. +the results of calling validate on a model that has been estimated. We may also add some other +features depending on demand and feedback.

Disclaimer

diff --git a/docs/src/guide/doublemachinelearning.md b/docs/src/guide/doublemachinelearning.md index 7447a16..c10ce67 100644 --- a/docs/src/guide/doublemachinelearning.md +++ b/docs/src/guide/doublemachinelearning.md @@ -54,7 +54,7 @@ We can get a summary of the model by pasing the model to the summarize method. !!!note To calculate the p-value, standard error, and confidence interval for the treatment - effect, you can set the inference keyword to false. However, these values are calculated + effect, you can set the inference keyword to true. However, these values are calculated via randomization inference, which will take a long time. This can be greatly sped up by launching Julia with more threads and lowering the number of iterations using the n keyword (at the expense of accuracy). diff --git a/docs/src/guide/gcomputation.md b/docs/src/guide/gcomputation.md index f35eb82..0a3dad6 100644 --- a/docs/src/guide/gcomputation.md +++ b/docs/src/guide/gcomputation.md @@ -55,7 +55,7 @@ We can get a summary of the model by pasing the model to the summarize method. !!!note To calculate the p-value, standard error, and confidence interval for the treatment - effect, you can set the inference keyword to false. However, these values are calculated + effect, you can set the inference keyword to true. However, these values are calculated via randomization inference, which will take a long time. This can be greatly sped up by launching Julia with more threads and lowering the number of iterations using the n keyword (at the expense of accuracy). diff --git a/docs/src/guide/its.md b/docs/src/guide/its.md index bc19ea5..5c6bed2 100644 --- a/docs/src/guide/its.md +++ b/docs/src/guide/its.md @@ -70,7 +70,7 @@ We can get a summary of the model by pasing the model to the summarize method. !!!note To calculate the p-value, standard error, and confidence interval for the treatment - effect, you can set the inference keyword to false. However, these values are calculated + effect, you can set the inference keyword to true. However, these values are calculated via randomization inference, which will take a long time. This can be greatly sped up by launching Julia with more threads and lowering the number of iterations using the n keyword (at the expense of accuracy). diff --git a/docs/src/guide/metalearners.md b/docs/src/guide/metalearners.md index 0ea02a1..536635c 100644 --- a/docs/src/guide/metalearners.md +++ b/docs/src/guide/metalearners.md @@ -79,7 +79,7 @@ We can get a summary of the model by pasing the model to the summarize method. !!!note To calculate the p-value, standard error, and confidence interval for the treatment - effect, you can set the inference keyword to false. However, these values are calculated + effect, you can set the inference keyword to true. However, these values are calculated via randomization inference, which will take a long time. This can be greatly sped up by launching Julia with more threads and lowering the number of iterations using the n keyword (at the expense of accuracy). diff --git a/docs/src/index.md b/docs/src/index.md index 5b777f0..118f319 100644 --- a/docs/src/index.md +++ b/docs/src/index.md @@ -27,6 +27,8 @@ weights and provides a form of regularization that does not have to be tuned thr validation. These attributes make CausalELM a very simple and powerful package for estimating treatment effects. +For a more interactive overview, see our JuliaCon 2024 talk[here](https://www.youtube.com/watch?v=hh_cyj8feu8&t=26s) + ### Features * Estimate a causal effect, get a summary, and validate assumptions in just four lines of code * Bagging improves performance and reduces variance without the need to tune a regularization parameter @@ -38,12 +40,12 @@ estimating treatment effects. * Codebase is high-quality, well tested, and regularly updated ### What's New? +* Model summaries contain confidence intervals and marginal effects * Now includes doubly robust estimator for CATE estimation * All estimators now implement bagging to reduce predictive performance and reduce variance * Counterfactual consistency validation simulates more realistic violations of the counterfactual consistency assumption * Uses a simple heuristic to choose the number of neurons, which reduces training time and still works well in practice * Probability clipping for classifier predictions and residuals is no longer necessary due to the bagging procedure -* CausalELM talk has been accepted to JuliaCon 2024! ### What makes CausalELM different? Other packages, mainly EconML, DoWhy, CausalAI, and CausalML, have similar funcitonality.