Rank deficiency in mixed-effects models
The (column) rank of a matrix refers to the number of linearly independent columns in the matrix. Clearly, the rank can never be more than the number of columns; however, the rank can be less than the number of columns. In a regression context, this corresponds to a (linear) dependency in the predictors. The simplest case of rank deficiency is a duplicated predictor or a predictor that is exactly a multiple of another predictor. However, rank deficiency can also arise in more subtle ways, such as from missing cells in a two-factor experimental design. Rank deficiency can also arise as an extreme case of multicollinearity. In all cases, it is important to remember that we can only assess the numerical rank of a matrix, which may be less than its theoretical rank, and that evaluation of this numerical rank requires setting some numerical tolerance levels. These choices are not always well defined. In other words, the rank of a matrix is well-defined in theory but in practice can be difficult to evaluate.
Rank deficiency can occur in two ways in mixed-effects models: in the fixed effects and in the random effects. The implications of rank deficiency and thus the handling of it differ between these.
Fixed effects
The consequences of rank deficiency in the fixed effects are similar to those in classical ordinary least squares (OLS) regression. If one or more predictors can be expressed as a linear combination of the other columns, then this column is redundant and the model matrix is rank deficient. Note however, that the redundant column is not defined uniquely. For example, in the case that of two columns a
and b
where b = 2a
, then the rank deficiency can be handled by eliminating either a
or b
. While we defined b
here in terms of a
, it may be that b
is actually the more 'fundamental' predictor and hence we may define a
in terms of b
as a = 0.5b
. The user may of course possess this information, but the choice is not apparent to the modelling software. As such, the handling of rank deficiency in MixedModels.jl
should not be taken as a replacement for thinking about the nature of the predictors in a given model.
There is a widely accepted convention for how to make the coefficient estimates for these redundant columns well-defined: we set their value to zero and their standard errors to NaN
(and thus also their $z$ and $p$-values). The values that have been defined to be zero, as opposed to evaluating to zero, are displayed as -0.0
as an additional visual aid to distinguish them from the other coefficients. In practice the determination of rank and the redundant coefficients is done via a 'pivoting' scheme during a decomposition to move the surplus columns to the right side of the model matrix. In subsequent calculations, these columns are effectively ignored (as their estimates are zero and thus won't contribute to any other computations). For display purposes, this pivoting is unwound when the coef
values are displayed.
Both the pivoted and unpivoted coefficients are available in MixedModels. The fixef
extractor returns the pivoted, truncated estimates (i.e. the non redundant terms), while the coef
extractor returns the unpivoted estimates (i.e. all terms, included the redundant ones). The same holds for the associated fixefnames
and coefnames
.
Pivoting is platform dependent
In MixedModels.jl, we use standard numerical techniques to detect rank deficiency. We currently offer no guarantees as to which exactly of the standard techniques (pivoted QR decomposition, pivoted Cholesky decomposition, etc.) will be used. This choice should be viewed as an implementation detail. Similarly, we offer no guarantees as to which of columns will be treated as redundant. This choice may vary between releases and even between platforms (both in broad strokes of "Linux" vs. "Windows" and at the level of which BLAS options are loaded on a given processor architecture) for the same release. In other words, you should not rely on the order of the pivoted columns being consistent! when you switch to a different computer or a different operating system. If consistency in the pivoted columns is important to you, then you should instead determine your rank ahead of time and remove extraneous columns / predictors from your model specification.
This lack of consistency guarantees arises from a more fundamental issue: numeric linear algebra is challenging and sensitive to the underlying floating point operations. Due to rounding error, floating point arithmetic is not associative:
0.1 + 0.1 + 0.1 - 0.3 == 0.1 + 0.1 + (0.1 - 0.3)
false
This means that "nearly" / numerically rank deficient matrices may or may not be detected as rank deficient, depending on details of the platform. Determining the rank of a matrix is the type of problem that is well-defined in theory but not in practice.
Currently, a coarse heuristic is applied to reduce the chance that the intercept column will be pivoted, but even this behavior is not guaranteed.
Undetected Rank Deficiency
Undetected rank deficiency in the fixed effects will lead to numerical issues, such as nonsensical estimates. A PosDefException
may indicate rank deficiency because the covariance matrix will only be positive semidefinite and not positive definite (see Details of the parameter estimation). In other words, checking that the fixed effects are full rank is a great first step in debugging a PosDefException
.
Note that PosDefException
is not specific to rank deficiency and may arise in other ill-conditioned models. In any case, examining the model specification and the data to verify that they work together is the first step. For generalized linear mixed-effects models, it may also be worthwhile to try out fast=true
instead of the default fast=false
. See this GitHub issue and linked Discourse discussion for more information.
Random effects
Rank deficiency presents less of a problem in the random effects than in the fixed effects because the "estimates" (more formally, the conditional modes of the random effects given the observed data) are determined as the solution to a penalized least squares problem. The shrinkage effect which moves the conditional modes (group-level predictions) towards the grand mean is a form of regularization, which provides well-defined "estimates" for overparameterized models. (For more reading on this general idea, see also this blog post on the model complexity myth.)
The nature of the penalty in the penalized least squares solution is such that the "estimates" are well-defined even when the covariance matrix of the random effects converges to a "singular" or "boundary" value. In other words, singularity of the covariance matrix for the random effects, which means that there are one or more directions in which there is no variability in the random effects, is different from singularity of the model matrix for the random effects, which would affect the ability to define uniquely these coefficients. The penalty term always provides a unique solution for the random-effects coefficients.
In addition to handling naturally occurring rank deficiency in the random effects, the regularization allows us to fit explicitly overparameterized random effects. For example, we can use fulldummy
to fit both an intercept term and $n$ indicator variables in the random effects for a categorical variable with $n$ levels instead of the usual $n-1$ contrasts.
kb07 = MixedModels.dataset(:kb07)
contrasts = Dict(var => HelmertCoding() for var in (:spkr, :prec, :load))
fit(MixedModel, @formula(rt_raw ~ spkr * prec * load + (1|subj) + (1+prec|item)), kb07; contrasts=contrasts)
Linear mixed model fit by maximum likelihood
rt_raw ~ 1 + spkr + prec + load + spkr & prec + spkr & load + prec & load + spkr & prec & load + (1 | subj) + (1 + prec | item)
logLik -2 logLik AIC AICc BIC
-14846.7248 29693.4496 29719.4496 29719.6547 29790.8120
Variance components:
Column Variance Std.Dev. Corr.
item (Intercept) 158695.302 398.366
prec: maintain 93637.344 306.002 -0.81
subj (Intercept) 105909.620 325.438
Residual 842822.907 918.054
Number of obs: 1789; levels of grouping factors: 32, 56
Fixed-effects parameters:
──────────────────────────────────────────────────────────────────────────────
Coef. Std. Error z Pr(>|z|)
──────────────────────────────────────────────────────────────────────────────
(Intercept) 2225.71 85.5665 26.01 <1e-99
spkr: old 74.1916 21.7062 3.42 0.0006
prec: maintain -377.212 58.2866 -6.47 <1e-10
load: yes 101.958 21.7062 4.70 <1e-05
spkr: old & prec: maintain -28.0275 21.7062 -1.29 0.1966
spkr: old & load: yes 26.8642 21.7062 1.24 0.2159
prec: maintain & load: yes -18.6514 21.7062 -0.86 0.3902
spkr: old & prec: maintain & load: yes 15.4985 21.7062 0.71 0.4752
──────────────────────────────────────────────────────────────────────────────
fit(MixedModel, @formula(rt_raw ~ spkr * prec * load + (1|subj) + (1+fulldummy(prec)|item)), kb07; contrasts=contrasts)
Linear mixed model fit by maximum likelihood
rt_raw ~ 1 + spkr + prec + load + spkr & prec + spkr & load + prec & load + spkr & prec & load + (1 | subj) + (1 + prec | item)
logLik -2 logLik AIC AICc BIC
-14846.7248 29693.4496 29725.4496 29725.7566 29813.2802
Variance components:
Column Variance Std.Dev. Corr.
item (Intercept) 1225798.531 1107.158
prec: break 493552.843 702.533 -0.82
prec: maintain 1006118.561 1003.055 -0.98 +0.80
subj (Intercept) 105911.865 325.441
Residual 842822.319 918.054
Number of obs: 1789; levels of grouping factors: 32, 56
Fixed-effects parameters:
──────────────────────────────────────────────────────────────────────────────
Coef. Std. Error z Pr(>|z|)
──────────────────────────────────────────────────────────────────────────────
(Intercept) 2225.71 85.567 26.01 <1e-99
spkr: old 74.1916 21.7062 3.42 0.0006
prec: maintain -377.212 58.287 -6.47 <1e-10
load: yes 101.958 21.7062 4.70 <1e-05
spkr: old & prec: maintain -28.0275 21.7062 -1.29 0.1966
spkr: old & load: yes 26.8642 21.7062 1.24 0.2159
prec: maintain & load: yes -18.6514 21.7062 -0.86 0.3902
spkr: old & prec: maintain & load: yes 15.4985 21.7062 0.71 0.4752
──────────────────────────────────────────────────────────────────────────────
This may be useful when the PCA
property suggests a random effects structure larger than only main effects but smaller than all interaction terms. This is also similar to the functionality provided by dummy
in lme4
, but as in the difference between zerocorr
in Julia and ||
in R, there are subtle differences in how this expansion interacts with other terms in the random effects.