从混合模型中获取随机效应矩阵

时间:2021-02-02 20:44:44

标签: r regression lme4 mixed-models nlme

在我下面的代码中,我想知道如何从 out 中的 Ts 对象获取 lme()library(nlme) 的等效项?

dat <- read.csv("https://raw.githubusercontent.com/rnorouzian/v/main/mv.l.csv")

library(lme4)

x <- lmer(value ~0 + name+ (1| School/Student), data = dat,
          control = lmerControl(check.nobs.vs.nRE= "ignore"))

    lwr <- getME(x, "lower")
    theta <- getME(x, "theta")

    out = any(theta[lwr == 0] < 1e-4)  # find this from `x1` object below
     Ts = getME(x, "Tlist")            # find this from `x1` object below


# Fitting the above model using `library(nlme)`:

library(nlme)

x1 <- lme(value ~0 + name, random = ~1| School/Student, data = dat)

1 个答案:

答案 0 :(得分:1)

我强烈建议阅读文档! lme4 和 nlme 使用固有不同的方法来拟合混合模型——lme4 使用基于较低 Cholesky 因子 (theta) 的惩罚最小二乘公式,而 nlme4 使用广义最小二乘公式,可以选择存储为 Cholesky 因子——但是他们的文档为您提供了从内部表示中获取所需信息的信息。之后,由您来进行数学运算以在表示之间进行转换。

如果你做 ?lme 那么有一行

<块引用>

请参阅 lmeObject 以了解拟合的组成部分

然后您执行 ?lmeObject,您会在其中找到两个有希望的条目:

<块引用>

apVar
方差-协方差系数的近似协方差矩阵。如果在调用 apVar = FALSE 时使用的控件值中有 lme,则该组件为 NULL

<块引用>

modelStruct 从类 lmeStruct 继承的对象,表示混合效果模型组件的列表,例如 reStructcorStructvarFunc 对象。

好吧,我们实际上并不想要 var-cov 系数,而是想要随机效应矩阵。所以我们可以看看reStruct。这在 nlme 中比 lme4 灵活得多,但通常只是随机效应矩阵。要执行与 lme4 相当的任何操作,您需要将它们转换为较低的 Cholesky 因子。以下是使用 sleepstudy 数据的示例:

> library("nlme")
> library("lme4")
> 
> data("sleepstudy")
> m_nlme <- lme(fixed=Reaction ~ 1 + Days,
+               random=~ 1 + Days | Subject,
+               data=sleepstudy,
+               method = "ML")
> m_lme4 <- lmer(Reaction ~ 1 + Days + (1 + Days|Subject),
+                data=sleepstudy,
+                REML=FALSE)
> 
> re_lme4 <- getME(m_lme4, "Tlist")$Subject
> print(re_lme4)
           [,1]      [,2]
[1,] 0.92919061 0.0000000
[2,] 0.01816575 0.2226432
> 
> re_nlme <- m_nlme$modelStruct$reStruct$Subject
> # entire covariance matrix
> print(re_nlme)
Positive definite matrix structure of class pdLogChol representing
            (Intercept)       Days
(Intercept)  0.86344433 0.01688228
Days         0.01688228 0.04990040
> # get the lower cholesky factor
> re_nlme <- t(chol(re_nlme)) # could also use pdMatrix(re_nlme, TRUE)
> print(re_nlme)
            (Intercept)      Days
(Intercept)  0.92921705 0.0000000
Days         0.01816829 0.2226439

lme4 的 theta 向量只是给定分组变量下 Cholesky 因子的下三角形的行主要表示。 (对于具有多个分组变量的模型,您只需将它们连接在一起。)较低的 Cholesky 因子被约束为对角线上的条目不小于零(因为这将对应于负方差),否则不受约束。换句话说,对角线条目的下限为 0,所有其他条目的下限为 -Inf。

所以,在 lme4 中:

> re_lme4[lower.tri(re_lme4,diag = TRUE)]
[1] 0.92919061 0.01816575 0.22264321
> getME(m_lme4, "theta")
     Subject.(Intercept) Subject.Days.(Intercept)             Subject.Days 
              0.92919061               0.01816575               0.22264321 
> getME(m_lme4, "lower")
[1]    0 -Inf    0

我们可以为 nlme 实现这个(不是最有效的方法,但它显示了事物是如何构建的):

> lowerbd <- function(x){
+   dd <- diag(0, nrow=nrow(x))
+   dd[lower.tri(dd)] <- -Inf
+   dd[lower.tri(dd, diag=TRUE)]
+ }
> lowerbd(re_nlme)
[1]    0 -Inf    0
> lowerbd(re_lme4)
[1]    0 -Inf    0

请注意,这是 nlme 实际上比 lme4 更强大的地方:整个 pdMatrix 限制集可以为不同的条目创建不同的下限(以及例如约束条目之间的关系)。