How to visually assess the convergence of a mixed-effects model by plotting various optimizers
Want to share your content on R-bloggers? click here if you have a blog, or here if you don't.
To assess whether convergence warnings render the results invalid, or on the contrary, the results can be deemed valid in spite of the warnings, Bates et al. (2023) suggest refitting models affected by convergence warnings with a variety of optimizers. The authors argue that, if the different optimizers produce practically-equivalent results, the results are valid. The allFit function from the ‘lme4’ package allows the refitting of models using a number of optimizers. To use the seven optimizers listed above, two extra packages must be installed: ‘dfoptim’ and ‘optimx’ (see lme4 manual). The output from allFit() contains several statistics on the fixed and the random effects fitted by each optimizer.
library(lme4) #> Loading required package: Matrix library(dfoptim) library(optimx) # Create data using code by Ben Bolker from # https://stackoverflow.com/a/38296264/7050882 set.seed(101) spin = runif(600, 1, 24) reg = runif(600, 1, 15) ID = rep(c("1","2","3","4","5", "6", "7", "8", "9", "10")) day = rep(1:30, each = 10) testdata <- data.frame(spin, reg, ID, day) testdata$fatigue <- testdata$spin * testdata$reg/10 * rnorm(30, mean=3, sd=2) # Model fit = lmer(fatigue ~ spin * reg + (1|ID), data = testdata, REML = TRUE) # Refit model using all available algorithms multi_fit = allFit(fit) #> bobyqa : [OK] #> Nelder_Mead : [OK] #> nlminbwrap : [OK] #> nmkbw : [OK] #> optimx.L-BFGS-B : [OK] #> nloptwrap.NLOPT_LN_NELDERMEAD : [OK] #> nloptwrap.NLOPT_LN_BOBYQA : [OK] # Show results summary(multi_fit)$fixef #> (Intercept) spin reg spin:reg #> bobyqa -2.975678 0.5926561 0.1437204 0.1834016 #> Nelder_Mead -2.975675 0.5926559 0.1437202 0.1834016 #> nlminbwrap -2.975677 0.5926560 0.1437203 0.1834016 #> nmkbw -2.975678 0.5926561 0.1437204 0.1834016 #> optimx.L-BFGS-B -2.975680 0.5926562 0.1437205 0.1834016 #> nloptwrap.NLOPT_LN_NELDERMEAD -2.975666 0.5926552 0.1437196 0.1834017 #> nloptwrap.NLOPT_LN_BOBYQA -2.975678 0.5926561 0.1437204 0.1834016
Created on 2023-06-24 with reprex v2.0.2
Numbers are hard for the common people, however. We are far better at vision. So, could there be a way to visualise the output from allFit(), and look at the parameters for a set of predictors across various optimizers (e.g., bobyqa, Nelder-Mead, etc.)?
Yes, there is a way, using a custom function called plot.fixef.allFit
, described in this post.
library(lme4) #> Loading required package: Matrix library(dfoptim) library(optimx) # Create data using code by Ben Bolker from # https://stackoverflow.com/a/38296264/7050882 set.seed(101) spin = runif(600, 1, 24) reg = runif(600, 1, 15) ID = rep(c("1","2","3","4","5", "6", "7", "8", "9", "10")) day = rep(1:30, each = 10) testdata <- data.frame(spin, reg, ID, day) testdata$fatigue <- testdata$spin * testdata$reg/10 * rnorm(30, mean=3, sd=2) # Model fit = lmer(fatigue ~ spin * reg + (1|ID), data = testdata, REML = TRUE) # Refit model using all available algorithms multi_fit = allFit(fit) #> bobyqa : [OK] #> Nelder_Mead : [OK] #> nlminbwrap : [OK] #> nmkbw : [OK] #> optimx.L-BFGS-B : [OK] #> nloptwrap.NLOPT_LN_NELDERMEAD : [OK] #> nloptwrap.NLOPT_LN_BOBYQA : [OK] # Show results summary(multi_fit)$fixef #> (Intercept) spin reg spin:reg #> bobyqa -2.975678 0.5926561 0.1437204 0.1834016 #> Nelder_Mead -2.975675 0.5926559 0.1437202 0.1834016 #> nlminbwrap -2.975677 0.5926560 0.1437203 0.1834016 #> nmkbw -2.975678 0.5926561 0.1437204 0.1834016 #> optimx.L-BFGS-B -2.975680 0.5926562 0.1437205 0.1834016 #> nloptwrap.NLOPT_LN_NELDERMEAD -2.975666 0.5926552 0.1437196 0.1834017 #> nloptwrap.NLOPT_LN_BOBYQA -2.975678 0.5926561 0.1437204 0.1834016 # Read in function from GitHub source('https://raw.githubusercontent.com/pablobernabeu/plot.fixef.allFit/main/plot.fixef.allFit.R') plot.fixef.allFit(multi_fit, select_predictors = c('spin', 'reg', 'spin:reg'), # Increase padding at top and bottom of Y axis multiply_y_axis_limits = 1.3, y_title = 'Fixed effect (*b*)') #> Loading required package: dplyr #> #> Attaching package: 'dplyr' #> The following objects are masked from 'package:stats': #> #> filter, lag #> The following objects are masked from 'package:base': #> #> intersect, setdiff, setequal, union #> Loading required package: reshape2 #> Loading required package: stringr #> Loading required package: scales #> Loading required package: ggplot2 #> Loading required package: ggtext #> Loading required package: patchwork
Created on 2023-06-24 with reprex v2.0.2
R-bloggers.com offers daily e-mail updates about R news and tutorials about learning R and many other topics. Click here if you're looking to post or find an R/data-science job.
Want to share your content on R-bloggers? click here if you have a blog, or here if you don't.