modifier - modifier le code - modifier Wikidata
En apprentissage automatique, la machine de Boltzmann restreinte (en anglais : restricted boltzmann machine ou RBM) est un type de réseau de neurones artificiels pour l'apprentissage non supervisé.
Une RBM est constituée de deux couches de neurones : une couche visible représentant les variables d'entrée et une couche cachée représentant les variables latentes apprises par le modèle. Les neurones visibles sont connectés aux neurones cachés, sans connexion au sein de chaque couche. Fondés sur des modèles de probabilité, les états des neurones sont déterminés par des probabilités conditionnelles. L'apprentissage se fait par l'algorithme de contrastive divergence (CD), qui ajuste les poids et biais pour minimiser la divergence entre les distributions des données d'entrée et celles générées par le modèle.
La machine de Boltzmann restreinte est couramment utilisée pour estimer la distribution probabiliste d'un jeu de données. Elle a initialement été inventée sous le nom de Harmonium en 1986 par Paul Smolenski[1]. Elle entre dans le cadre des modèles graphiques et des modèles à base d'énergie[2].
La machine de Boltzmann restreinte est en fait un cas particulier de machine de Boltzmann où les neurones d'une même couche sont indépendants entre eux.
Dans sa forme la plus simple, une machine de Boltzmann est composée d'une couche de neurones qui reçoit l'entrée, ainsi que d'une couche de neurones cachée.
On définit l'énergie pour une configuration de donnée de la manière suivante :
E = − ( ∑ i , j w i j x i h j + ∑ i b i x i + ∑ j c j h j ) {\displaystyle E=-\left(\sum _{i,j}w_{ij}\,x_{i}\,h_{j}+\sum _{i}b_{i}\,x_{i}+\sum _{j}c_{j}h_{j}\right)}
où :
La probabilité conjointe d'avoir une configuration ( x i , h j ) {\displaystyle (x_{i},h_{j})} est alors donnée par[3] :
P ( x i , h j ) = exp ( − E ( x i , h j ) ) / Z {\displaystyle P(x_{i},h_{j})=\exp(-E(x_{i},h_{j}))/Z}
avec :
La machine de Boltzmann s’entraîne à l'aide d'un apprentissage non supervisé. On cherche à minimiser la log-vraisemblance. La dérivée de la log-vraisemblance donne l'expression suivante :
∂ [ − log ( p ( x ( t ) ) ) ] ∂ θ = E h [ ∂ E ( x ( t ) , h ) ∂ θ | x ( t ) ] − E x , y [ ∂ E ( x , h ) ∂ θ ] {\displaystyle {\frac {\partial \left[-\log(p(x^{(t)}))\right]}{\partial \theta }}=\mathbb {E} _{h}\left[{\frac {\partial E(x^{(t)},h)}{\partial \theta }}|x^{(t)}\right]-\mathbb {E} _{x,y}\left[{\frac {\partial E(x,h)}{\partial \theta }}\right]}
On remarque la présence de deux termes dans cette expression, appelés phase positive et phase négative. La phase positive se calcule aisément pour le biais et pour la matrice des poids.
On obtient alors[4] :
E h [ ∂ E ( x ( t ) , h ) ∂ W i j | x ( t ) ] = − h ( x ( t ) ) ∗ x ( t ) T {\displaystyle \mathbb {E} _{h}\left[{\frac {\partial E(x^{(t)},h)}{\partial W_{ij}}}|x^{(t)}\right]=-h(x^{(t)})*{x^{(t)}}^{\mathsf {T}}}
avec h(x) l'état de la couche cachée sachant x donnée par la formule
h ( x ) = s i g m ( W ∗ x + b ) {\displaystyle h(x)=sigm(W*x+b)} .
La partie la plus compliquée est de calculer ce qu'on appelle la phase négative. On ne peut pas la calculer directement car on ne connaît pas la constante de normalisation du système. Pour pouvoir effectuer une descente de gradient, on calcule ce que l'on appelle la reconstruction de l'entrée x ( t ) {\displaystyle x^{(t)}} . En effet, les propriétés de symétrie du système permettent de calculer l'entrée estimée par le modèle, il suffit d'appliquer la formule :
x r e c = W T ∗ h ( x ) + c {\displaystyle x_{rec}=W^{\mathsf {T}}*h(x)+c}
avec c {\displaystyle c} le biais de la couche cachée de neurones H {\displaystyle H} .
De la même manière, on peut recalculer l'état de la couche cachée en réitérant le procédé. Finalement, on peut résumer l'algorithme de descente du gradient ainsi[5] (on parle de l'algorithme de contrastive divergence, couramment abrégé CD-k) :
x <= x(t) h <= W*x + b phasePositive <= -h*Transpose(x) Pour i allant de 1 à k: x = Transpose(W) * h(x) + c h = W*x + b phaseNegative <= -h*transpose(x) gradient <= phasePositive-phaseNegative W <= W + alpha*gradient c <= c + alpha*(x(t)-x) b <= b + alpha*(h(x(t)) - h)