MadVfold: accelerating NLO event generation and reducing negative weights with SIMD vectorization and GPUs

Dans le monde à enjeux élevés de la physique des particules, les scientifiques du Grand Collisionneur de Hadrons font s'entrechoquer des protons pour recréer les conditions de l'univers primitif. Pour donner un sens aux débris, ils s'appuient sur des simulations informatiques qui prédisent ce qui devrait se passer lors de ces collisions. Cependant, calculer ces prédictions avec une précision extrême est une tâche monumentale. Les méthodes les plus précises, connues sous le nom de calculs au niveau suivant de l'ordre (next-to-leading-order), sont incroyablement lentes et coûteuses en ressources de calcul. Un obstacle majeur dans ces simulations est l'apparition de « poids négatifs », une particularité mathématique où certains événements simulés portent une valeur négative. Pour annuler le bruit que ces valeurs négatives créent et obtenir une image claire, les chercheurs doivent générer des lots d'événements bien plus vastes que d'habitude, consommant ainsi des quantités énormes de puissance de calcul et de temps.

Pour résoudre ce problème, les physiciens ont développé une technique appelée « pliage » (folding). Imaginez que vous essayiez d'estimer la valeur d'une intégrale complexe en échantillonnant un seul point ; cela peut être bruité et imprécis. Le pliage revient à calculer la même fonction complexe pour plusieurs variations différentes des variables cinématiques d'un même événement, puis à en faire la moyenne. Cette approche fournit une estimation plus stable de l'intégrale, ce qui aide à lisser le bruit statistique et à réduire le besoin de poids négatifs. Bien qu'efficace, cette méthode est elle-même très lente car elle nécessite que l'ordinateur effectue la même tâche complexe de calcul de nombreuses fois pour un seul événement. Une nouvelle approche décrite par Andrea Valassi, au CERN, vise à accélérer ce processus en utilisant le matériel informatique moderne pour effectuer ces calculs répétés simultanément, plutôt que les uns après les autres.

Le travail de Valassi introduit une méthode appelée « pliage vectorisé », qui exploite la puissance de traitement parallèle des cartes graphiques modernes et des processeurs avancés. Au lieu de demander à l'ordinateur de calculer la physique pour une variation spécifique d'un événement, puis la suivante, et ainsi de suite, le nouveau logiciel regroupe des dizaines de ces variations. Il envoie ensuite ce lot entier au processeur de l'ordinateur, qui calcule tous ces éléments en même temps. Il s'agit d'un changement significatif par rapport à la manière traditionnelle dont ces simulations fonctionnent, qui traite les événements de manière séquentielle. En réorganisant le logiciel pour tirer parti de cette capacité de parallélisation, le chercheur a créé un nouvel outil appelé MADVFOLD, conçu pour fonctionner avec le package de simulation largement utilisé Madgraph5_aMC@NLO.

Les résultats de cette réorganisation sont frappants. Dans des tests impliquant la collision d'électrons et de positrons pour produire des quarks de fond (bottom quarks), la nouvelle méthode a considérablement réduit le temps nécessaire pour générer les événements simulés. En utilisant les processeurs avancés que l'on trouve dans les centres de données modernes, la technique de pliage vectorisé a rendu le calcul environ six à neuf fois plus rapide que la méthode standard. Lorsque les calculs ont été déchargés vers un processeur graphique (GPU), l'accélération a été encore plus prononcée, atteignant près de dix fois la vitesse de l'approche originale. Ces gains ont été obtenus non pas en modifiant la physique sous-jacente, mais en changeant la manière dont l'ordinateur exécute les mathématiques, lui permettant de gérer la charge lourde du pliage de manière beaucoup plus efficace.

La recherche a également exploré si cette approche parallèle pouvait être appliquée à la génération d'événements sans la technique de pliage elle-même. Même dans ce scénario « déplié » (unfolded), où l'objectif est simplement d'accélérer la simulation standard, le nouveau logiciel a délivré une accélération d'un facteur trois. Cela suggère que les changements architecturaux apportés pour supporter le pliage ont des bénéfices plus larges, rendant l'ensemble du pipeline de simulation plus efficace. Le travail a été développé grâce à un processus novateur où le chercheur s'est fortement appuyé sur des modèles de langage de grande taille pour l'aider à écrire et tester le code, une méthode qui a permis au projet de passer du concept au logiciel opérationnel en seulement environ six semaines.

Malgré les accélérations impressionnantes, l'article note que le pliage reste une technique coûteuse en termes de calcul. Même avec cette accélération matérielle, exécuter une simulation avec pliage prend toujours nettement plus de temps que d'en exécuter une sans pliage. Cependant, ce compromis est souvent nécessaire car le pliage réduit le nombre d'événements à poids négatif, ce qui réduit en retour le nombre total d'événements nécessaires pour atteindre un résultat statistiquement significatif. En rendant le processus de pliage lui-même beaucoup plus rapide, la nouvelle méthode rend ce compromis nécessaire plus gérable pour les expériences futures, y compris le programme de Haute Luminosité du LHC à venir. L'étude conclut que, bien que le logiciel soit prêt pour des tests et des raffinements supplémentaires, il représente une étape importante pour rendre les simulations de physique des particules de haute précision plus réalisables face aux ensembles de données massifs attendus dans les années à venir.