Pile matérielle et logicielle de l'IA
Pour un grand modèle de langage de N paramètres, quel est l'ordre de grandeur du nombre d'opérations nécessaires à l'inférence d'un jeton (hors attention) ?
- AEnviron 2N
- BEnviron 6N
- CEnviron N²
Afficher la réponse et l'explication
Réponse : A. Environ 2N
La propagation avant de l'inférence effectue, pour chaque poids, une multiplication avec l'entrée et une accumulation, soit environ 2N opérations par jeton. Pour 7 milliards de paramètres, environ 14 milliards par jeton. C'est peu pour un GPU actuel, mais le décodage doit lire pendant ce temps 14 GB de poids ; c'est donc la bande passante mémoire, et non le calcul, qui détermine le temps.
Continuer : 9 autres questions sur Pile matérielle et logicielle de l'IA