A proof-of-concept for automated AI-driven stellarator coil optimization with in-the-loop finite-element calculations
Organizations: Courant Institute of Mathematical Sciences, New York University, New York, NY 10012, USA · Max-Planck-Institut für Plasmaphysik, 85748 Garching, Germany
Abstract
Finding feasible coils for stellarator fusion devices is a critical challenge of realizing this concept for future power plants. Years of research work can be put into the design of even a single reactor-scale stellarator design. To rapidly speed up and automate the workflow of designing stellarator coils, we have designed an end-to-end ``runner'' for performing stellarator coil optimization. The entirety of pre and post-processing steps have been automated; the user specifies only a few basic input parameters, and final coil solutions are updated on an open-source leaderboard. Two policies are available for performing non-stop automated coil optimizations through a genetic algorithm or a context-aware LLM. Lastly, we construct a novel in-the-loop optimization of Von Mises stresses in the coils, opening up important future capabilities for in-the-loop finite-element calculations.
Figures & tables
| Component | Motivation | Outcome |
| Automated coil initialization | Manual coil initialization is usually required for different plasma surfaces. | Circular coils are initialized without coil-coil or coil-surface intersections, and modular coils always interlink the plasma. |
| Automated coil optimization | Manual weight tuning is expensive and very different weights are needed for different plasmas. | Objective weights are no longer user-defined; they are auto-tuned with a penalty method. All SIMSOPT functionality is available. |
| Genetic or LLM-driven policy | Learning the optimal hyperparameters such as thresholds, convergence tolerances, number of iterations, “schedules”, etc. is very challenging. | Genetic algorithm or context-aware LLM makes the decisions about which runs to submit every batch. Can keep a running log of justifications. |
| Automated post-processing | Post-processing requires installing a host of complex simulation codes and substantial training of researchers to accurately use the codes. | Users can run new cases and plasma surfaces on cloud devices (through the continuous integration) where all post-processing is auto-installed and auto-run |
| Automated documentation & leaderboard | Historically, it has been challenging to make precise comparisons between stellarator coil solutions. | Runs receive an overall composite score by a weighted sum of coil objectives. Runs are added in real-time to an open-source, online ReadTheDocs leaderboard. |
| Reproducibility | Historically, it has been challenging to make precise comparisons between stellarator coil solutions. | Plasma surfaces and other parameters are standardized and cannot be altered. Full metadata, all final solutions, and all metrics are saved in a database. |
| Property | 3-coil QA (policy) | 3-coil QA [ 22 ] |
| Total coil length (m) | 182.0 | |
| Max curvature, (m -1 ) | 0.53 | |
| Max mean-squared , (m -2 ) | 0.05 | |
| Min coil–coil dist., (m) | ||
| Min coil–surface dist., (m) | ||
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| Parameter | Default | Description |
| width , height | 0.35 m | Cross-section dimensions |
| E | 100 GPa | Young’s modulus |
| nu | 0.3 | Poisson ratio |
| stress_metric | Scalar metric | |
| fd_step | Finite-difference step | |
| eval_interval | 1 | Evaluate every eval_interval iterations |