NuMagSANS: a GPU-accelerated open-source software package for the generic computation of nuclear and magnetic small-angle neutron scattering observables of complex systems

Resumen Técnico de NuMagSANS

Planteamiento del Problema
Evaluar con precisión los observables de dispersión de neutrones de ángulo pequeño (SANS) nucleares y magnéticos en sistemas estructural y magnéticamente complejos presenta desafíos computacionales significativos. Las soluciones de software existentes, como SASfit, mag2exp o los calculadores genéricos dentro de SasView, a menudo carecen de la flexibilidad para manejar datos de densidad nuclear y de magnetización arbitrarios y dependientes de la posición, o de la velocidad computacional requerida para conjuntos de datos a gran escala. Específicamente, existe la necesidad de una herramienta que pueda integrarse perfectamente con las salidas de simulaciones micromagnéticas (por ejemplo, de MuMax3) para computar secciones transversales de SANS en dos dimensiones y funciones de correlación para texturas magnéticas complejas, incluyendo aquellas con características quirales e interferencia interpartícula, sin depender de factores de forma analíticos simplificados.

Metodología
Los autores presentan NuMagSANS, un paquete de software implementado en C++ y CUDA, diseñado para aprovechar las capacidades de procesamiento paralelo de las GPU NVIDIA. La metodología central consiste en:

• Gestión de Entradas: El software acepta datos de espacio real definidos por el usuario para la densidad de longitud de dispersión nuclear ($N(\mathbf{r})$), campos de vectores de magnetización ($\mathbf{M}(\mathbf{r})$) y posiciones de objetos estructurales. Soporta tanto modos atómicos (objetos de puntos discretos) como micromagnéticos (modos de vóxel de continuo).
• Marco Teórico: El código computa los observables de SANS basados en las transformadas de Fourier de los datos de espacio real de entrada. Calcula explícitamente las secciones transversales no polarizadas, polarizadas (POLARIS), semipolarizadas (SANSPOL) y quirales, así como los términos de interferencia núcleo-magnética. La implementación tiene en cuenta diversas geometías de dispersión, incluyendo campos magnéticos aplicados perpendicularmente ($B_0 \perp k_0$) o paralelamente ($B_0 \parallel k_0$) al haz de neutrones incidente.
• Aceleración por GPU: Las transformadas de Fourier en espacio discreto (DSFTs) necesarias para convertir los datos de espacio real en espacio recíproco se ejecutan en paralelo en la GPU. El software separa las partes real e imaginaria de las transformadas de Fourier para adaptarse a las restricciones de los kernels de CUDA, utilizando transformadas de coseno y seno discretas.
• Flujo de Trabajo: El proceso implica el análisis de un archivo de configuración, la carga de datos de espacio real en la RAM, la transferencia de datos a la memoria de la GPU, la realización de sumas de alta velocidad para las amplitudes de dispersión, y el cálculo de secciones transversas, funciones de correlación y funciones de distribución de distancia de pares.
• Salida: El software genera secciones transversales de SANS en dos dimensiones, intensidades con promedio azimutal, descomposiciones de Fourier angulares (intensidades modales) y funciones de correlación tanto de una como de dos dimensiones en formato CSV.

Contribuciones Clave

1. Rendimiento Acelerado por GPU: NuMagSANS ofrece mejoras sustanciales de rendimiento sobre los enfoques convencionales basados en CPU, permitiendo el cálculo rápido de observables de SANS para conjuntos de datos que contienen decenas de millones de celdas de discretización.
2. Flexibilidad y Modularidad: El software permite la importación de datos nucleares y magnéticos arbitrarios dependientes de la posición, proporcionando flexibilidad para analizar sistemas complejos que no pueden describirse mediante factores de forma analíticos simples. Incluye una biblioteca de aproximadamente 100 funciones de respuesta que cubren secciones transversales 2D, funciones de correlación y cantidades con promedio azimutal.
3. Integración con Micromagnetismo: Una característica clave es la integración fluida con herramientas de simulación micromagnética como MuMax3, permitiendo a los investigadores computar directamente observables de SANS magnéticos a partir de configuraciones de espín simuladas.
4. Separación de Contribuciones: El software incluye la funcionalidad para separar los factores de forma de las partículas de las contribuciones de interferencia interpartícula, facilitando el análisis de los efectos del factor de estructura en sistemas ordenados o densamente empaquetados.
5. Conjunto Integral de Observables: Computa una amplia gama de observables, incluyendo términos de dispersión quiral ($\chi$) e interferencia núcleo-magnética, los cuales son críticos para analizar sistemas con rotura de simetría de inversión o texturas magnéticas específicas.

Resultados y Verificación
El artículo valida NuMagSANS a través de varios casos de prueba con soluciones analíticas conocidas:

• Esfera Uniformemente Magnetizada: El software reproduce correctamente las secciones transversales de SANS analíticas y las funciones de distribución de distancia de pares para una esfera con magnetización uniforme, incluyendo la simetría de cuatro pliegues característica en la dispersión de cambio de espín (spin-flip).
• Textura de Vórtice Lineal: NuMagSres modela con éxito un nanomagneto esférico con una textura de vórtice lineal, capturando con precisión los términos de dispersión quiral no nulos y la anisotropía angular específica derivada de la rotura de la simetría de inversión.
• Interferencia Interpartícula: El software se aplicó a un conjunto de 800 nanopartículas de hierro en configuraciones diluidas, aleatorias y ordenadas (cúbica simple). Demostró con éxito la transición de oscilaciones suaves del factor de forma en sistemas diluidos hacia señales ruidosas con características tipo pico de Bragg en redes ordenadas, capturando efectos colectivos.
• Escalabilidad del Rendimiento: Los experimentos de tiempo de ejecución en una GPU NVIDIA RTX 3090 mostraron que el tiempo de ejecución escala aproximadamente de forma lineal con el número de celdas de discretización a lo largo de dos órdenes de magnitud (desde $\sim 4 \times 10^4$ hasta $\sim 10^8$ celdas). Un sistema con $\sim 3.38 \times 10^6$ celdas fue procesado en aproximadamente 60 segundos.

Significado
Los autores afirman que NuMagSANS proporciona una herramienta potente y eficiente para el análisis avanzado de SANS, cerrando la brecha entre las simulaciones a escala de escritorio y los enfoques de computación de alto rendimiento a gran escala. Su capacidad para manejar datos de espacio real complejos y definidos por el usuario, junto con su integración con resolvedores micromagnéticos, permite obtener información detallada sobre las características estructurales y magnéticas de las muestras, particularmente en materiales magnéticos con texturas de espín intrincadas. Al permitir el examen sistemático de cómo las correlaciones interpartícula influyen en las secciones transversales de dispersión, NuMagSANS ofrece una plataforma versátil para avanzar en la comprensión de los fenómenos de dispersión en sistemas puramente magnéticos e híbridos. El software se distribuye como código abierto bajo la Licencia MIT para facilitar la reproducibilidad y el desarrollo posterior dentro de la comunidad.