The margins and
prediction packages are a combined effort to port the
functionality of Stata’s (closed source) margins
command to (open source) R. These tools provide ways of obtaining common
quantities of interest from regression-type models.
margins provides “marginal effects” summaries of models
and prediction provides unit-specific and sample
average predictions from models. Marginal effects are partial
derivatives of the regression equation with respect to each variable in
the model for each unit in the data; average marginal effects are simply
the mean of these unit-specific partial derivatives over some sample. In
ordinary least squares regression with no interactions or higher-order
term, the estimated slope coefficients are marginal effects. In other
cases and for generalized linear models, the coefficients are not
marginal effects at least not on the scale of the response variable.
margins therefore provides ways of calculating the
marginal effects of variables to make these models more
interpretable.
The major functionality of Stata’s margins
command -
namely the estimation of marginal (or partial) effects - is provided
here through a single function, margins()
. This is an S3
generic method for calculating the marginal effects of covariates
included in model objects (like those of classes “lm” and “glm”). Users
interested in generating predicted (fitted) values, such as the
“predictive margins” generated by Stata’s margins
command,
should consider using prediction()
from the sibling
project, prediction.
Stata’s margins
command is very simple and intuitive to
use:
. import delimited mtcars.csv
. quietly reg mpg c.cyl##c.hp wt
. margins, dydx(*)
------------------------------------------------------------------------------
| Delta-method
| dy/dx Std. Err. t P>|t| [95% Conf. Interval]
-------------+----------------------------------------------------------------
cyl | .0381376 .5998897 0.06 0.950 -1.192735 1.26901
hp | -.0463187 .014516 -3.19 0.004 -.076103 -.0165343
wt | -3.119815 .661322 -4.72 0.000 -4.476736 -1.762894
------------------------------------------------------------------------------
. marginsplot
With margins in R, replicating Stata’s results is
incredibly simple using just the margins()
method to obtain
average marginal effects and its summary()
method to obtain
Stata-like output:
## Average marginal effects
## lm(formula = mpg ~ cyl * hp + wt, data = mtcars)
## cyl hp wt
## 0.03814 -0.04632 -3.12
## factor AME SE z p lower upper
## cyl 0.0381 0.5999 0.0636 0.9493 -1.1376 1.2139
## hp -0.0463 0.0145 -3.1909 0.0014 -0.0748 -0.0179
## wt -3.1198 0.6613 -4.7175 0.0000 -4.4160 -1.8236
With the exception of differences in rounding, the above results
match identically what Stata’s margins
command produces. A
slightly more concise expression relies on the syntactic sugar provided
by margins_summary()
:
## factor AME SE z p lower upper
## cyl 0.0381 0.5999 0.0636 0.9493 -1.1376 1.2139
## hp -0.0463 0.0145 -3.1909 0.0014 -0.0748 -0.0179
## wt -3.1198 0.6613 -4.7175 0.0000 -4.4160 -1.8236
Using the plot()
method also yields an aesthetically
similar result to Stata’s marginsplot
:
margins()
margins is intended as a port of (some of) the
features of Stata’s margins
command, which includes
numerous options for calculating marginal effects at the mean values of
a dataset (i.e., the marginal effects at the mean), an average of the
marginal effects at each value of a dataset (i.e., the average marginal
effect), marginal effects at representative values, and any of those
operations on various subsets of a dataset. (The functionality of
Stata’s command to produce predictive margins is not ported, as
this is easily obtained from the prediction
package.) In particular, Stata provides the following options:
at
: calculate marginal effects at (potentially
representative) specified values (i.e., replacing observed values with
specified replacement values before calculating marginal effects)atmeans
: calculate marginal effects at the mean (MEMs)
of a dataset rather than the default behavior of calculating average
marginal effects (AMEs)over
: calculate marginal effects (including MEMs and/or
AMEs at observed or specified values) on subsets of the original data
(e.g., the marginal effect of a treatment separately for men and
women)The at
argument has been translated into
margins()
in a very similar manner. It can be used by
specifying a list of variable names and specified values for those
variables at which to calculate marginal effects, such as
margins(x, at = list(hp=150))
. When using at
,
margins()
constructs modified datasets - using
build_datalist()
- containing the specified values and
calculates marginal effects on each modified dataset,
rbind
-ing them back into a single “margins” object.
Stata’s atmeans
argument is not implemented in
margins()
for various reasons, including because it is
possible to achieve the effect manually through an operation like
data$var <- mean(data$var, na.rm = TRUE)
and passing the
modified data frame to margins(x, data = data)
.
At present, margins()
does not implement the
over
option. The reason for this is also simple: R already
makes data subsetting operations quite simple using simple
[
extraction. If, for example, one wanted to calculate
marginal effects on subsets of a data frame, those subsets can be passed
directly to margins()
via the data
argument
(as in a call to prediction()
).
The rest of this vignette shows how to use at
and
data
to obtain various kinds of marginal effects, and how
to use plotting functions to visualize those inferences.
We can start by loading the margins package:
We’ll use a simple example regression model based on the built-in
mtcars
dataset:
To obtain average marginal effects (AMEs), we simply call
margins()
on the model object created by
lm()
:
## Average marginal effects
## lm(formula = mpg ~ cyl + hp * wt, data = mtcars)
## cyl hp wt
## -0.3652 -0.02527 -3.838
The result is a data frame with special class "margins"
.
"margins"
objects are printed in a tidy summary format, by
default, as you can see above. The only difference between a
"margins"
object and a regular data frame are some
additional data frame-level attributes that dictate how the object is
printed.
The default method calculates marginal effects for all variables
included in the model (ala Stata’s , dydx(*)
option). To
limit calculation to only a subset of variables, use the
variables
argument:
## factor AME SE z p lower upper
## hp -0.0253 0.0105 -2.4046 0.0162 -0.0459 -0.0047
In an ordinary least squares regression, there is really only one way
of examining marginal effects (that is, on the scale of the outcome
variable). In a generalized linear model (e.g., logit), however, it is
possible to examine true “marginal effects” (i.e., the marginal
contribution of each variable on the scale of the linear predictor) or
“partial effects” (i.e., the contribution of each variable on the
outcome scale, conditional on the other variables involved in the link
function transformation of the linear predictor). The latter are the
default in margins()
, which implicitly sets the argument
margins(x, type = "response")
and passes that through to
prediction()
methods. To obtain the former, simply set
margins(x, type = "link")
. There’s some debate about which
of these is preferred and even what to call the two different quantities
of interest. Regardless of all of that, here’s how you obtain
either:
## Warning: glm.fit: fitted probabilities numerically 0 or 1 occurred
## Average marginal effects
## glm(formula = am ~ cyl + hp * wt, family = binomial, data = mtcars)
## cyl hp wt
## 0.02156 0.002667 -0.5158
## Average marginal effects
## glm(formula = am ~ cyl + hp * wt, family = binomial, data = mtcars)
## cyl hp wt
## 0.5156 0.05151 -12.24
Note that some other packages available for R, as well as Stata’s
margins
and mfx
packages enable calculation of
so-called “marginal effects at means” (i.e., the marginal effect for a
single observation that has covariate values equal to the means of the
sample as a whole). The substantive interpretation of these is fairly
ambiguous. While it was once common practice to estimate MEMs - rather
than AMEs or MERs - this is now considered somewhat inappropriate
because it focuses on cases that may not exist (e.g., the average of a
0/1 variable is not going to reflect a case that can actually exist in
reality) and we are often interested in the effect of a variable at
multiple possible values of covariates, rather than an arbitrarily
selected case that is deemed “typical” in this way. As such,
margins()
defaults to reporting AMEs, unless modified by
the at
argument to calculate average “marginal effects for
representative cases” (MERs). MEMs could be obtained by manually
specifying at
for every variable in a way that respects the
variables classes and inherent meaning of the data, but that
functionality is not demonstrated here.
at
ArgumentThe at
argument allows you to calculate marginal effects
at representative cases (sometimes “MERs”) or marginal effects at means
- or any other statistic - (sometimes “MEMs”), which are marginal
effects for particularly interesting (sets of) observations in a
dataset. This differs from marginal effects on subsets of the original
data (see the next section for a demonstration of that) in that it
operates on a modified set of the full dataset wherein particular
variables have been replaced by specified values. This is helpful
because it allows for calculation of marginal effects for
counterfactual datasets (e.g., what if all women were instead
men? what if all democracies were instead autocracies? what if all
foreign cars were instead domestic?).
As an example, if we wanted to know if the marginal effect of
horsepower (hp
) on fuel economy differed across different
types of automobile transmissions, we could simply use at
to obtain separate marginal effect estimates for our data as if every
car observation were a manual versus if every car observation were an
automatic. The output of margins()
is a simplified summary
of the estimated marginal effects across the requested variable
levels/combinations specified in at
:
## Average marginal effects at specified values
## lm(formula = mpg ~ cyl + wt + hp * am, data = mtcars)
## at(am) cyl wt hp am
## 0 -0.9339 -2.812 -0.008945 1.034
## 1 -0.9339 -2.812 -0.026392 1.034
Because of the hp * am
interaction in the regression,
the marginal effect of horsepower differs between the two sets of
results. We can also specify more than one variable to at
,
creating a potentially long list of marginal effects results. For
example, we can produce marginal effects at both levels of
am
and the values from the five-number summary (minimum,
Q1, median, Q3, and maximum) of observed values of hp
. This
produces 2 * 5 = 10 sets of marginal effects estimates:
## Average marginal effects at specified values
## lm(formula = mpg ~ cyl + wt + hp * am, data = mtcars)
## at(am) at(hp) cyl wt hp am
## 0 52 -0.9339 -2.812 -0.008945 2.6864
## 1 52 -0.9339 -2.812 -0.026392 2.6864
## 0 96 -0.9339 -2.812 -0.008945 1.9188
## 1 96 -0.9339 -2.812 -0.026392 1.9188
## 0 123 -0.9339 -2.812 -0.008945 1.4477
## 1 123 -0.9339 -2.812 -0.026392 1.4477
## 0 180 -0.9339 -2.812 -0.008945 0.4533
## 1 180 -0.9339 -2.812 -0.026392 0.4533
## 0 335 -0.9339 -2.812 -0.008945 -2.2509
## 1 335 -0.9339 -2.812 -0.026392 -2.2509
Because this is a linear model, the marginal effects of
cyl
and wt
do not vary across levels of
am
or hp
. The minimum and Q1 value of
hp
are also the same, so the marginal effects of
am
are the same in the first two results. As you can see,
however, the marginal effect of hp
differs when
am == 0
versus am == 1
(first and second rows)
and the marginal effect of am
differs across levels of
hp
(e.g., between the first and third rows). As should be
clear, the at
argument is incredibly useful for getting a
better grasp of the marginal effects of different covariates.
This becomes especially apparent when a model includes power-terms (or any other alternative functional form of a covariate). Consider, for example, the simple model of fuel economy as a function of weight, with weight included as both a first- and second-order term:
##
## Call:
## lm(formula = mpg ~ wt + I(wt^2), data = mtcars)
##
## Residuals:
## Min 1Q Median 3Q Max
## -3.483 -1.998 -0.773 1.462 6.238
##
## Coefficients:
## Estimate Std. Error t value Pr(>|t|)
## (Intercept) 49.9308 4.2113 11.856 1.21e-12 ***
## wt -13.3803 2.5140 -5.322 1.04e-05 ***
## I(wt^2) 1.1711 0.3594 3.258 0.00286 **
## ---
## Signif. codes: 0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1
##
## Residual standard error: 2.651 on 29 degrees of freedom
## Multiple R-squared: 0.8191, Adjusted R-squared: 0.8066
## F-statistic: 65.64 on 2 and 29 DF, p-value: 1.715e-11
Looking only at the regression results table, it is actually quite
difficult to understand the effect of wt
on fuel economy
because it requires performing mental multiplication and addition on all
possible values of wt
. Using the at
option to
margins, you could quickly obtain a sense of the average marginal effect
of wt
at a range of plausible values:
## Average marginal effects at specified values
## lm(formula = mpg ~ wt + I(wt^2), data = mtcars)
## at(wt) wt
## 1.513 -9.8366
## 2.542 -7.4254
## 3.325 -5.5926
## 3.650 -4.8314
## 5.424 -0.6764
The marginal effects in the first column of results reveal that the
average marginal effect of wt
is large and negative except
when wt
is very large, in which case it has an effect not
distinguishable from zero. We can easily plot these results using the
cplot()
function to see the effect visually in terms of
either predicted fuel economy or the marginal effect of
wt
:
A really nice feature of Stata’s margins command is that it handles
factor variables gracefully. This functionality is difficult to emulate
in R, but the margins()
function does its best. Here we see
the marginal effects of a simple regression that includes a factor
variable:
## Average marginal effects
## lm(formula = mpg ~ factor(cyl) * hp + wt, data = mtcars)
## hp wt cyl6 cyl8
## -0.04475 -3.06 1.473 0.8909
margins()
recognizes the factor and displays the
marginal effect for each level of the factor separately. (Caveat: this
may not work with certain at
specifications, yet.)
Stata’s margins
command includes an over()
option, which allows you to very easily calculate marginal effects on
subsets of the data (e.g., separately for men and women). This is useful
in Stata because the program only allows one dataset in memory. Because
R does not impose this restriction and further makes subsetting
expressions very simple, that feature is not really useful and can be
achieved using standard subsetting notation in R:
x <- lm(mpg ~ factor(cyl) * am + hp + wt, data = mtcars)
# automatic vehicles
margins(x, data = mtcars[mtcars$am == 0, ])
## Average marginal effects
## lm(formula = mpg ~ factor(cyl) * am + hp + wt, data = mtcars)
## am hp wt cyl6 cyl8
## 1.706 -0.03115 -2.441 -1.715 -1.586
## Average marginal effects
## lm(formula = mpg ~ factor(cyl) * am + hp + wt, data = mtcars)
## am hp wt cyl6 cyl8
## 2.167 -0.03115 -2.441 -4.218 -2.656
Because a "margins"
object is just a data frame, it is
also possible to obtain the same result by subsetting the
output of margins()
:
## $`0`
## Average marginal effects
## lm(formula = mpg ~ factor(cyl) * am + hp + wt, data = mtcars)
## am hp wt cyl6 cyl8
## 1.706 -0.03115 -2.441 -1.715 -1.586
##
## $`1`
## Average marginal effects
## lm(formula = mpg ~ factor(cyl) * am + hp + wt, data = mtcars)
## am hp wt cyl6 cyl8
## 2.167 -0.03115 -2.441 -4.218 -2.656
Using margins()
to calculate marginal effects enables
several kinds of plotting. The built-in plot()
method for
objects of class "margins"
creates simple diagnostic plots
for examining the output of margins()
in visual rather than
tabular format. It is also possible to use the output of
margins()
to produce more typical marginal effects plots
that show the marginal effect of one variable across levels of another
variable. This section walks through the plot()
method and
then shows how to produce marginal effects plots using base
graphics.
plot()
method for “margins” objectsThe margins package implements a plot()
method for objects of class "margins"
(seen above). This
produces a plot similar (in spirit) to the output of Stata’s
marginsplot
. It is highly customizable, but is meant
primarily as a diagnostic tool to examine the results of
margins()
. It simply produces, by default, a plot of
marginal effects along with 95% confidence intervals for those effects.
The confidence level can be modified using the levels
argument, which is vectorized to allow multiple levels to be specified
simultaneously.
There are two common ways of visually representing the substantive results of a regression model: (1) fitted values plots, which display the fitted conditional mean outcome across levels of a covariate, and (2) marginal effects plots, which display the estimated marginal effect of a variable across levels of a covariate. This section discusses both approaches.
Fitted value plots can be created using cplot()
(to
provide conditional predicted value plots or
conditional effect plots) and both the persp()
method and image()
method for "lm"
objects,
which display the same type of relationships in three-dimensions (i.e.,
across two conditioning covariates).
For example, we can use cplot()
to quickly display the
predicted fuel economy of a vehicle from a model:
The slopes of the predicted value lines are the marginal effect of
wt
when am == 0
and am == 1
. We
can obtain these slopes using margins()
and specifying the
at
argument:
## Average marginal effects at specified values
## lm(formula = mpg ~ cyl + wt * am, data = mtcars)
## at(am) cyl wt am
## 0 -1.181 -2.369 -1.566
## 1 -1.181 -6.566 -1.566
Another plotting function - the persp()
method for “lm”
objects - gives even more functionality:
This three-dimensional representation can be easily rotated using the
theta
and phi
arguments. If multiple values
are specified for each, facetting is automatically provided:
Because these three-dimensional representations can be challenging to
interpret given the distortion of displaying a three-dimensional surface
in two dimensions, the image()
method provides a “flat”
alternative to convey the same information using the same semantics:
One of principal motives for developing margins is to facilitate the substantive interpretation of interaction terms in regression models. A large literature now describes the difficulties of such interpretations in both linear and non-linear regression models. This vignette walks through some of that interpretation.
If we begin with a simple example of a regression model with an
interaction term, the difficulties of interpretation become almost
immediately clear. In this first model, we’ll use the
mtcars
dataset to understand vehicle fuel economy as a
function of drat
(rear axle ratio), wt
(weight), and their interaction. As Brambor et al. (2006) make clear,
the most common mistake in such models is estimating the model without
the constituent variables. We can see why this is a problem by
estimating the model with and without the constituent terms:
##
## Call:
## lm(formula = mpg ~ drat:wt, data = mtcars)
##
## Residuals:
## Min 1Q Median 3Q Max
## -7.3319 -2.3669 -0.6585 2.0952 8.2124
##
## Coefficients:
## Estimate Std. Error t value Pr(>|t|)
## (Intercept) 40.6488 3.1335 12.972 7.77e-14 ***
## drat:wt -1.8339 0.2728 -6.723 1.89e-07 ***
## ---
## Signif. codes: 0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1
##
## Residual standard error: 3.87 on 30 degrees of freedom
## Multiple R-squared: 0.6011, Adjusted R-squared: 0.5878
## F-statistic: 45.2 on 1 and 30 DF, p-value: 1.886e-07
##
## Call:
## lm(formula = mpg ~ drat * wt, data = mtcars)
##
## Residuals:
## Min 1Q Median 3Q Max
## -3.8913 -1.8634 -0.3398 1.3247 6.4730
##
## Coefficients:
## Estimate Std. Error t value Pr(>|t|)
## (Intercept) 5.550 12.631 0.439 0.6637
## drat 8.494 3.321 2.557 0.0162 *
## wt 3.884 3.798 1.023 0.3153
## drat:wt -2.543 1.093 -2.327 0.0274 *
## ---
## Signif. codes: 0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1
##
## Residual standard error: 2.839 on 28 degrees of freedom
## Multiple R-squared: 0.7996, Adjusted R-squared: 0.7782
## F-statistic: 37.25 on 3 and 28 DF, p-value: 6.567e-10
Clearly the models produce radically different estimates and
goodness-of-fit to the original data. As a result, it’s important to use
all constituent terms in the model even if they are thought a
priori to have coefficients of zero. Now let’s see how we can use
margins()
to interpret a more complex three-way
interaction:
## factor AME SE z p lower upper
## am -2.4729 3.0354 -0.8147 0.4152 -8.4223 3.4764
## drat 0.8882 1.7047 0.5210 0.6024 -2.4530 4.2294
## wt -5.8203 0.8769 -6.6370 0.0000 -7.5391 -4.1015
By default, margins()
will supply the average marginal
effects of the constituent variables in the model. Note how the
drat:wt
term is not expressed in the margins results. This
is because the contribution of the drat:wt
term is
incorporated into the marginal effects for the constituent variables.
Because there is a significant interaction, we can see this by examining
margins at different levels of the constituent variables. The
drat
variable is continuous, taking values from 2.76 to
4.93:
## Average marginal effects at specified values
## lm(formula = mpg ~ drat * wt * am, data = mtcars)
## at(drat) drat wt am
## 2.76 0.8882 -4.989 -0.04699
## 4.93 0.8882 -7.271 -6.09246
Now margins()
returns two "margins"
objects, one for each combination of values specified in
drat
. We can see in the above that cars with a low axle
ratio (drat == 2.76
), the average marginal effect of weight
is a reduction in fuel economy of 4.99 miles per gallon. For vehicles
with a higher ratio (drat == 4.93
), this reduction in fuel
economy is lower at 7.27 miles per gallon. Yet this is also not fully
accurate because it mixes the automatic and manual cars, so we may want
to further break out the results by transmission type.
The at
argument accepts multiple named combinations of
variables, so we can specify particular values of both drat
and wt
and am
for which we would like to
understand the marginal effect of each variable. For example, we might
want to look at the effects of each variable for vehicles with varying
axle ratios but also across some representative values of the
wt
distribution, separately for manual and automatic
vehicles. (Note that the order of values in the at
object
does matter and it can inclue variables that are not explicitly being
modelled).
wts <- prediction::seq_range(mtcars$wt, 10)
m1 <- margins(x1, at = list(wt = wts, drat = range(mtcars$drat), am = 0:1))
nrow(m1)/nrow(mtcars)
## [1] 40
As you can see above, the result is a relatively long data frame
(basically a stack of "margins"
objects specific to each of
the - 40 - requested combinations of at
values). We can
examine the whole stack using summary()
, but it’s an
extremely long output (one row for each estimate times the number of
unique combinations of at
values, so 120 rows in this
instance).
An easier way to understand all of this is to use graphing. The
cplot()
function, in particular, is useful here. We can,
for example, plot the marginal effect of axle ratio across levels of
vehicle weight, separately for automatic vehicles (in red) and manual
vehicles (in blue):
cplot(x1, x = "wt", dx = "drat", what = "effect",
data = mtcars[mtcars[["am"]] == 0,],
col = "red", se.type = "shade",
xlim = range(mtcars[["wt"]]), ylim = c(-20, 20),
main = "AME of Axle Ratio on Fuel Economy")
cplot(x1, x = "wt", dx = "drat", what = "effect",
data = mtcars[mtcars[["am"]] == 1,],
col = "blue", se.type = "shade",
draw = "add")
cplot()
only provides AME displays over the observed
range of the data (noted by the rug), so we can see that it would be
inappropriate to compare the average marginal effect of axle ratio for
the two transmission types at extreme values of weight.
Another use of these types of plots can be to interpret power terms, to see the average marginal effect of a variable across values of itself. Consider the following, for example:
x1b <- lm(mpg ~ am * wt + am * I(wt^2), data = mtcars)
cplot(x1b, x = "wt", dx = "wt", what = "effect",
data = mtcars[mtcars[["am"]] == 0,],
col = "red", se.type = "shade",
xlim = range(mtcars[["wt"]]), ylim = c(-20, 20),
main = "AME of Weight on Fuel Economy")
cplot(x1b, x = "wt", dx = "wt", what = "effect",
data = mtcars[mtcars[["am"]] == 1,],
col = "blue", se.type = "shade",
draw = "add")
Note, however, that interpreting these kind of
continuous-by-continuous interaction terms is slightly more complex
because the marginal effect of both constituent variables always depends
on the level of the other variable. We’ll use the horsepower
(hp
) variable from mtcars
to understand this
type of interaction. We can start by looking at the AMEs:
## Average marginal effects
## lm(formula = mpg ~ hp * wt, data = mtcars)
## hp wt
## -0.03051 -4.132
On average across the cases in the dataset, the effect of horsepower
is slightly negative. On average, the effect of weight is also negative.
Both decrease fuel economy. But what is the marginal effect of each
variable across the range of values we actually observe in the data? To
get a handle on this, we can use the persp()
method
provided by margins.
To make sense of this set of plots (actually, the same plot seen from four different angles), it will also be helpful to have the original regression results close at-hand:
##
## Call:
## lm(formula = mpg ~ hp * wt, data = mtcars)
##
## Residuals:
## Min 1Q Median 3Q Max
## -3.0632 -1.6491 -0.7362 1.4211 4.5513
##
## Coefficients:
## Estimate Std. Error t value Pr(>|t|)
## (Intercept) 49.80842 3.60516 13.816 5.01e-14 ***
## hp -0.12010 0.02470 -4.863 4.04e-05 ***
## wt -8.21662 1.26971 -6.471 5.20e-07 ***
## hp:wt 0.02785 0.00742 3.753 0.000811 ***
## ---
## Signif. codes: 0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1
##
## Residual standard error: 2.153 on 28 degrees of freedom
## Multiple R-squared: 0.8848, Adjusted R-squared: 0.8724
## F-statistic: 71.66 on 3 and 28 DF, p-value: 2.981e-13
If we express the regression results as an equation:
mpg = 49.81 + (-0.12 * hp) + (-8.22 * wt) + (0.03 * hp*wt)
,
it will be easy to see how the three-dimensional surface above reflects
various partial derivatives of the regression equation.
For example, if we take the partial derivative of the regression
equation with respect to wt
(i.e., the marginal effect of
weight), the equation is:
d_mpg/d_wt = (-8.22) + (0.03 * hp)
. This means that the
marginal effect of weight is large and negative when horsepower is zero
(which never occurs in the mtcars
dataset) and decreases in
magnitude and becoming more positive as horsepower increases. We can see
this in the above graph that the marginal effect of weight is constant
across levels of weight
because wt
does not
enter into the partial derivative. Across levels of horsepower, however,
the marginal effect becomes more positive. This is clear looking at the
“front” or “back” edges of the surface, which are straight-linear
increases. The slope of those edges is 0.03 (the coefficient on the
interaction term).
If we then take the partial derivative with respect to
hp
(to obtain the marginal effect of horsepower), the
equation is: d_mpg/d_hp = (-0.12) + (0.03 * wt)
. When
wt
is zero, this partial derivative (or marginal effect) is
-0.12 miles/gallon. The observed range of wt
, however, is
only: 1.513, 5.424. We can see these results in the analogous graph of
the marginal effects of horsepower (below). The “front” and “back” edges
of the graph are now flat (reflecting how the marginal effect of
horsepower is constant across levels of horsepower), while the
“front-left” and “right-back” edges of the surface are lines with slope
0.03, reflecting the coefficient on the interaction term.
An alternative way of plotting these results is to take “slices” of the three-dimensional surface and present them in a two-dimensional graph, similar to what we did above with the indicator-by-continuous approach. That strategy would be especially appropriate for a categorical-by-continuous interaction where the categories of the first variable did not necessarily have a logical ordering sufficient to draw a three-dimensional surface.
It should be noted that cplot()
returns a fairly tidy
data frame, making it possible to use ggplot2 as an alternative plotting
package to display the kinds of relationships of typical interest. For
example, returning to our earlier example:
## xvals yvals upper lower
## 1 1.513000 28.66765 33.04903 24.28627
## 2 1.675958 27.71126 31.71243 23.71010
## 3 1.838917 26.75488 30.38319 23.12656
## 4 2.001875 25.79849 29.06385 22.53313
## 5 2.164833 24.84210 27.75810 21.92610
## 6 2.327792 23.88571 26.47147 21.29996
From this structure, it is very easy to draw a predicted values plot
library("ggplot2")
ggplot(cdat, aes(x = xvals)) +
geom_line(aes(y = yvals)) +
geom_line(aes(y = upper), linetype = 2) +
geom_line(aes(y = lower), linetype = 2) +
geom_hline(yintercept = 0) +
ggtitle("Predicted Fuel Economy (mpg) by Weight") +
xlab("Weight (1000 lbs)") + ylab("Predicted Value")
And the same thing is possible with a marginal effect calculation:
cdat <- cplot(x1, "wt", "drat", what = "effect", draw = FALSE)
ggplot(cdat, aes(x = xvals)) +
geom_line(aes(y = yvals)) +
geom_line(aes(y = upper), linetype = 2) +
geom_line(aes(y = lower), linetype = 2) +
geom_hline(yintercept = 0) +
ggtitle("AME of Axle Ratio on Fuel Economy (mpg) by Weight") +
xlab("Weight (1000 lbs)") + ylab("AME of Axle Ratio")
Interaction terms in generalized linear models have been even more
controversial than interaction terms in linear regression (Norton et
al. 2004). The result is a realization over the past decade that almost
all extant interpretations of GLMs with interaction terms (or
hypothesizing moderating effects) have been misinterpreted. Stata’s
margins
command and the debate that preceded it have led to
a substantial change in analytic practices.
For this, we’ll use the Pima.te
dataset from
MASS, which should be preinstalled in R, but we’ll
manually load it just in case:
## npreg glu bp skin bmi ped age type
## 1 6 148 72 35 33.6 0.627 50 Yes
## 2 1 85 66 29 26.6 0.351 31 No
## 3 1 89 66 23 28.1 0.167 21 No
## 4 3 78 50 32 31.0 0.248 26 Yes
## 5 2 197 70 45 30.5 0.158 53 Yes
## 6 5 166 72 19 25.8 0.587 51 Yes
This dataset contains data on 332 women, including whether or not
they are diabetic (type
). We’ll examine a simple model with
an interaction term between age and a skin thickness measure to explain
diabetes status in these women:
##
## Call:
## glm(formula = type ~ age * skin, family = binomial, data = Pima.te)
##
## Coefficients:
## Estimate Std. Error z value Pr(>|z|)
## (Intercept) -7.880929 1.469932 -5.361 8.26e-08 ***
## age 0.164912 0.042398 3.890 0.00010 ***
## skin 0.177608 0.044946 3.952 7.77e-05 ***
## age:skin -0.003628 0.001316 -2.756 0.00585 **
## ---
## Signif. codes: 0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1
##
## (Dispersion parameter for binomial family taken to be 1)
##
## Null deviance: 420.3 on 331 degrees of freedom
## Residual deviance: 364.1 on 328 degrees of freedom
## AIC: 372.1
##
## Number of Fisher Scoring iterations: 4
Logit models (like all GLMs) present the dual challenges of having
coefficients that are directly uninterpretable and marginal effects that
depend on the values of the data. As a result, we can see coefficients
and statistical significance tests above but it’s hard to make sense of
those results without converting them into a more intuitive quantity,
such as the predicted probability of having diabetes. We can see that
increasing age and increasing skin thickness are associated with higher
rates of diabetes, but the negative coefficient on the interaction term
makes it hard to express the substantive size of these relationships. We
can use margins
to achieve this. By default, however,
margins()
reports results on the response scale (thus
differing from the default behavior of
stats::predict()
):
## Average marginal effects
## glm(formula = type ~ age * skin, family = binomial, data = Pima.te)
## age skin
## 0.009283 0.01064
These marginal effects reflect, on average across all of the cases in our data, how much more likely a woman is to have diabetes. Because this is an instantaneous effect, it can be a little hard to conceptualize. I find it helpful to take a look at a predicted probability plot to understand what is going on. Let’s take a look, for example, at the effect of age on the probability of having diabetes:
The above graph shows that as age increase, the probability of having
diabetes increases. When a woman is 20 years old, the probability is
about .20 whereas when a woman is 80, the probability is about 0.80. In
essence, the marginal effect of age
reported above is the
slope of this predicted probability plot at the mean age of women in the
dataset (which is 31.3162651). Clearly, this quantity is useful (it’s
the impact of age for the average-aged woman) but the logit curve shows
that the marginal effect of age
differs considerably across
ages and, as we know from above with linear models, also depends on the
other variable in the interaction (skin thickness). To really understand
what is going on, we need to graph the data. Let’s look at the
perspective plot like the one we drew for the OLS model, above:
This graph is much more complicated than the analogous graph for an OLS model, this is because of the conversion between the log-odds scale of the linear predictors and the distribution of the data. What we can see is that the average marginal effect of age (across all of the women in our dataset) is highest when a woman is middle age and skin thickness is low. In these conditions, the marginal change in the probability of diabetes is about 3%, but the AME of age is nearly zero at higher and lower ages. Indeed, as skin thickness increases, the marginal effect of age flattens out and actually becomes negative (such that increasing age actually decreases the probability of diabetes for women with thick arms).
Now let’s examine the average marginal effects of skin thickness across levels of age and skin thickness:
This graphs is somewhat flatter, indicating that the average marginal effects of skin thickness vary less than those for age. Yet, the AME of skin thickness is as high as 2% when age is low and skin thickness is high. Interestingly, however, this effect actually becomes negative as age increases. The marginal effect of skin thickness is radically different for young and old women.
For either of these cases, it would also be appropriate to draw two-dimensional plots of “slides” of these surfaces and incorporate measures of uncertainty. It is extremely difficult to convey standard errors or confidence intervals in a three-dimensional plot, so those could be quite valuable.
It is also worth comparing the above graphs to those that would
result from a model without the interaction term between
age
and skin
. It looks quite different (note,
however, that it is also drawn from a higher perspective to better see
the shape of the surface):
Our inferences about the impact of a variable on the outcome in a GLM therefore depend quite heavily on how the interaction is modelled. It also worth pointing out that the surface of the AMEs on the log-odds (linear) scale are actually flat (as in an OLS model), so the the curved shape of the plot immediately above reflects only the conversion of the effects from the linear scale to the probability scale (and the distributions of the covariates), whereas the much more unusual surface from earlier reflects that conversion and the interaction between the two variables (and the distributions thereof). If we were to plot the interaction model on the scale of the log-odds, it would look much more like the plot from the OLS models (this is left as an exercise to the reader).
Bartus, Tamas. 2005. “Estimation of marginal effects using margeff.” Stata Journal 5: 309-329.
Brambor, Thomas, William Clark & Matt Golder. 2006. “Understanding Interaction Models: Improving Empirical Analyses.” Political Analysis 14: 63-82.
Greene, William H. 2012. Econometric Analysis. 7th ed. Upper Saddle River, NJ: Prentice Hall.
Long, J. Scott. 1997. Regression Models for Categorical and Limited Dependent Variables. Sage Publications: Thousand Oaks, CA.
Norton, Edward C., Hua Wang, and Chunrong Ai. 2004. “Computing interaction effects and standard errors in logit and probit models.” The Stata Journal 4(2): 154-167.
Stata Corporation. “margins”. Software Documentation. Available from: https://www.stata.com/manuals13/rmargins.pdf.
Williams, Richard. 2012. “Using the margins command to estimate and interpret adjusted predictions and marginal effects.” Stata Journal 12: 308-331.