Un cours en R et Stan avec brms
Cours n°01 : Introduction à l’inférence bayésienne
Cours n°02 : Modèle Beta-Binomial
Cours n°03 : Introduction à brms, modèle de régression linéaire
Cours n°04 : Modèle de régression linéaire (suite)
Cours n°05 : Markov Chain Monte Carlo
Cours n°06 : Modèle linéaire généralisé
Cours n°07 : Comparaison de modèles
Cours n°08 : Modèles multi-niveaux (généralisés)
Cours n°09 : Examen final
\[\newcommand\given[1][]{\:#1\vert\:}\]
Le but est de construire un modèle qui puisse apprendre à plusieurs niveaux, qui puisse produire des estimations qui seront informées par les différents groupes présents dans les données. Nous allons suivre l’exemple suivant tout au long de ce cours.
Imaginons que nous ayons construit un robot visiteur de cafés, et que celui-ci s’amuse à mesurer le temps d’attente après avoir commandé. Ce robot visite 20 cafés différents, 5 fois le matin et 5 fois l’après-midi, et mesure le temps de service après avoir commandé un cappuccino.
cafe afternoon wait
1 2 0 2.521509
2 2 0 2.387150
3 2 0 1.954398
4 3 1 3.044962
5 7 0 4.132367
6 8 0 5.246669
7 9 1 2.236393
8 11 1 2.103433
9 12 0 3.900817
10 13 0 4.044824
11 15 0 4.752380
12 16 1 2.533639
13 18 0 5.841681
14 18 0 6.156569
15 19 0 3.582722
df %>%
ggplot(aes(x = factor(cafe), y = wait, fill = factor(afternoon) ) ) +
geom_dotplot(
stackdir = "center", binaxis = "y",
dotsize = 1, show.legend = FALSE
) +
geom_hline(yintercept = mean(df$wait), linetype = 2) +
facet_wrap(~afternoon, ncol = 2) +
labs(x = "Café", y = "Temps d'attente (en minutes)")On peut construire un premier modèle, qui estime le temps moyen (sur tous les cafés confondus) pour être servi.
\[ \begin{align} \color{orangered}{w_{i}} \ &\color{orangered}{\sim \mathrm{Normal}(\mu_{i}, \sigma)} \\ \color{black}{\mu_{i}} \ &\color{black}{= \alpha} \\ \color{steelblue}{\alpha} \ &\color{steelblue}{\sim \mathrm{Normal}(5, 10)} \\ \color{steelblue}{\sigma} \ &\color{steelblue}{\sim \mathrm{HalfCauchy}(0, 2)} \\ \end{align} \]
\[ p(x \given x_{0}, \gamma) = \left(\pi \gamma \left[1 + \left(\frac{x-x_{0}}{\gamma}\right)^{2}\right] \right)^{-1} \]
Deuxième modèle qui estime un intercept par café. Équivalent à construire 20 dummy variables.
\[ \begin{align} \color{orangered}{w_{i}} \ &\color{orangered}{\sim \mathrm{Normal}(\mu_{i}, \sigma)} \\ \color{black}{\mu_{i}} \ &\color{black}{= \alpha_{\text{café}[i]}} \\ \color{steelblue}{\alpha_{\text{café}[i]}} \ &\color{steelblue}{\sim \mathrm{Normal}(5, 10)} \\ \color{steelblue}{\sigma} \ &\color{steelblue}{\sim \mathrm{HalfCauchy}(0, 2)} \\ \end{align} \]
Estimate Est.Error Q2.5 Q97.5
b_factorcafe1 3.446720 0.2581097 2.9570223 3.955792
b_factorcafe2 1.735698 0.2547330 1.2214308 2.255546
b_factorcafe3 3.321396 0.2538071 2.8150220 3.827093
b_factorcafe4 2.792376 0.2622491 2.2902496 3.304397
b_factorcafe5 1.463107 0.2495957 0.9666945 1.962709
b_factorcafe6 3.645492 0.2606125 3.1280079 4.164413
b_factorcafe7 2.944369 0.2537705 2.4381786 3.436924
b_factorcafe8 3.171952 0.2711110 2.6295728 3.690205
b_factorcafe9 3.333783 0.2657974 2.8171111 3.862563
b_factorcafe10 3.098585 0.2603172 2.5906690 3.619480
b_factorcafe11 1.920268 0.2672054 1.4035323 2.435402
b_factorcafe12 3.492849 0.2569514 2.9907473 3.989367
b_factorcafe13 3.220294 0.2559399 2.7253452 3.721055
b_factorcafe14 2.632583 0.2506545 2.1207012 3.127929
b_factorcafe15 3.479847 0.2600625 2.9548957 3.995019
b_factorcafe16 3.002896 0.2608633 2.5028342 3.505349
b_factorcafe17 3.878754 0.2685568 3.3476324 4.409492
b_factorcafe18 5.533190 0.2599808 5.0222679 6.055488
b_factorcafe19 2.975092 0.2587433 2.4756302 3.488851
b_factorcafe20 3.361872 0.2606892 2.8497260 3.878911
Est-ce qu’on ne pourrait pas faire en sorte que le temps mesuré au café 1 informe la mesure réalisée au café 2, et au café 3 ? Ainsi que le temps moyen pour être servi ? Nous allons apprendre le prior à partir des données…
\[ \begin{align} \text{Niveau 1}: \color{orangered}{w_{i}} \ &\color{orangered}{\sim \mathrm{Normal}(\mu_{i}, \sigma)} \\ \color{black}{\mu_{i}} \ &\color{black}{= \alpha_{\text{café}[i]}} \\ \text{Niveau 2}: \color{steelblue}{\alpha_{\text{café}}} \ &\color{steelblue}{\sim \mathrm{Normal}(\alpha,\sigma_{\text{café}})} \\ \color{steelblue}{\alpha} \ &\color{steelblue}{\sim \mathrm{Normal}(5, 10)} \\ \color{steelblue}{\sigma_{\text{café}}} \ &\color{steelblue}{\sim \mathrm{HalfCauchy}(0, 2)} \\ \color{steelblue}{\sigma} \ &\color{steelblue}{\sim \mathrm{HalfCauchy}(0, 2)} \\ \end{align} \]
Le prior de l’intercept pour chaque café (\(\alpha_{\text{café}}\)) est maintenant fonction de deux paramètres (\(\alpha\) et \(\sigma_{\text{café}}\)). \(\alpha\) et \(\sigma_{\text{café}}\) sont appelés des hyper-paramètres, ce sont des paramètres pour des paramètres, et leurs priors sont appelés des hyperpriors. Il y a deux niveaux dans le modèle…
\[ \begin{align} \color{orangered}{w_{i}} \ &\color{orangered}{\sim \mathrm{Normal}(\mu_{i}, \sigma)} \\ \color{black}{\mu_{i}} \ &\color{black}{= \alpha_{\text{café}[i]}} \\ \color{steelblue}{\alpha_{\text{café}}} \ &\color{steelblue}{\sim \mathrm{Normal}(\alpha,\sigma_{\text{café}})} \\ \end{align} \]
NB : \(\alpha\) est ici défini dans le prior de \(\alpha_{\text{café}}\) mais on pourrait, de la même manière, le définir dans le modèle linéaire :
\[ \begin{align} \color{orangered}{w_{i}} \ &\color{orangered}{\sim \mathrm{Normal}(\mu_{i}, \sigma)} \\ \color{black}{\mu_{i}} \ &\color{black}{= \alpha + \alpha_{\text{café}[i]}} \\ \color{steelblue}{\alpha_{\text{café}}} \ &\color{steelblue}{\sim \mathrm{Normal}(0,\sigma_{\text{café}})} \\ \end{align} \]
On peut toujours “enlever” la moyenne d’une distribution gaussienne et la considérer comme une constante plus une gaussienne centrée sur zéro.
NB : quand \(\alpha\) est défini dans le modèle linéaire, les \(\alpha_{\text{café}}\) représentent des déviations de l’intercept moyen. Il faut donc ajouter \(\alpha\) et \(\alpha_{\text{café}}\) pour obtenir le temps d’attente moyen par café…
Ce modèle a 23 paramètres, l’intercept général \(\alpha\), la variabilité résiduelle \(\sigma\), la variabilité entre les cafés \(\sigma_{\text{café}}\), et un intercept par café.
L’estimateur James-Stein est défini comme \(z = \bar{y} + c(y - \bar{y})\), où \(\bar{y}\) désigne la moyenne de l’échantillon, \(y\) une observation individuelle, et \(c\) une constante, le shrinking factor (Efron & Morris, 1977).
Le shrinking factor est déterminé à la fois par la variabilité (imprécision) de la mesure (e.g., son écart-type) et par la distance à l’estimation moyenne (i.e., \(y - \bar{y}\)). En d’autres termes, cet estimateur fait moins “confiance” (i.e., accorde moins de poids) aux observations imprécises et/ou extrêmes. En pratique, le shrinkage agit comme une protection contre le sur-apprentissage (overfitting).
Le shrinkage observé slide précédente est dû à des phénomènes de partage (pooling) de l’information entre les cafés. L’estimation de l’intercept pour chaque café informe l’estimation de l’intercept des autres cafés, ainsi que l’estimation de l’intercept général (i.e., la moyenne générale).
On distingue en général trois perspectives (ou stratégies) :
Complete pooling : on suppose que le temps d’attente est invariant, on estime un intercept commun (mod1).
No pooling : on suppose que les temps d’attente de chaque café sont uniques et indépendants : on estime un intercept par café, mais sans informer le niveau supérieur (mod2).
Partial pooling : on utilise un prior adaptatif, comme dans l’exemple précédent (mod3).
La stratégie complete pooling en général underfitte les données (faibles capacités de prédiction) tandis que le stratégie no pooling revient à overfitter les données (faibles capacités de prédiction ici aussi). La stratégie partial pooling (i.e., celle des modèles multi-niveaux) permet d’’équilibrer underfitting et overfitting.
On peut comparer ces trois modèles en utilisant le WAIC (discuté au Cours n°07).
elpd_diff se_diff elpd_waic se_elpd_waic p_waic se_p_waic waic se_waic
mod3 0.0 0.0 -253.9 8.3 18.2 1.5 507.7 16.6
mod2 -0.6 1.3 -254.5 8.4 19.5 1.6 509.0 16.8
mod1 -57.4 10.6 -311.2 10.5 2.1 0.4 622.4 21.1
On remarque que le modèle 3 a seulement 18 “effective parameters” (pWAIC) et moins de paramètres que le modèle 2, alors qu’il en a en réalité 2 de plus… posterior_summary(mod3)[3, 1] nous donne le sigma du prior adaptatif des \(\alpha_{\text{café}}\) (\(\sigma_{\text{café}} = 0.82\)). On remarque que ce sigma est très faible et correspond à assigner un prior très contraignant, ou régularisateur.
On compare les estimations du premier modèle (complete pooling model) et du troisième modèle (partial pooling model).
Estimate Est.Error Q2.5 Q97.5
b_Intercept 3.12029 0.08305618 2.961966 3.278956
sigma 1.14329 0.05821889 1.035996 1.263067
Estimate Est.Error Q2.5 Q97.5
b_Intercept 3.1192094 0.20762244 2.7129150 3.5327098
sigma 0.8220335 0.04318892 0.7431579 0.9115307
Les deux modèles font la même prédiction (en moyenne) pour \(\alpha\), mais le modèle 3 est plus incertain de sa prédiction que le modèle 1 (voir l’erreur standard pour \(\alpha\))…
L’estimation de \(\sigma\) du modèle 3 est plus petite que celle du modèle 1 car le modèle 3 décompose la variabilité non expliquée en deux sources : la variabilité du temps d’attente entre les cafés \(\sigma_{\text{café}}\) et la variabilité résiduelle \(\sigma\).
Imaginons que notre robot ne visite pas tous les cafés le même nombre de fois (comme dans le cas précédent) mais qu’il visite plus souvent les cafés proches de chez lui…
On observe que les cafés qui sont souvent visités (à droite) subissent moins l’effet du shrinkage. Leur estimation est moins “tirée” vers la moyenne générale que les estimations des cafés les moins souvent visités (à gauche).
Cinq définitions (contradictoires) relevées par Gelman (2005).
Gelman & Hill (2006) suggèrent plutôt l’utilisation des termes de constant effects et varying effects, et de toujours utiliser la modélisation multi-niveaux, en considérant que ce qu’on appelle effet fixe peut simplement être considéré comme un effet aléatoire dont la variance serait égale à \(0\).
Le fait de faire varier les intercepts par café est simplement une autre manière de régulariser (de manière adaptative), c’est à dire de diminuer le poids accordé aux données dans l’estimation. Le modèle devient à même d’estimer à quel point les groupes (ici, les cafés) sont différents, tout en estimant les caractéristiques de chaque café…
Différence entre les cross-classified (ou “crossed”) multilevel models et nested or hierarchical multilevel models. Le premier type de modèle concerne des données structurées selon deux (ou plus) facteurs aléatoires non “nichés”. Le deuxième type de modèles concerne des données structurées de manière hiérarchique (e.g., un élève dans une classe dans une école dans une ville…). Voir cette discussion pour plus de détails.
Les deux types de modèles s’écrivent cependant de manière similaire, sur plusieurs “niveaux”. Le terme “multi-niveaux” (dans notre terminologie) fait donc référence à la structure du modèle, à sa spécification. À distinguer de la structure des données.
On pourrait se poser la question de savoir si la récence des cafés (leur âge) ne serait pas une source de variabilité non contrôlée ? Il suffit d’ajouter un intercept qui varie par âge, et de lui attribuer un prior adaptatif.
\[ \begin{align} \color{orangered}{w_{i}} \ &\color{orangered}{\sim \mathrm{Normal}(\mu_{i}, \sigma)} \\ \color{black}{\mu_{i}} \ &\color{black}{= \alpha + \alpha_{\text{café}[i]} + \alpha_{\text{âge}[i]}} \\ \color{steelblue}{\alpha_{\text{café}}} \ &\color{steelblue}{\sim \mathrm{Normal}(5, \sigma_{\text{café}})} \\ \color{steelblue}{\alpha_{\text{âge}}} \ &\color{steelblue}{\sim \mathrm{Normal}(5, \sigma_{\text{âge}})} \\ \color{steelblue}{\alpha} \ &\color{steelblue}{\sim \mathrm{Normal}(0, 10)} \\ \color{steelblue}{\sigma_{\text{café}}} \ &\color{steelblue}{\sim \mathrm{HalfCauchy}(0, 2)} \\ \color{steelblue}{\sigma_{\text{âge}}} \ &\color{steelblue}{\sim \mathrm{HalfCauchy}(0, 2)} \\ \color{steelblue}{\sigma} \ &\color{steelblue}{\sim \mathrm{HalfCauchy}(0, 2)} \\ \end{align} \]
On s’intéresse maintenant à l’effet du moment de la journée sur le temps d’attente. Attend-on plus le matin, ou l’après-midi ?
\[ \begin{align} \color{orangered}{w_{i}} \ &\color{orangered}{\sim \mathrm{Normal}(\mu_{i}, \sigma)} \\ \color{black}{\mu_{i}} \ &\color{black}{= \alpha_{\text{café}[i]} + \beta_{\text{café}[i]} \times A_{i}} \\ \end{align} \]
Où \(A_{i}\) est une dummy variable codée 0/1 pour le matin et l’après-midi et où \(\beta_{\text{café}}\) est donc un paramètre de différence (i.e., une pente) entre le matin et l’après-midi.
Remarque : on sait que les cafés ont des intercepts et des pentes qui co-varient… Les cafés populaires seront surchargés le matin et beaucoup moins l’après-midi, résultant en une pente importante. Ces cafés auront aussi un temps d’attente moyen plus long (i.e., un intercept plus grand). Dans ces cafés, \(\alpha\) est grand et \(\beta\) est loin de zéro. À l’inverse, dans un café peu populaire, le temps d’attente sera faible, ainsi que la différence entre matin et après-midi.
On pourrait donc utiliser la co-variation entre intercept et pente pour faire de meilleures inférences. Autrement dit, faire en sorte que l’estimation de l’intercept informe celle de la pente, et réciproquement.
On s’intéresse maintenant à l’effet du moment de la journée sur le temps d’attente. Attend-on plus le matin, ou l’après-midi ?
\[ \begin{align} \color{orangered}{w_{i}} \ &\color{orangered}{\sim \mathrm{Normal}(\mu_{i}, \sigma)} \\ \color{black}{\mu_{i}} \ &\color{black}{= \alpha_{\text{café}[i]} + \beta_{\text{café}[i]} \times A_{i}} \\ \color{steelblue}{\begin{bmatrix} \alpha_{\text{café}} \\ \beta_{\text{café}} \\ \end{bmatrix}} \ &\color{steelblue}{\sim \mathrm{MVNormal}\bigg(\begin{bmatrix} \alpha \\ \beta \end{bmatrix}, \textbf{S}\bigg)} \\ \end{align} \]
La troisième ligne postule que chaque café a un intercept \(\alpha_{\text{café}}\) et une pente \(\beta_{\text{café}}\), définis par un prior Gaussien bivarié (i.e., à deux dimensions) ayant comme moyennes \(\alpha\) et \(\beta\) et comme matrice de covariance \(\textbf{S}\).
\[\mathbf{x} \sim \mathcal{N}(\boldsymbol{\mu}, \boldsymbol{\Sigma})\]
Où \(\boldsymbol{\mu}\) est un vecteur (à \(k\) dimensions) de moyennes, par exemple: mu <- c(a, b).
\(\boldsymbol{\Sigma}\) est une matrice de covariance de \(k \times k\) dimensions, et qui correspond à la matrice donnée par la fonction vcov().
\[ \begin{align} \boldsymbol{\Sigma} &= \begin{pmatrix} \sigma_{\alpha}^{2} & \sigma_{\alpha} \sigma_{\beta} \rho \\ \sigma_{\alpha} \sigma_{\beta} \rho & \sigma_{\beta}^{2} \\ \end{pmatrix} \\ \end{align} \]
\[ \begin{align} \boldsymbol{\Sigma} &= \begin{pmatrix} \sigma_{\alpha}^{2} & \sigma_{\alpha} \sigma_{\beta} \rho \\ \sigma_{\alpha} \sigma_{\beta} \rho & \sigma_{\beta}^{2} \\ \end{pmatrix} \\ \end{align} \]
Cette matrice peut se construire de deux manières différentes, strictement équivalentes.
\[ \begin{align} \boldsymbol{\Sigma} &= \begin{pmatrix} \sigma_{\alpha}^{2} & \sigma_{\alpha} \sigma_{\beta} \rho \\ \sigma_{\alpha} \sigma_{\beta} \rho & \sigma_{\beta}^{2} \\ \end{pmatrix} \\ \end{align} \]
La deuxième méthode est pratique car elle considère séparément les écart-types et les corrélations.
\[ \begin{align} \color{orangered}{w_{i}} \ &\color{orangered}{\sim \mathrm{Normal}(\mu_{i}, \sigma)} \\ \color{black}{\mu_{i}} \ &\color{black}{= \alpha_{\text{café}[i]} + \beta_{\text{café}[i]} \times A_{i}} \\ \color{steelblue}{\begin{bmatrix} \alpha_{\text{café}} \\ \beta_{\text{café}} \\ \end{bmatrix}} \ &\color{steelblue}{\sim \mathrm{MVNormal}\bigg(\begin{bmatrix} \alpha \\ \beta \end{bmatrix}, \textbf{S}\bigg)} \\ \color{black}{\textbf{S}} \ &\color{black}{= \begin{pmatrix} \sigma_{\alpha} & 0 \\ 0 & \sigma_{\beta} \\ \end{pmatrix} \ \textbf{R} \begin{pmatrix} \sigma_{\alpha} & 0 \\ 0 & \sigma_{\beta} \\ \end{pmatrix}} \\ \color{steelblue}{\alpha} \ &\color{steelblue}{\sim \mathrm{Normal} (0, 10)} \\ \color{steelblue}{\beta} \ &\color{steelblue}{\sim \mathrm{Normal} (0, 10)} \\ \color{steelblue}{\sigma_{\alpha}} \ &\color{steelblue}{\sim \mathrm{HalfCauchy} (0, 2)} \\ \color{steelblue}{\sigma_{\beta}} \ &\color{steelblue}{\sim \mathrm{HalfCauchy} (0, 2)} \\ \color{steelblue}{\sigma} \ &\color{steelblue}{\sim \mathrm{HalfCauchy} (0, 2)} \\ \color{steelblue}{\textbf{R}} \ &\color{steelblue}{\sim \mathrm{LKJ}(2)} \\ \end{align} \]
\(\textbf{S}\) est définie en factorisant \(\sigma_{\alpha}\), \(\sigma_{\beta}\), et la matrice de corrélation \(\textbf{R}\). La suite du modèle définit simplement les priors pour les effets constants. La dernière ligne spécifie le prior pour \(\textbf{R}\).
Prior proposé par Lewandowski et al. (2009). Un seul paramètre \(\zeta\) (zeta) spécifie la concentration de la distribution du coefficient de corrélation. Le prior \(\mathrm{LKJ}(2)\) définit un prior peu informatif pour \(\rho\) (rho) qui est sceptique des corrélations extrêmes (i.e., des valeurs proches de \(-1\) ou \(1\)).
Le paquet brms utilise la même syntaxe que les fonctions de base R (comme lm) ou que le paquet lme4.
La partie gauche représente notre variable dépendante (ou “outcome”, i.e., ce qu’on essaye de prédire).
La première partie de la partie droite de la formule représente les effets constants (effets fixes), tandis que la seconde partie (entre parenthèses) représente les effets “variants” ou “variables” (effets aléatoires).
Le premier modèle ci-dessus contient seulement un intercept variable, qui varie par Subject. Le deuxième modèle contient également un intercept variable, mais aussi une pente variable pour l’effet de Days.
Lorsqu’on inclut plusieurs effets variants (e.g., un intercept et une pente variables), brms postule qu’on souhaite aussi estimer la corrélation entre ces deux effets. Dans le cas contraire, on peut supprimer cette corrélation (i.e., la fixer à 0) en utilisant ||.
On spécifie un intercept et une pente (pour l’effet d’afternoon) qui varient par cafe.
post <- as_draws_df(x = mod5) # extracts posterior samples
R <- rethinking::rlkjcorr(n = 16000, K = 2, eta = 2) # samples from prior
data.frame(prior = R[, 1, 2], posterior = post$cor_cafe__Intercept__afternoon) %>%
gather(type, value, prior:posterior) %>%
ggplot(aes(x = value, color = type, fill = type) ) +
geom_histogram(position = "identity", alpha = 0.2) +
labs(x = expression(rho), y = "Nombre d'échantillons")Estimated Marginal Means
afternoon | Median | 95% CI | pd | ROPE | % in ROPE
--------------------------------------------------------------------
0 | 3.74 | [3.64, 3.83] | 100% | [-0.10, 0.10] | 0%
1 | 2.50 | [2.41, 2.60] | 100% | [-0.10, 0.10] | 0%
Variable predicted: wait
Predictors modulated: afternoon
Predictors averaged: cafe (1)
On compare le premier modèle (complete pooling model), le troisième modèle (partial pooling model), et le dernier modèle (avec intercept et pente variable).
elpd_diff se_diff elpd_waic se_elpd_waic p_waic se_p_waic waic se_waic
mod5 0.0 0.0 -154.9 10.1 26.4 2.6 309.9 20.1
mod3 -98.9 8.3 -253.9 8.3 18.2 1.5 507.7 16.6
mod2 -99.5 8.3 -254.5 8.4 19.5 1.6 509.0 16.8
mod1 -156.3 13.7 -311.2 10.5 2.1 0.4 622.4 21.1
mod1 mod2 mod3 mod5
1.339634e-68 5.838915e-44 1.087725e-43 1.000000e+00
L’estimation du temps d’attente moyen est plus incertaine lorsqu’on prend en compte de nouvelles sources d’erreur. Cependant, l’erreur du modèle (i.e., ce qui n’est pas expliqué), la variation résiduelle \(\sigma\), diminue…
Estimate Est.Error Q2.5 Q97.5
b_Intercept 3.12029 0.08305618 2.961966 3.278956
sigma 1.14329 0.05821889 1.035996 1.263067
Estimate Est.Error Q2.5 Q97.5
b_Intercept 3.1192094 0.20762244 2.7129150 3.5327098
sigma 0.8220335 0.04318892 0.7431579 0.9115307
Estimate Est.Error Q2.5 Q97.5
b_Intercept 3.7362476 0.21873485 3.3013415 4.1752168
b_afternoon -1.2328637 0.08678994 -1.4036481 -1.0601241
sigma 0.4895081 0.02684615 0.4394686 0.5450169
Les modèles multi-niveaux (ou “modèles mixtes”) sont des extensions naturelles des modèles de régression classiques, où les paramètres de ces derniers se voient eux-même attribués des “modèles”, gouvernés par des hyper-paramètres.
Cette extension permet de faire des prédictions plus précises en prenant en compte la variabilité liée aux groupes ou structures (clusters) présent(e)s dans les données. Autrement dit, en modélisant les populations d’où sont tirés les effets aléatoires (e.g., la population de participants ou de stimuli).
Un modèle de régression classique est équivalent à un modèle multi-niveaux où la variabilité des effets aléatoires serait fixée à \(0\).
La cadre bayésien permet une interprétation naturelle des distributions desquelles proviennent les effets aléatoires (varying effects). En effet, ces distributions peuvent être interprétées comme des distributions a priori, dont les paramètres sont estimés à partir des données.
Travailler avec des sujets humains implique un minimum de coopération réciproque. Mais ce n’est pas toujours le cas. Une partie non-négligeable des étudiants qui s’inscrivent pour passer des expériences de psychologie ne se présentent pas le jour prévu… On a voulu estimer la probabilité de présence d’un étudiant inscrit en fonction de l’envoi (ou non) d’un mail de rappel (cet exemple est présenté en détails dans deux blogposts, accessibles ici, et ici).
reminder researcher presence absence total
1 -0.5 8 38 54 92
2 0.5 7 64 16 80
3 0.5 8 81 11 92
4 -0.5 4 61 34 95
5 0.5 6 82 6 88
6 -0.5 10 23 58 81
\[ \begin{aligned} \color{orangered}{y_{i}} \ &\color{orangered}{\sim \mathrm{Binomial}(n_{i}, p_{i})} \\ \color{black}{\text{logit}(p_{i})} \ &\color{black}{= \alpha_{\text{researcher}_{[i]}} + \beta_{\text{researcher}_{[i]}} \times \text{reminder}_{i}} \\ \color{steelblue}{\begin{bmatrix} \alpha_{\text{researcher}} \\ \beta_{\text{researcher}} \\ \end{bmatrix}} \ & \color{steelblue}{\sim \mathrm{MVNormal}\left(\begin{bmatrix} \alpha \\ \beta \end{bmatrix}, \textbf{S}\right)} \\ \color{steelblue}{\textbf{S}} \ &\color{steelblue}{= \begin{pmatrix} \sigma_{\alpha} & 0 \\ 0 & \sigma_{\beta} \\ \end{pmatrix} \textbf{R} \begin{pmatrix} \sigma_{\alpha} & 0 \\ 0 & \sigma_{\beta} \\ \end{pmatrix}} \\ \color{steelblue}{\alpha} \ &\color{steelblue}{\sim \mathrm{Normal}(0, 1)} \\ \color{steelblue}{\beta} \ &\color{steelblue}{\sim \mathrm{Normal}(0, 1)} \\ \color{steelblue}{(\sigma_{\alpha}, \sigma_{\beta})}\ &\color{steelblue}{\sim \mathrm{HalfCauchy}(0, 1)} \\ \color{steelblue}{\textbf{R}} \ &\color{steelblue}{\sim \mathrm{LKJcorr}(2)} \\ \end{aligned} \]
Il s’agit du même modèle de régression logistique vu au Cours n°06, avec une fonction de lien logit, mais cette fois-ci sur plusieurs niveaux.
Attention, les estimations ci-dessous sont dans l’espace log-odds…
Estimate Est.Error Q2.5 Q97.5
b_Intercept 0.8048514 0.2601641 0.2620946 1.304932
b_reminder 2.8921913 0.3472082 2.1004525 3.487141
sd_researcher__Intercept 0.7821731 0.2323250 0.4528986 1.344695
sd_researcher__reminder 0.9024132 0.3337226 0.4323950 1.710281
Afin de pouvoir les interpréter il faut appliquer la transformation logit-inverse. Par exemple, la probabilité de présence en moyenne (i.e., quel que soit le chercheur et pour toutes conditions confondues) est égale à \(p = \exp(\alpha) / (1 + \exp(\alpha) )\).
On s’est ensuite interrogé sur l’effet du mail de rappel. Ici encore, on ne peut pas interpréter la pente directement… mais on sait que \(\text{exp}(\beta)\) nous donne un odds ratio (i.e., un rapport de cotes).
Envoyer un mail de rappel multiplie par environ 18 le rapport des cotes.
Une manière de représenter les prédictions du modèle est de plotter directement quelques échantillons issus de la distribution a posteriori. On appelle ce genre de plot un “spaghetti plot”.
absence_data %>%
group_by(researcher, total) %>%
data_grid(reminder = seq_range(reminder, n = 1e2) ) %>%
add_linpred_draws(object = mod6, newdata = ., ndraws = 200) %>%
mutate(estimate = plogis(.linpred) ) %>%
ggplot(aes(x = reminder, y = estimate, group = .draw) ) +
geom_hline(yintercept = 0.5, lty = 2) +
geom_line(aes(y = estimate, group = .draw), size = 0.5, alpha = 0.1) +
facet_wrap(~researcher, nrow = 2) +
labs(x = "Mail de rappel", y = "Pr(présent)")Plusieurs manières de tester des hypothèses avec brms. La fonction hypothesis() calcule un evidence ratio (équivalent au Bayes factor). Lorsque l’hypothèse testée est une hypothèse ponctuelle (on teste une valeur précise du paramètre, e.g., \(\theta = 0\)), cet evidence ratio est approximé via la méthode de Savage-Dickey. Cette méthode consiste simplement à comparer la densité du point testé accordée par le prior à la densité accordée par la distribution a posteriori.
Hypothesis Tests for class b:
Hypothesis Estimate Est.Error CI.Lower CI.Upper Evid.Ratio Post.Prob Star
1 (reminder) = 0 2.89 0.35 2.1 3.49 0 0 *
---
'CI': 90%-CI for one-sided and 95%-CI for two-sided hypotheses.
'*': For one-sided hypotheses, the posterior probability exceeds 95%;
for two-sided hypotheses, the value tested against lies outside the 95%-CI.
Posterior probabilities of point hypotheses assume equal prior probabilities.
[1] Inf
Voir la vignette détaillée du paquet bayestestR concernant les facteurs de Bayes : https://easystats.github.io/bayestestR/articles/bayes_factors.html.
data.frame(prior = hyp1$prior_samples$H1, posterior = hyp1$samples$H1) %>%
gather(type, value) %>%
mutate(type = factor(type, levels = c("prior", "posterior") ) ) %>%
ggplot(aes(x = value) ) +
geom_histogram(bins = 50, alpha = 0.8, col = "white", fill = "steelblue") +
geom_vline(xintercept = 0, lty = 2, size = 1) +
facet_wrap(~type, scales = "free") +
labs(x = expression(beta[reminder]), y = "Nombre d'échantillons")Une deuxième solution consiste à étendre l’approche par comparaison de modèles. Tester une hypothèse revient à comparer deux modèles : un modèle avec l’effet d’intérêt et un modèle sans l’effet d’intérêt.
prior7 <- c(
prior(normal(0, 10), class = Intercept, coef = ""),
prior(exponential(1), class = sd),
prior(lkj(2), class = cor)
)
mod7 <- brm(
presence | trials(total) ~ 1 + reminder + (1 + reminder | researcher),
family = binomial(link = "logit"),
prior = prior7,
data = absence_data,
# this line is important for bridgesampling
save_pars = save_pars(all = TRUE),
warmup = 2000, iter = 1e4, cores = 4,
control = list(adapt_delta = 0.95) )
mod8 <- brm(
presence | trials(total) ~ 1 + (1 + reminder | researcher),
family = binomial(link = "logit"),
prior = prior7,
data = absence_data,
save_pars = save_pars(all = TRUE),
warmup = 2000, iter = 1e4, cores = 4,
control = list(adapt_delta = 0.95) )On peut ensuite comparer la vraisemblance marginale de ces modèles, c’est à dire calculer un Bayes factor. Le paquet brms propose la méthode bayes_factor() qui repose sur une approximation de la vraisemblance marginale via le paquet bridgesampling (Gronau et al., 2017).
Estimated Bayes factor (based on medians of log marginal likelihood estimates)
in favor of mod7 over mod8: 2040385.29720
Range of estimates: 1854375.02469 to 2360917.83565
Interquartile range: 123700.20764
On peut également s’intéresser aux capacités de prédiction de ces deux modèles et les comparer en utilisant des critères d’information. La fonction waic() calcule le Widely Applicable Information Criterion (cf. Cours n°07).
Output of model 'mod7':
Computed from 32000 by 20 log-likelihood matrix.
Estimate SE
elpd_waic -59.7 2.4
p_waic 10.6 1.4
waic 119.4 4.9
15 (75.0%) p_waic estimates greater than 0.4. We recommend trying loo instead.
Output of model 'mod8':
Computed from 32000 by 20 log-likelihood matrix.
Estimate SE
elpd_waic -59.3 1.5
p_waic 10.3 0.6
waic 118.5 3.1
20 (100.0%) p_waic estimates greater than 0.4. We recommend trying loo instead.
Une autre manière d’examiner les capacités de prédiction d’un modèle est le posterior predictive checking (PPC). L’idée est simple : il s’agit de comparer les données observées à des données simulées à partir de la distribution a posteriori. Une fois qu’on a une distribution a posteriori sur \(\theta\), on peut simuler des données à partir de la posterior predictive distribution :
\[p(\widetilde{y} \given y) = \int p(\widetilde{y} \given \theta) p(\theta \given y) d \theta\]
Si le modèle est un bon modèle, il devrait pouvoir générer des données qui ressemblent aux données qu’on a observées (e.g., Gabry et al., 2019).
On représente ci-dessous la distribution de nos données.
Cette procédure est implémentée dans brms via la méthode pp_check() qui permet de réaliser de nombreux checks. Par exemple, ci-dessous on compare les prédictions a posteriori (n = 100) aux données observées.
En fittant des modèles un peu compliqués, il se peut que vous obteniez des messages d’avertissement du genre “There were x divergent transitions after warmup”. Dans cette situation, on peut ajuster le comportement de Stan directement dans un appel de la fonction brm() en utilisant l’argument control.
On peut par exemple augmenter le pas de l’algorithme, via adapt_delta (par défaut fixé à 0.8), ce qui ralentira probablement l’échantillonnage mais améliorera la validité des échantillons obtenus. Plus généralement, soyez attentifs aux messages d’erreur et d’avertissement générés par brms.
Une liste d’articles de blog sur brms : https://paul-buerkner.github.io/blog/brms-blogposts/.
L’article d’introduction du paquet brms (Bürkner, 2017) et la version “advanced” (Bürkner, 2018).
Un tutoriel sur les modèles de régression logistique ordinale (Bürkner & Vuorre, 2019).
Notre article tutoriel d’introduction aux modèles multi-niveaux avec brms (Nalborczyk et al., 2019).
Application des modèles généralisés additifs multi-niveaux (GAMMs) aux séries temporelles (e.g., M/EEG, iEEG, pupillometry, finger- or mouse-tracking) (Nalborczyk & Bürkner, 2025), voir le package neurogam (interface à brms): https://lnalborczyk.github.io/neurogam/.
La statistique bayésienne est une approche générale de l’estimation de paramètres. Cette approche utilise la théorie des probabilités pour quantifier l’incertitude vis à vis de la valeur des paramètres de modèles statistiques.
Ces modèles sont composés de différents blocs (e.g., fonction de vraisemblance, priors, modèle linéaire ou non-linéaire) qui sont modifiables à souhait. Ce qu’on appelle classiquement “conditions d’application” sont simplement les conséquences des choix de modélisation réalisés par l’utilisateur. Autrement dit, c’est l’utilisateur qui choisit (et ne subit pas) les conditions d’application.
Nous avons vu que le modèle de régression linéaire est un modèle très flexible qui permet de décrire, via la modification de la fonction de vraisemblance et via l’introduction de fonctions de lien, des relations complexes (e.g., non-linéaires) entre variable prédite et variables prédictrices. Ces modèles peuvent gagner en précision par la prise en compte de la variabilité et des structures présentes dans les données (cf. modèles multi-niveaux).
Le paquet brms est un véritable couteau suisse de l’analyse statistique bayésienne en R. Il permet de fitter presque n’importe quel type de modèle de régression. Cela comprend tous les modèles que nous avons vu en cours, mais également bien d’autres. Entre autres, des modèles multivariés (i.e., avec plusieurs variables à prédire), des modèles “distributionnels” (e.g., pour prédire des différence de variance), des modèles additifs, des procesus Gaussiens (Gaussian processes), des modèles issus de la théorie de détection du signal, des modèles de mélange (mixture models), des modèles de diffusion, des modèles non-linéaires…
N’hésitez pas à me contacter pour plus d’informations sur ces modèles ou si vous avez des questions par rapport à vos propres données. Vous pouvez aussi contacter le créateur du paquet brms, très actif en ligne (voir son site). Voir aussi le forum Stan.
Reaction Days Subject
1 249.5600 0 308
2 258.7047 1 308
3 250.8006 2 308
4 321.4398 3 308
5 356.8519 4 308
6 414.6901 5 308
7 382.2038 6 308
8 290.1486 7 308
9 430.5853 8 308
10 466.3535 9 308
11 222.7339 0 309
12 205.2658 1 309
13 202.9778 2 309
14 204.7070 3 309
15 207.7161 4 309
16 215.9618 5 309
17 213.6303 6 309
18 217.7272 7 309
19 224.2957 8 309
20 237.3142 9 309
À vous de construire les modèles mathématiques et les modèles brms correspondant aux modèles suivants :
Days.Days + un effet aléatoire de Subject (varying intercept).Days + un effet aléatoire de Subject. (varying intercept) + un effet aléatoire de Days (varying slope).Comparez ensuite ces modèles en utilisant les outils discutés aux cours précédents (e.g., WAIC) et concluez.
Data: sleepstudy
Models:
fmod1: Reaction ~ Days + (1 | Subject)
fmod2: Reaction ~ Days + (1 + Days | Subject)
npar AIC BIC logLik -2*log(L) Chisq Df Pr(>Chisq)
fmod1 4 1802.1 1814.8 -897.04 1794.1
fmod2 6 1763.9 1783.1 -875.97 1751.9 42.139 2 7.072e-10 ***
---
Signif. codes: 0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1
Estimate Est.Error Q2.5 Q97.5
b_Intercept 251.97454 6.5202042 239.212332 264.54436
b_Days 10.31425 1.2211528 7.938771 12.73782
sigma 47.77668 2.5350084 43.125370 53.09019
Intercept 298.38868 3.5579627 291.396690 305.36825
lprior -15.69215 0.1655652 -16.040987 -15.39135
lp__ -963.46497 1.2147926 -966.554399 -962.07969
elpd_diff se_diff elpd_waic se_elpd_waic p_waic se_p_waic waic se_waic
mod12 0.0 0.0 -59.3 1.5 10.3 0.6 118.5 3.1
mod11 -0.4 1.5 -59.7 2.4 10.6 1.4 119.4 4.9
mod10 -0.7 1.2 -59.9 2.1 10.8 1.3 119.9 4.2
mod10 mod11 mod12
0.2347421 0.3025556 0.4627023
Ce jeu de données recense des données concernant 2000 élèves dans 100 écoles différentes. L’outcome principal est la popularité de l’élève, évaluée sur une échelle de 1 à 10, et estimée en utilisant une procédure sociométrique (i.e., on a demandé aux élèves de se noter mutuellement). Ces élèves étaient également notés par leurs professeurs (colonne teachpop), sur une échelle de 1 à 7. On dispose comme prédicteurs du genre de l’élève (boy = 0, girl = 1) et de l’expérience du professeur (texp, en années).
À vous d’explorer ce jeu de données, de fitter quelques modèles (avec brms) pour essayer de comprendre quels sont les facteurs qui expliquent (permettent de prédire) la popularité d’un élève…
d <- d %>%
mutate(
# using a sum contrast for gender
sex = ifelse(sex == "boy", -0.5, 0.5),
# centering and standardising teacher experience
texp = scale(texp) %>% as.numeric
)
prior13 <- c(
prior(normal(5, 2.5), class = Intercept),
prior(cauchy(0, 10), class = sd),
prior(cauchy(0, 10), class = sigma)
)
mod13 <- brm(
formula = popular ~ 1 + (1 | school),
data = d,
prior = prior13,
save_all_pars = TRUE,
warmup = 2000, iter = 1e4, cores = 4
)prior14 <- c(
prior(normal(0, 1), class = Intercept),
prior(normal(0, 1), class = b),
prior(cauchy(0, 1), class = sd),
prior(cauchy(0, 10), class = sigma)
)
mod14 <- brm(
formula = popular ~ 1 + texp + (1 | school),
data = d,
prior = prior14,
save_all_pars = TRUE,
warmup = 2000, iter = 1e4, cores = 4
)prior15 <- c(
prior(normal(0, 1), class = Intercept),
prior(normal(0, 1), class = b),
prior(cauchy(0, 1), class = sd),
prior(cauchy(0, 10), class = sigma),
prior(lkj(2), class = cor)
)
mod15 <- brm(
formula = popular ~ 1 + sex + texp + (1 + sex | school),
data = d,
prior = prior15,
save_all_pars = TRUE,
warmup = 2000, iter = 1e4, cores = 4
)# comparaison des WAIC de chaque modèle
model_comparison_table <- loo_compare(mod13, mod14, mod15, mod16, criterion = "waic") %>%
data.frame() %>%
rownames_to_column(var = "model")
weights <- data.frame(weight = model_weights(mod13, mod14, mod15, mod16, weights = "waic") ) %>%
round(digits = 3) %>%
rownames_to_column(var = "model")
left_join(model_comparison_table, weights, by = "model") model elpd_diff se_diff elpd_waic se_elpd_waic p_waic se_p_waic
1 mod16 0.000000 0.000000 -1990.534 32.53232 161.09701 5.171938
2 mod15 -1.661692 2.083195 -1992.196 32.71562 164.01858 5.288606
3 mod14 -448.197634 25.860908 -2438.732 30.21279 92.86614 2.780577
4 mod13 -449.374635 25.930249 -2439.909 30.26516 95.24930 2.855821
waic se_waic weight
1 3981.068 65.06464 0.84
2 3984.391 65.43124 0.16
3 4877.463 60.42557 0.00
4 4879.817 60.53032 0.00
Les prédictions du modèle ne coïncident pas exactement avec les données car ces dernières sont discrètes. Les élèves étaient notés sur une échelle discrète allant de 1 à 10 (un élève ne pouvait pas avoir une note de 3.456). Ce type de données peut-être approximée par une distribution normale (comme nous l’avons fait) mais ce choix n’est pas optimal en termes de prédiction…
On pourrait choisir un modèle qui se rapproche du processus de génération des données. C’est le cas du modèle de régression logistique ordinale. Ce modèle est une sorte de généralisation à plus de 2 catégories du modèle de régression logistique vu au Cours n°06 (voir ce blogpost pour plus de détails, ou le chapitre 11 de Statistical Rethinking), sauf que les catégories sont ordonnées.
\[ \begin{aligned} \color{orangered}{\text{pop}_{i}} \ &\color{orangered}{\sim \mathrm{Categorical}(\mathbf{p})} \\ \color{black}{\text{logit}(p_{k})} \ &\color{black}{= \alpha_{k}} \\ \color{steelblue}{\alpha_{k}} \ &\color{steelblue}{\sim \mathrm{Normal}(0, 10)} \\ \end{aligned} \]
Où la distribution \(\mathrm{Categorical}\) est une distribution discrète qui prend un vecteur de probabilités \(\mathbf{p} = \{p_{1}, p_{2}, p_{3}, p_{4}, p_{5}, p_{6}, p_{7}, p_{8}, p_{9}\}\) qui correspondent aux probabilités cumulées de chaque réponse (entre 1 et 10, 10 ayant une probabilité cumulée de 1).
On définit une série de \(N - 1\) intercepts \(\mathbf{p} = \{p_{1}, p_{2}, p_{3}, p_{4}, p_{5}, p_{6}, p_{7}, p_{8}, p_{9}\}\) sur le logarithme de la cote cumulée (log-cumulative-odds).
\[\text{logit}(p_{k}) = \log \frac{\Pr(y_{i} \leq k)}{1 - \Pr(y_{i} \leq k)} = \alpha_{k}\]
La vraisemblance de l’observation \(k\) (e.g., pop = 3) est donnée par soustraction des proportions cumulées. Cette vraisemblance est représentée par les barres verticales sur le graphique ci-dessous.
\[p_{k} = \Pr(y_{i} = k) = \Pr(y_{i} \leq k) - \Pr(y_{i} \leq k -1)\]
NB : Ce modèle peut prendre plusieurs heures selon votre système…
prior18 <- c(
brms::prior(normal(0, 10), class = Intercept),
brms::prior(normal(0, 10), class = b),
brms::prior(cauchy(0, 10), class = sd)
)
mod18 <- brm(
popular ~ 1 + sex + texp + sex:texp + (1 | school),
data = d,
family = cumulative(link = "logit"),
prior = prior18,
chains = 4, cores = 4,
control = list(adapt_delta = 0.99, max_treedepth = 15),
threads = threading(threads = 2),
file = "models/mod18", backend = "cmdstanr"
)Output of model 'mod17':
Computed from 16000 by 2000 log-likelihood matrix.
Estimate SE
elpd_waic -2086.4 31.3
p_waic 96.6 3.0
waic 4172.8 62.6
8 (0.4%) p_waic estimates greater than 0.4. We recommend trying loo instead.
Output of model 'mod18':
Computed from 4000 by 2000 log-likelihood matrix.
Estimate SE
elpd_waic -2074.2 32.4
p_waic 105.9 2.6
waic 4148.4 64.8
4 (0.2%) p_waic estimates greater than 0.4. We recommend trying loo instead.
Ladislas Nalborczyk - IMSB2026