On the Acceleration of Pulsar Timing computations using Normalising Flows and Parallelisation

Imaginez que l'univers soit un vaste orchestre cosmique, et qu'au sein de sa symphonie se cachent les battements rythmiques et ténus d'un fond diffus d'ondes gravitationnelles. Il ne s'agit pas d'ondes sonores, mais de rides dans le tissu même de l'espace-temps, créées par des trous noirs massifs dansant ensemble il y a des milliards d'années. Pour entendre ce bourdonnement cosmique, les scientifiques utilisent des « Pulsar Timing Arrays » (réseaux de chronométrage de pulsars). Considérez les pulsars comme les métronomes les plus fiables de l'univers : des étoiles mortes qui tournent si vite et si régulièrement qu'elles projettent des faisceaux de lumière comme des phares. En chronométrant ces éclats avec une précision extrême, les astronomes peuvent détecter si l'espace-temps entre nous et le pulsar a été étiré ou compressé par une onde gravitationnelle passante.

Cependant, écouter cette musique cosmique est incroyablement difficile. Les données sont désordonnées, remplies de « bruit » provenant de la rotation instable du pulsar lui-même, du milieu interstellaire agissant comme une lentille brumeuse, et des limites de nos horloges. C'est comme essayer d'entendre un chuchotement dans un ouragan. Pour donner un sens aux données, les scientifiques utilisent des mathématiques complexes pour séparer le signal du bruit, mais les mathématiques sont si lourdes et les données si volumineuses qu'il faut souvent des semaines, voire des mois, pour qu'un seul ordinateur puisse traiter les chiffres. Ce document s'attaque à ce goulot d'étranglement, en posant une question simple mais vitale : pouvons-nous trouver un moyen plus rapide de résoudre ces énigmes cosmiques sans attendre éternellement ?

Les auteurs de cette étude, Churchil Dwivedi, Hiya Shah et Hemanga Tahbildar, ont décidé de tester un nouvel outil de haute technologie appelé « Normalising Flows » (flux de normalisation) contre les anciennes méthodes éprouvées utilisées par la communauté astronomique. Ils ont traité le problème comme une course. D'un côté de la piste, il y avait les coureurs traditionnels : PTMCMCSAMPLER, un bourreau de travail qui utilise une technique appelée « Parallel Tempering » (imaginez l'envoi de nombreux randonneurs à différentes températures pour explorer une chaîne de montagnes, échangeant leurs places pour trouver les sommets les plus hauts), et PARALLEL BILBY, un coureur plus récent qui utilise le « Nested Sampling » (comme une équipe d'explorateurs réduisant systématiquement une zone de recherche). De l'autre côté, ils ont introduit un sprinteur nommé POCOMC, qui utilise une astuce ingénieuse : les « Normalising Flows ».

Pour comprendre l'astuce, imaginez le paysage de données comme une pelote de laine tordue et nouée. Les méthodes traditionnelles tentent de la démêler en tirant sur les extrémités une par une, ce qui prend beaucoup de temps. POCOMC, cependant, utilise une « baguette magique » mathématique (le Normalising Flow) pour redresser instantanément la pelote en une ligne lisse et simple, rendant la recherche de la solution beaucoup plus facile. Les chercheurs ont testé ces outils sur des données simulées qui imitaient les observations réelles de pulsars, incluant des scénarios où le bruit était simple et des scénarios où il s'agissait d'un réseau complexe de corrélations entrelacées.

Les résultats étaient un mélange de « le plus rapide sur une seule machine » et « le plus rapide avec une équipe immense ». Lors de l'exécution sur un seul nœud informatique, POCOMC fut le champion incontesté. Il a résolu une recherche complexe qui aurait pris plus de 10 jours à l'ancienne méthode (PTMCMCSAMPLER) en seulement environ 10 heures. Plus impressionnant encore, pour un type spécifique d'analyse de pulsar unique, il a terminé en environ 10 heures, alors que l'ancienne méthode aurait pris des jours. Les auteurs ont constaté que POCOMC était si efficace qu'il n'avait même pas besoin d'un supercalculateur ou de cartes graphiques spéciales pour être incroyablement rapide ; il lui suffisait simplement d'un ordinateur standard fonctionnant intelligemment.

Cependant, l'histoire change lorsque vous introduisez une équipe entière d'ordinateurs. Lorsque les chercheurs ont réparti le travail sur de nombreux nœuds de calcul (jusqu'à 16 nœuds, chacun avec 48 threads), l'architecture PARALLEL BILBY a pris la tête. Elle était la meilleure pour diviser le travail au sein de l'équipe. Pour une recherche impliquant des corrélations complexes à travers l'ensemble du réseau de pulsars (appelée recherche « Hellings et Downs »), PARALLEL BILBY a réduit le temps à environ 100 minutes en utilisant 16 nœuds, alors que POCOMC sur un seul nœud aurait pris beaucoup plus de temps. Curieusement, POCOMC a atteint un mur lorsqu'ils ont essayé d'ajouter trop d'ordinateurs ; après environ 4 nœuds, l'ajout de plus de puissance ne le rendait pas plus rapide, et dans certains cas, cela le ralentissait. Les auteurs soupçonnent que l'étape de la « baguette magique » devient trop encombrée lorsque trop de processeurs tentent de travailler dessus simultanément.

Le papier a également comparé différentes manières de gérer un type spécifique de bruit appelé ECORR. Ils ont trouvé que la méthode « fast-sherman-morrison » était la plus efficace pour calculer ce bruit, tandis que la méthode « sherman-morrison » standard était la plus lente. Ils ont également noté que PTMCMCSAMPLER était le moins efficace de tous, prenant nettement plus de temps que les autres, et que dans certains cas, ajouter plus d'ordinateurs le rendait plus lent plutôt que plus rapide.

En fin de compte, les auteurs suggèrent que l'avenir de la chronométrie des pulsars dépend du choix du bon outil pour la tâche. Si vous travaillez sur un seul ordinateur ou si vous analysez un seul pulsar à la fois, POCOC est la nouvelle étoile, offrant une accélération massive sans nécessiter de supercalculateur. Mais si vous avez accès à un supercalculateur massif avec des dizaines de nœuds et que vous devez analyser l'ensemble du réseau de pulsars à la fois, PARALLEL BILBY est le meilleur choix. Ces conclusions suggèrent qu'en utilisant ces méthodes plus rapides, les scientifiques pourront bientôt explorer des modèles de l'univers beaucoup plus complexes qui étaient auparavant impossibles à calculer, nous rapprochant potentiellement de l'écoute finale de ce bourdonnement cosmique d'ondes gravitationnelles.