Pile matérielle et logicielle de l'IA

Quelle approximation est couramment utilisée pour estimer le nombre d'opérations nécessaires à l'entraînement d'un modèle Transformer, avec N le nombre de paramètres et D le nombre de jetons d'entraînement ?

  1. AEnviron 2ND
  2. BEnviron N+D
  3. CEnviron 6ND
Afficher la réponse et l'explication

Réponse : C. Environ 6ND

La propagation avant d'un jeton demande, pour chaque paramètre, une multiplication et une addition, soit environ 2N opérations. La rétropropagation en demandant environ le double, 4N, l'entraînement d'un jeton coûte environ 6N opérations, et D jetons environ 6ND. Par exemple, entraîner un modèle de 7 milliards de paramètres sur 1000 milliards de jetons représente environ 4,2×10²² opérations. Diviser cette valeur par les performances effectives des GPU donne une estimation de la durée d'entraînement.

S'entraîner sur ce thème Retour au thème

Autres questions sur Pile matérielle et logicielle de l'IA

  1. 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) ?
  2. Lors du calcul de la fonction softmax exp(xi)/Σexp(xj), pourquoi soustrait-on souvent le maximum de chaque xi avant le calcul ?
  3. Quel est le principe du « décodage spéculatif », qui accélère l'inférence des grands modèles de langage ?
  4. Pourquoi la consultation de grandes tables d'embeddings, utilisées notamment dans les modèles de systèmes de recommandation, est-elle difficile à traiter pour le matériel d'IA ?
  5. Quel ordre de la pile logicielle d'IA, de la couche la plus proche de l'utilisateur à la plus proche du matériel, est correct ?