Pile matérielle et logicielle de l'IA
Laquelle des propositions suivantes n'explique pas pourquoi la mémoire nécessaire à l'entraînement est bien plus grande que pour l'inférence du même modèle ?
- AParce que l'entraînement multiplie le nombre de poids par rapport à l'inférence
- BParce qu'il faut conserver les résultats intermédiaires de la propagation avant (activations) pour la rétropropagation
- CParce qu'il faut conserver autant de gradients que de poids, ainsi que l'état de l'optimiseur
Afficher la réponse et l'explication
Réponse : A. Parce que l'entraînement multiplie le nombre de poids par rapport à l'inférence
Le nombre de poids du modèle est le même à l'entraînement et à l'inférence. Mais l'entraînement doit conserver, couche par couche, les résultats intermédiaires de la propagation avant pour calculer les gradients en rétropropagation, ainsi qu'autant de gradients que de poids et les valeurs par poids tenues par l'optimiseur, comme Adam (deux moyennes mobiles, etc.). La mémoire d'entraînement est donc plusieurs fois celle de l'inférence, qui ne demande que les poids, et l'entraînement des grands modèles exige de nombreux GPU.
Continuer : 9 autres questions sur Pile matérielle et logicielle de l'IA