NAG Library Routine Document
g02jaf
(mixeff_reml)
1
Purpose
g02jaf fits a linear mixed effects regression model using restricted maximum likelihood (REML).
2
Specification
Fortran Interface
Subroutine g02jaf ( |
n,
ncol,
lddat,
dat,
levels,
yvid,
cwid,
nfv,
fvid,
fint,
nrv,
rvid,
nvpr,
vpr,
rint,
svid,
gamma,
nff,
nrf,
df,
reml,
lb,
b,
se,
maxit,
tol,
warn,
ifail) |
Integer, Intent (In) | :: |
n,
ncol,
lddat,
levels(ncol),
yvid,
cwid,
nfv,
fvid(nfv),
fint,
nrv,
rvid(nrv),
nvpr,
vpr(nrv),
rint,
svid,
lb,
maxit | Integer, Intent (Inout) | :: |
ifail | Integer, Intent (Out) | :: |
nff,
nrf,
df,
warn | Real (Kind=nag_wp), Intent (In) | :: |
dat(lddat,ncol),
tol | Real (Kind=nag_wp), Intent (Inout) | :: |
gamma(nvpr+2) | Real (Kind=nag_wp), Intent (Out) | :: |
reml,
b(lb),
se(lb) |
|
C Header Interface
#include nagmk26.h
void |
g02jaf_ (
const Integer *n,
const Integer *ncol,
const Integer *lddat,
const double dat[],
const Integer levels[],
const Integer *yvid,
const Integer *cwid,
const Integer *nfv,
const Integer fvid[],
const Integer *fint,
const Integer *nrv,
const Integer rvid[],
const Integer *nvpr,
const Integer vpr[],
const Integer *rint,
const Integer *svid,
double gamma[],
Integer *nff,
Integer *nrf,
Integer *df,
double *reml,
const Integer *lb,
double b[],
double se[],
const Integer *maxit,
const double *tol,
Integer *warn,
Integer *ifail) |
|
3
Description
g02jaf fits a model of the form:
where
- is a vector of observations on the dependent variable,
- is a known by design matrix for the fixed independent variables,
- is a vector of length of unknown fixed effects,
- is a known by design matrix for the random independent variables,
- is a vector of length of unknown random effects,
and
- is a vector of length of unknown random errors.
Both
and
are assumed to have a Gaussian distribution with expectation zero and
where
,
is the
identity matrix and
is a diagonal matrix. It is assumed that the random variables,
, can be subdivided into
groups with each group being identically distributed with expectations zero and variance
. The diagonal elements of matrix
therefore take one of the values
, depending on which group the associated random variable belongs to.
The model therefore contains three sets of unknowns, the fixed effects,
, the random effects
and a vector of
variance components,
, where
. Rather than working directly with
,
g02jaf uses an iterative process to estimate
. Due to the iterative nature of the estimation a set of initial values,
, for
is required.
g02jaf allows these initial values either to be supplied by you or calculated from the data using the minimum variance quadratic unbiased estimators (MIVQUE0) suggested by
Rao (1972).
g02jaf fits the model using a quasi-Newton algorithm to maximize the restricted log-likelihood function:
where
Once the final estimates for
have been obtained, the value of
is given by:
Case weights, , can be incorporated into the model by replacing and with and respectively, for a diagonal weight matrix .
The log-likelihood,
, is calculated using the sweep algorithm detailed in
Wolfinger et al. (1994).
4
References
Goodnight J H (1979) A tutorial on the SWEEP operator The American Statistician 33(3) 149–158
Harville D A (1977) Maximum likelihood approaches to variance component estimation and to related problems JASA 72 320–340
Rao C R (1972) Estimation of variance and covariance components in a linear model J. Am. Stat. Assoc. 67 112–115
Stroup W W (1989) Predictable functions and prediction space in the mixed model procedure Applications of Mixed Models in Agriculture and Related Disciplines Southern Cooperative Series Bulletin No. 343 39–48
Wolfinger R, Tobias R and Sall J (1994) Computing Gaussian likelihoods and their derivatives for general linear mixed models SIAM Sci. Statist. Comput. 15 1294–1310
5
Arguments
- 1: – IntegerInput
-
On entry: , the number of observations.
Constraint:
.
- 2: – IntegerInput
-
On entry: the number of columns in the data matrix,
dat.
Constraint:
.
- 3: – IntegerInput
-
On entry: the first dimension of the array
dat as declared in the (sub)program from which
g02jaf is called.
Constraint:
.
- 4: – Real (Kind=nag_wp) arrayInput
-
On entry: array containing all of the data. For the
th observation:
- holds the dependent variable, ;
- if , holds the case weights;
- if , holds the subject variable.
The remaining columns hold the values of the independent variables.
Constraints:
- if , ;
- if , .
- 5: – Integer arrayInput
-
On entry:
contains the number of levels associated with the
th variable of the data matrix
dat. If this variable is continuous or binary (i.e., only takes the values zero or one) then
should be
; if the variable is discrete then
is the number of levels associated with it and
is assumed to take the values
to
, for
.
Constraint:
, for .
- 6: – IntegerInput
-
On entry: the column of
dat holding the dependent,
, variable.
Constraint:
.
- 7: – IntegerInput
-
On entry: the column of
dat holding the case weights.
If , no weights are used.
Constraint:
.
- 8: – IntegerInput
-
On entry: the number of independent variables in the model which are to be treated as being fixed.
Constraint:
.
- 9: – Integer arrayInput
-
On entry: the columns of the data matrix
dat holding the fixed independent variables with
holding the column number corresponding to the
th fixed variable.
Constraint:
, for .
- 10: – IntegerInput
-
On entry: flag indicating whether a fixed intercept is included ().
Constraint:
or .
- 11: – IntegerInput
-
On entry: the number of independent variables in the model which are to be treated as being random.
- 12: – Integer arrayInput
-
On entry: the columns of the data matrix holding the random independent variables with holding the column number corresponding to the th random variable.
Constraint:
, for .
- 13: – IntegerInput
-
On entry: if
and
,
nvpr is the number of variance components being
, (
), else
.
If , is not referenced.
Constraint:
if , .
- 14: – Integer arrayInput
-
On entry: holds a flag indicating the variance of the th random variable. The variance of the th random variable is , where if and and otherwise. Random variables with the same value of are assumed to be taken from the same distribution.
Constraint:
, for .
- 15: – IntegerInput
-
On entry: flag indicating whether a random intercept is included (
).
If
,
rint is not referenced.
Constraint:
or .
- 16: – IntegerInput
-
On entry: the column of
dat holding the subject variable.
If , no subject variable is used.
Specifying a subject variable is equivalent to specifying the interaction between that variable and all of the random-effects. Letting the notation denote the interaction between variables and , fitting a model with , random-effects and subject variable is equivalent to fitting a model with random-effects and no subject variable. If the model is equivalent to fitting and no subject variable.
Constraint:
.
- 17: – Real (Kind=nag_wp) arrayInput/Output
-
On entry: holds the initial values of the variance components,
, with
the initial value for
, for
. If
and
,
, else
.
If
, the remaining elements of
gamma are ignored and the initial values for the variance components are estimated from the data using MIVQUE0.
On exit: , for , holds the final estimate of and holds the final estimate for .
Constraint:
, for .
- 18: – IntegerOutput
-
On exit: the number of fixed effects estimated (i.e., the number of columns, , in the design matrix ).
- 19: – IntegerOutput
-
On exit: the number of random effects estimated (i.e., the number of columns, , in the design matrix ).
- 20: – IntegerOutput
-
On exit: the degrees of freedom.
- 21: – Real (Kind=nag_wp)Output
-
On exit:
where
is the log of the restricted maximum likelihood calculated at
, the estimated variance components returned in
gamma.
- 22: – IntegerInput
-
On entry: the size of the array
b.
Constraint:
where if and otherwise.
- 23: – Real (Kind=nag_wp) arrayOutput
-
On exit: the parameter estimates,
, with the first
nff elements of
b containing the fixed effect parameter estimates,
and the next
nrf elements of
b containing the random effect parameter estimates,
.
Fixed effects
If
,
contains the estimate of the fixed intercept. Let
denote the number of levels associated with the
th fixed variable, that is
. Define
- if , else if , ;
- , .
Then for
:
- if ,
contains the parameter estimate for the th level of the th fixed variable, for ;
- if , contains the parameter estimate for the th fixed variable.
Random effects
Redefining
to denote the number of levels associated with the
th random variable, that is
. Define
- if , else if , ;
, .
Then for
:
- if ,
- if ,
contains the parameter estimate for the th level of the th random variable, for ;
- if , contains the parameter estimate for the th random variable;
- if ,
- let denote the number of levels associated with the subject variable, that is ;
- if ,
contains the parameter estimate for the interaction between the th level of the subject variable and the th level of the th random variable, for and ;
- if ,
contains the parameter estimate for the interaction between the th level of the subject variable and the th random variable, for ;
- if , contains the estimate of the random intercept.
- 24: – Real (Kind=nag_wp) arrayOutput
-
On exit: the standard errors of the parameter estimates given in
b.
- 25: – IntegerInput
-
On entry: the maximum number of iterations.
If , the default value of is used.
If
, the parameter estimates
and corresponding standard errors are calculated based on the value of
supplied in
gamma.
- 26: – Real (Kind=nag_wp)Input
-
On entry: the tolerance used to assess convergence.
If , the default value of is used, where is the machine precision.
- 27: – IntegerOutput
-
On exit: is set to
if a variance component was estimated to be a negative value during the fitting process. Otherwise
warn is set to
.
If , the negative estimate is set to zero and the estimation process allowed to continue.
- 28: – IntegerInput/Output
-
On entry:
ifail must be set to
,
. If you are unfamiliar with this argument you should refer to
Section 3.4 in How to Use the NAG Library and its Documentation for details.
For environments where it might be inappropriate to halt program execution when an error is detected, the value
is recommended. If the output of error messages is undesirable, then the value
is recommended. Otherwise, if you are not familiar with this argument, the recommended value is
.
When the value is used it is essential to test the value of ifail on exit.
On exit:
unless the routine detects an error or a warning has been flagged (see
Section 6).
6
Error Indicators and Warnings
If on entry
or
, explanatory error messages are output on the current error message unit (as defined by
x04aaf).
Errors or warnings detected by the routine:
-
On entry, and .
Constraint: and any supplied weights must be .
On entry, .
Constraint: or .
On entry,
lb too small:
.
On entry, and .
Constraint: .
On entry, .
Constraint: number of observations with nonzero weights must be greater than one.
On entry, .
Constraint: .
On entry, .
Constraint: .
On entry, and .
Constraint: .
On entry, and .
Constraint: and .
On entry, and .
Constraint: and ( or ).
On entry, .
Constraint: or .
On entry, and .
Constraint: .
On entry, and .
Constraint: .
-
On entry, , for at least one .
On entry, invalid data: categorical variable with value greater than that specified in
levels.
On entry, , for at least one .
On entry, .
Constraint: , for all .
On entry, .
Constraint: , for all .
On entry, .
Constraint: , for all .
-
Degrees of freedom : .
This is due to the number of parameters exceeding the effective number of observations.
-
Routine failed to converge in
maxit iterations:
.
See
Section 10 for advice.
Routine failed to converge to specified tolerance:
.
See
Section 10 for advice.
An unexpected error has been triggered by this routine. Please
contact
NAG.
See
Section 3.9 in How to Use the NAG Library and its Documentation for further information.
Your licence key may have expired or may not have been installed correctly.
See
Section 3.8 in How to Use the NAG Library and its Documentation for further information.
Dynamic memory allocation failed.
See
Section 3.7 in How to Use the NAG Library and its Documentation for further information.
7
Accuracy
The accuracy of the results can be adjusted through the use of the
tol argument.
8
Parallelism and Performance
g02jaf is threaded by NAG for parallel execution in multithreaded implementations of the NAG Library.
g02jaf makes calls to BLAS and/or LAPACK routines, which may be threaded within the vendor library used by this implementation. Consult the documentation for the vendor library for further information.
Please consult the
X06 Chapter Introduction for information on how to control and interrogate the OpenMP environment used within this routine. Please also consult the
Users' Note for your implementation for any additional implementation-specific information.
Wherever possible any block structure present in the design matrix
should be modelled through a subject variable, specified via
svid, rather than being explicitly entered into
dat.
g02jaf uses an iterative process to fit the specified model and for some problems this process may fail to converge (see
). If the routine fails to converge then the maximum number of iterations (see
maxit) or tolerance (see
tol) may require increasing; try a different starting estimate in
gamma. Alternatively, the model can be fit using maximum likelihood (see
g02jbf) or using the noniterative MIVQUE0.
To fit the model just using MIVQUE0, the first element of
gamma should be set to
and
maxit should be set to zero.
Although the quasi-Newton algorithm used in
g02jaf tends to require more iterations before converging compared to the Newton–Raphson algorithm recommended by
Wolfinger et al. (1994), it does not require the second derivatives of the likelihood function to be calculated and consequentially takes significantly less time per iteration.
10
Example
The following dataset is taken from
Stroup (1989) and arises from a balanced split-plot design with the whole plots arranged in a randomized complete block-design.
In this example the full design matrix for the random independent variable,
, is given by:
where
The block structure evident in
(1) is modelled by specifying a four-level subject variable, taking the values
. The first column of
is added to
by setting
. The remaining columns of
are specified by a three level factor, taking the values,
.
10.1
Program Text
Program Text (g02jafe.f90)
10.2
Program Data
Program Data (g02jafe.d)
10.3
Program Results
Program Results (g02jafe.r)