Yeah so the paper I linked on contextual bayesian optimization (https://papers.nips.cc/paper/2019/hash/7876acb66640bad41f1e1...) does a combination of controlling for beta and optimizing linear MHD stability --- which is part of the problem. One of our collaborators has been working precisely on disruption prediction and mitigation but it's on most of our minds: https://www.pppl.gov/news/2021/artificial-intelligence-helps....