VEQ: a fast parametric Grad--Shafranov solver for fixed-boundary tokamak equilibria with flexible source profiles

El artículo presenta VEQ, un resolvedor paramétrico rápido para equilibrios de tokamak de frontera fija que utiliza perfiles de fuente flexibles y expansiones armónicas para lograr una reconstrucción de equilibrio de baja latencia y alta precisión adecuada para consultas repetidas en flujos de trabajo de acoplamiento de geometría de transporte.

Lo esencial

Imagina un tokamak (un reactor de fusión nuclear con forma de dónut) como un globo gigante e invisible lleno de plasma supercaliente. Para evitar que este globo estalle o colapse, los científicos necesitan saber exactamente qué forma tiene la "piel" magnética que lo mantiene unido y cómo se equilibran las fuerzas en su interior. Esto se llama encontrar el "equilibrio".

Normalmente, calcular este equilibrio es como intentar resolver un rompecabezas 3D masivo donde cada pieza cambia de forma al tocarla. Es preciso, pero toma mucho tiempo, demasiado tiempo si necesitas comprobar la forma cientos de veces por segundo para controlar el reactor.

Este artículo presenta VEQ (Veloce EQuilibrium), una nueva herramienta diseñada para ser el botón de "avance rápido" de estos cálculos. Así es como funciona, utilizando analogías sencillas:

1. El "Cambiaformas" frente al "Pintor de Píxeles"

Los métodos tradicionales son como un Pintor de Píxeles. Intentan mapear todo el campo magnético calculando el valor en millones de diminutos puntos de rejilla (píxeles). Es detallado, pero pesado y lento.

VEQ es más como un Cambiaformas. En lugar de pintar cada píxel, describe la forma del plasma usando un "esqueleto" matemático flexible hecho de unos pocos controles y diales clave (llamados parámetros).

• Piensa en la forma del plasma como una pieza de arcilla.
• Los resolvedores tradicionales intentan esculpir cada pequeña protuberancia de la arcilla.
• VEQ utiliza un conjunto de herramientas de "estiramiento" y "doblez" predefinidas (armónicos matemáticos y polinomios). Solo tienes que girar unos pocos diales para estirar la arcilla hasta darle la forma correcta. Esto hace que el cálculo sea increíblemente rápido porque hay muchos menos diales que girar que píxeles que pintar.

2. El "Traductor Universal"

Uno de los mayores dolores de cabeza en la investigación de la fusión es que diferentes programas de computadora hablan lenguajes distintos. Un programa puede darte la "presión", otro la "corriente" y otro el "factor de seguridad" (una medida de estabilidad).

VEQ actúa como un Traductor Universal. El artículo muestra que VEQ tiene seis "rutas de entrada" diferentes (como diferentes puertos USB). Puedes conectar datos de cualquiera de estas fuentes diferentes, y VEQ los traduce todos a su propio lenguaje interno para resolver el problema.

• La afirmación: Los autores probaron esto alimentando al sistema con los mismos datos perfectos a través de los seis puertos. Descubrieron que, sin importar qué puerto usaras, VEQ producía exactamente el mismo resultado. Esto demuestra que el traductor funciona perfectamente y no introduce errores solo por haber usado un cable de entrada diferente.

3. La relación "Velocidad vs. Precisión"

El artículo no solo dice que es rápido; explora la compensación entre velocidad y precisión, como elegir entre un boceto y una fotografía.

• El Boceto (Pocos Parámetros): Usas muy pocos diales. Es instantáneo (milisegundos) y suficiente para un vistazo rápido, pero podría perderse pequeños detalles.
• La Foto (Muchos Parámetros): Usas muchos diales. Tarda un poco más (sigue siendo muy rápido, alrededor de 19 milisegundos para formas complejas) pero captura la forma con alta precisión.
• El Resultado: Los autores probaron esto en tres tipos diferentes de formas de plasma (una forma de "D" estándar, una de alto rendimiento "H-mode" y una forma compleja con un "punto X"). Descubrieron que, incluso con un pequeño número de diales, VEQ podía recrear las formas complejas con un error tan pequeño que es casi invisible (menos del 0.2% del tamaño del reactor).

4. La "Prueba de Esfuerzo" (Donde aparecen las grietas)

Los autores fueron honestos sobre los límites. Comprobaron dónde el "equilibrio de fuerzas" (la tensión que sostiene el plasma) era perfecto y dónde no lo era.

• El Interior: En el medio del plasma, VEQ es excelente. Las fuerzas están equilibradas casi perfectamente.
• El Borde: Cerca de la piel exterior muy fina (el límite), los errores son ligeramente mayores. Esto se debe a que VEQ utiliza un esqueleto suave y flexible, pero los límites reales del plasma pueden ser dentados o tener esquinas afiladas (como un punto X).
• La Conclusión: VEQ es excelente para el "medio" del reactor. Si necesitas saber exactamente qué está pasando en el borde mismo, es posible que necesites volver a comprobarlo con una herramienta más lenta y detallada. Pero para la mayoría de las tareas de control, VEQ es lo suficientemente rápido y preciso.

5. La Prueba de "Transporte"

Finalmente, probaron si los pequeños errores en la forma de VEQ causarían grandes problemas si se intentara usar para predecir cómo se mueve el calor a través del plasma (una prueba de "transporte").

• El Resultado: Los errores fueron minúsculos (menos del 1%). Es como si midieras una habitación con una regla ligeramente torcida; el error en el tamaño de la habitación no cambiaría tu decisión sobre qué tamaño de alfombra comprar.

Resumen

VEQ es una calculadora nueva y superrápida para reactores de fusión. En lugar de mapear cada punto individual, utiliza un esqueleto matemático flexible para describir la forma del plasma.

• Es rápido: Puede resolver formas complejas en milisegundos.
• Es flexible: Acepta datos de muchas fuentes diferentes.
• Es fiable: Funciona bien para el interior del reactor y es lo suficientemente preciso para la mayoría de los sistemas de control, siempre que se vigile muy de cerca el borde exterior.

Los autores concluyen que VEQ es perfecto para sistemas que necesitan preguntar "¿Cómo se ve el plasma justo ahora?" una y otra vez, como en el control de reactores en tiempo real o al ejecutar miles de simulaciones para encontrar las mejores condiciones de operación.

Resumen técnico

Resumen Técnico: VEQ – Un resolvedor de Grad–Shafranov paramétrico rápido para equilibrios de Tokamak de frontera fija

Planteamiento del Problema
Los flujos de trabajo de modelado integrado para tokamaks (p. ej., IMAS, RAPTOR, SuperCode) requieren consultas repetidas de la geometría del equilibrio, los coeficientes métricos y los perfiles de diagnóstico. Estos flujos de trabajo difieren de las reconstrucciones de alta fidelidad de una sola vez; demandan una representación lo suficientemente compacta para escaneos de parámetros e iteraciones orientadas al control, pero que retenga suficiente información de forma 2D y de balance de fuerzas para ser útil más allá de los cierres analíticos simples. Las herramientas existentes suelen caer en dos extremos: los cierres de ingeniería de bajo orden (p. ej., modelos Miller) son rápidos pero carecen de diagnósticos residuales, mientras que los resolvedores de equilibrio completos (p. ej., CHEASE, EFIT, DESC) proporcionan salidas nativas de alta fidelidad pero suelen ser demasiado costosos computacionalmente o pesados en infraestructura para su uso en bucles internos. Existe una brecha para una representación continua de frontera fija que preserve la utilidad de la forma 2D y los diagnósticos residuales, permaneciendo económica de regenerar.

Metodología
El artículo presenta VEQ (Veloce EQuilibrium), un marco paramétrico compacto diseñado para llenar este régimen intermedio. La implementación específica evaluada, VEQPy, es un resolvedor de Grad–Shafranov axisimétrico de frontera fija.

• Representación Paramétrica: El equilibrio se define mediante un mapa de superficie de flujo paramétrico $(\rho, \theta) \to [R(\rho, \theta), Z(\rho, \theta)]$. La geometría se parametriza utilizando armónicos de superficie de flujo tipo MXH (ángulos poloidales distorsionados) y polinomios de Chebyshev desplazados para los perfiles radiales. Las incógnitas activas son los coeficientes de estos armónicos y perfiles, en lugar de valores nodales de un mapa de flujo poloidal basado en una rejilla.
• Formulación Variacional: El resolvedor impone un residuo de Grad–Shafranov proyectado inducido por variación. En lugar de resolver la ecuación de forma fuerte punto a punto, el método restringe las variaciones admisibles a aquellas generadas por los coeficientes activos de la representación paramétrica. Esto resulta en un sistema de dimensión finita donde el bloque de forma es una proyección de estructura variacional (condición de Petrov–Galerkin) y los bloques de perfil se cierran mediante proyecciones de momentos.
• Rutas de Entrada Flexibles: Para asegurar la compatibilidad con diversos inputs provenientes de estudios analíticos, archivos de reconstrucción y modelos de control, VEQ soporta seis rutas de entrada (PF, PP, PI, PJ1, PJ2, PQ). Estas rutas aceptan gradientes de presión, funciones de campo toroidal, gradientes de flujo poloidal, corriente toroidal encerrada, densidad de corriente o perfiles del factor de seguridad, permitiendo que todas las rutas alimenten al mismo operador de residuo de dimensión finita a través de un mapeo a una representación de fuente normalizada común.
• Implementación: El resolvedor utiliza un resolvedor no lineal (método híbrido de Powell de SciPy) con un backend acelerado (Numba). Separa la configuración (precomputación dependiente de la rejilla, datos del operador) de la fase de resolución para permitir consultas de baja latencia repetidas.

Contribuciones Clave

1. Representación Paramétrica de Dimensión Finita: Definición de una representación de superficie de flujo compacta de MXH–Chebyshev y la derivación del residuo de Grad–Shafranov proyectado inducido por variación, incluyendo las variaciones convectivas asociadas con el movimiento de la superficie de flujo.
2. Cierres de Fuente Específicos por Ruta: Introducción de un marco que mapea entradas 1D heterogéneas (presión, corriente, factor de seguridad, etc.) a un sistema de residuo común, distinguiendo la consistencia de la ruta de entrada del error de aproximación geométrica.
3. Enfoque de Diagnóstico Dual: Reporte tanto de la norma del residuo proyectado (minimizada por el resolvedor) como de los residuos de la forma fuerte muestreados punto a punto. Esto distingue la convergencia del sistema reducido del cribado del balance de fuerzas puntual.
4. Evaluación Comparativa (Benchmarking): Pruebas exhaustivas en casos de Solov'ev, CHEASE y derivados de EFIT (G-EQDSK), cuantificando la consistencia de la ruta de entrada, el error de reconstrucción de la superficie de flujo, las compensaciones entre precisión–costo y la latencia de resolución repetida.

Resultos

• Consistencia de Ruta: Pruebas controladas utilizando entradas suaves y mutuamente compatibles generadas desde un equilibrio de referencia común demostraron que todas las seis rutas de entrada convergen al mismo equilibrio de frontera fija paramétrico, con diferencias atribuibles únicamente a las rutas de recuperación numérica (p. ej., pasos de integración/diferenciación).
• Precisión y Latencia: El resolvedor fue probado en tres casos representativos: un caso Solov'ev con forma de D, un caso CHEASE de modo H y un caso EFIT con punto X.
  • Forma de D: Logró un error de forma normalizado al radio menor de $1.4 \times 10^{-3}$ con 9 parámetros activos y un tiempo de resolución (solo solve) de 1.6 ms.
  • Modo H: Logró un error de $1.1 \times 10^{-3}$ con 65 parámetros y un tiempo de resolución de 19 ms.
  • Punto X: Logró un error de $1.9 \times 10^{-3}$ con 94 parámetros y un tiempo de resolución de 15 ms.
• Diagnósticos de Residuo: Los diagnósticos de la forma fuerte puntual revelaron que el enriquecimiento de la representación activa mejora principalmente el balance de fuerzas interior. En los casos de Modo H y Punto X, los valores de RMS global y de máximo residuo permanecieron dominados por las contribuciones cercanas a la frontera, indicando que el ansatz de frontera fija no puede compensar totalmente los desajustes de ajuste de frontera o las estructuras singulares (como los puntos X) mediante variaciones de forma interior por sí solo.
• Acoplamiento de Transporte: En una prueba aislada de acoplamiento de transporte-geometría 1D, la respuesta del perfil de temperatura ante los errores de geometría de VEQ se mantuvo por debajo de aproximadamente el 1%.

Significación y Reivindicaciones
El artículo afirma que VEQ es adecuado para consultas repetidas de geometría de equilibrio en flujos de trabajo de modelado integrado y control, siempre que se retengan los diagnósticos puntuales para cribar casos que requieran refinamiento de frontera, corrección local o resolvedores de mayor fidelidad.

Los autores posicionan a VEQ como complementario a los resolvedores de reconstrucción de alta fidelidad, no como un reemplazo universal. Su valor principal reside en proporcionar una representación continua y compacta con residuos proyectados explícitos y diagnósticos muestreados que son económicos de regenerar. El documento establece explícitamente que los resultados de tiempo reflejan la latencia de "solo resolución" tras la configuración y no deben interpretarse como un reemplazo de los tiempos de los pipelines completos que incluyen preprocesamiento, E/S y discretización. El método está destinado a cerrar la brecha entre los cierres analíticos y los pipelines nativos de resolución completa, ofreciendo un equilibrio entre riqueza geométrica y eficiencia computacional.