HIPED: machine learning framework for spherical tokamak pedestal prediction and optimization

Abstract We introduce a machine learning framework, HIPED (HeIght and width Predictor for Edge Dynamics), for predicting and optimizing pedestal and core performance in spherical tokamak plasmas. Trained on pedestal and core datasets from the third MAST-U campaign, HIPED provides accurate estimates of pedestal height vs. width. The results reveal notable differences compared with conventional aspect-ratio studies; for instance, a simple power-law relation between pedestal width and height has very low accuracy. Instead, additional parameters such as normalized plasma pressure, elongation, and Greenwald fraction significantly improve accuracy. HIPED can also be trained only on ‘control room parameters’ to inform experimentalists of which controllable parameters to adjust for improving core-integrated performance. The framework further includes a multi-objective optimization scheme that helps guide experimental planning and optimization. We find Pareto-optimal discharges with respect to various features, including distance from edge-localized modes and normalized plasma pressure, track their parameter trajectories over time, and identify the control room parameters required for these Pareto-optimal discharges. This provides a framework for systematically optimizing core and edge performance according to different experimental priorities.