G-Adaptivity: optimised graph-based mesh relocation for finite element methodsJames Rowbottom, Georg Maierhofer, Teo Deveney et al.
We present a novel, and effective, approach to achieve optimal mesh relocation in finite element methods (FEMs). The cost and accuracy of FEMs is critically dependent on the choice of mesh points. Mesh relocation (r-adaptivity) seeks to optimise the mesh geometry to obtain the best solution accuracy at given computational budget. Classical r-adaptivity relies on the solution of a separate nonlinear "meshing" PDE to determine mesh point locations. This incurs significant cost at remeshing, and relies on estimates that relate interpolation- and FEM-error. Recent machine learning approaches have focused on the construction of fast surrogates for such classical methods. Instead, our new approach trains a graph neural network (GNN) to determine mesh point locations by directly minimising the FE solution error from the PDE system Firedrake to achieve higher solution accuracy. Our GNN architecture closely aligns the mesh solution space to that of classical meshing methodologies, thus replacing classical estimates for optimality with a learnable strategy. This allows for rapid and robust training and results in an extremely efficient and effective GNN approach to online r-adaptivity. Our method outperforms both classical, and prior ML, approaches to r-adaptive meshing. In particular, it achieves lower FE solution error, whilst retaining the significant speed-up over classical methods observed in prior ML work.
2.4APApr 7
On scattering for NLS: rigidity properties and numerical simulations via the lens transformRémi Carles, Georg Maierhofer
We analyse the scattering operator associated with the defocusing nonlinear Schr{ö}dinger equation which captures the evolution of solutions over an infinite time-interval under the nonlinear flow of this equation. The asymptotic nature of the scattering operator (involving unbounded time) makes its computation particularly challenging. We overcome this by exploiting the space-time compactification provided by the lens transform, marking the first use of this technique in numerical simulations. This results in a highly efficient and reliable methodology for computing the scattering operator in various regimes. In developing this approach we introduce and prove several new identities and theoretical properties of the scattering operator. We support our construction with several numerical experiments which we show to agree with known analytical properties of the scattering operator, and also address the case of long-range scattering for the one-dimensional cubic Schr{ö}dinger equation. Our simulations permit us to further explore regimes beyond current analytical understanding, and lead us to formulate new conjectures concerning fixed and rotating points of the operator, as well as its existence in the long-range setting for both defocusing and focusing cases.
Stable Hermite transforms via the Golub-Welsch algorithmMarcus Webb, Georg Maierhofer
We introduce an efficient stable algorithm for transforms associated with expansions in Hermite functions interpolated at Hermite polynomial roots. The Hermite transform matrix can be factorised into a diagonal component and an orthogonal matrix, leading to a form which allows both the forward and inverse Hermite transforms to be computed stably. Our novel algorithm computes this factorisation based on the eigendecomposition of the Jacobi matrix associated with Hermite functions. Through numerical experiments, we demonstrate the stability and efficiency gains of this novel method over prior work. Numerical experiments show that the new approach matches or improves on the accuracy of existing stabilized methods, is substantially faster in practice, and enables reliable use of large Hermite expansions in downstream PDE computations. We also provide an open-source implementation, together with reference implementations of previous methods, to facilitate adoption by the community.
8.5NAApr 12
Computing nonlinear Schrödinger equations with Hermite functions beyond harmonic trapsValeria Banica, Georg Maierhofer, Katharina Schratz
Hermite basis functions are a powerful tool for the spatial discretisation of Schrödinger equations with harmonic potential. In this work, we show that their stability properties extend to the simulation of Schrödinger equations without harmonic potential, thus making them a natural basis for the computation of nonlinear dispersive equations on unbounded domains. Building on this spatial discretisation, we introduce a novel unconditionally stable numerical method for the derivative nonlinear Schrödinger equation. Our theoretical results are supported with extensive numerical examples.
12.3LGMay 15
CTF4Nuclear: Common Task Framework for Nuclear Fission and Fusion ModelsStefano Riva, Carolina Introini, Antonio Cammi et al.
The demand for clean energy is ever increasing, with new nuclear technologies presenting a complementary solution to renewable energies. However, designing and operating these systems is exceptionally difficult, given the complexity of the physical phenomena that interact to form the system dynamics. While high-fidelity simulations help to understand the non-linear, multi-physics interactions within a reactor, they are computationally expensive and rarely suitable for real-time applications. Furthermore, model-based approaches are inherently sensitive to simplifying assumptions required to derive their governing equations and parameters, leading to inevitable discrepancies with real-world measurements. In contrast, Machine Learning (ML) methods have the potential to generate reliable surrogate models which may be able to quickly predict the system's behaviour. However, the number of data-driven methods that can potentially be used for this task is large and diverse. In a safety-critical setting such as nuclear engineering, a fair comparison of different ML methods, and a clear understanding of their advantages and limitations, is of paramount importance. To address this, we introduce a Common Task Framework (CTF) for ML in nuclear engineering, building upon previous efforts in dynamical systems and seismology. This CTF considers a curated set of datasets from different nuclear and nuclear-adjacent systems. The CTF evaluates the performance of a method on 12 established metrics, alongside a new paradigm focused on system monitoring from sparse measurements only. We illustrate the framework by benchmarking standard ML baselines against these datasets, revealing current method limitations. Our vision is to replace ad hoc comparisons with standardized evaluations on hidden test sets, raising the bar for rigour and reproducibility in scientific ML for the nuclear industry.
2.3NAJul 4, 2024
A fast neural hybrid Newton solver adapted to implicit methods for nonlinear dynamicsTianyu Jin, Georg Maierhofer, Katharina Schratz et al.
The use of implicit time-stepping schemes for the numerical approximation of solutions to stiff nonlinear time-evolution equations brings well-known advantages including, typically, better stability behaviour and corresponding support of larger time steps, and better structure preservation properties. However, this comes at the price of having to solve a nonlinear equation at every time step of the numerical scheme. In this work, we propose a novel deep learning based hybrid Newton's method to accelerate this solution of the nonlinear time step system for stiff time-evolution nonlinear equations. We propose a targeted learning strategy which facilitates robust unsupervised learning in an offline phase and provides a highly efficient initialisation for the Newton iteration leading to consistent acceleration of Newton's method. A quantifiable rate of improvement in Newton's method achieved by improved initialisation is provided and we analyse the upper bound of the generalisation error of our unsupervised learning strategy. These theoretical results are supported by extensive numerical results, demonstrating the efficiency of our proposed neural hybrid solver both in one- and two-dimensional cases.
Learning the Sampling Pattern for MRIFerdia Sherry, Martin Benning, Juan Carlos De los Reyes et al.
The discovery of the theory of compressed sensing brought the realisation that many inverse problems can be solved even when measurements are "incomplete". This is particularly interesting in magnetic resonance imaging (MRI), where long acquisition times can limit its use. In this work, we consider the problem of learning a sparse sampling pattern that can be used to optimally balance acquisition time versus quality of the reconstructed image. We use a supervised learning approach, making the assumption that our training data is representative enough of new data acquisitions. We demonstrate that this is indeed the case, even if the training data consists of just 7 training pairs of measurements and ground-truth images; with a training set of brain images of size 192 by 192, for instance, one of the learned patterns samples only 35% of k-space, however results in reconstructions with mean SSIM 0.914 on a test set of similar images. The proposed framework is general enough to learn arbitrary sampling patterns, including common patterns such as Cartesian, spiral and radial sampling.
2.5CVFeb 8, 2018
Peekaboo - Where are the Objects? Structure Adjusting SuperpixelsGeorg Maierhofer, Daniel Heydecker, Angelica I. Aviles-Rivero et al.
This paper addresses the search for a fast and meaningful image segmentation in the context of $k$-means clustering. The proposed method builds on a widely-used local version of Lloyd's algorithm, called Simple Linear Iterative Clustering (SLIC). We propose an algorithm which extends SLIC to dynamically adjust the local search, adopting superpixel resolution dynamically to structure existent in the image, and thus provides for more meaningful superpixels in the same linear runtime as standard SLIC. The proposed method is evaluated against state-of-the-art techniques and improved boundary adherence and undersegmentation error are observed, whilst still remaining among the fastest algorithms which are tested.