COMMET: orders-of-magnitude speed-up in finite element method via batch-vectorized neural constitutive updates

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Alheit, Benjamin, Peirlinck, Mathias, Kumar, Siddhant
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908776807792640
author Alheit, Benjamin
Peirlinck, Mathias
Kumar, Siddhant
author_facet Alheit, Benjamin
Peirlinck, Mathias
Kumar, Siddhant
contents Constitutive evaluations often dominate the computational cost of finite element (FE) simulations whenever material models are complex. Neural constitutive models (NCMs) offer a highly expressive and flexible framework for modeling complex material behavior in solid mechanics. However, their practical adoption in large-scale FE simulations remains limited due to significant computational costs, especially in repeatedly evaluating stress and stiffness. NCMs thus represent an extreme case: their large computational graphs make stress and stiffness evaluations prohibitively expensive, restricting their use to small-scale problems. In this work, we introduce COMMET, an open-source FE framework whose architecture has been redesigned from the ground up to accelerate high-cost constitutive updates. Our framework features a novel assembly algorithm that supports batched and vectorized constitutive evaluations, compute-graph-optimized derivatives that replace automatic differentiation, and distributed-memory parallelism via MPI. These advances dramatically reduce runtime, with speed-ups exceeding three orders of magnitude relative to traditional non-vectorized automatic differentiation-based implementations. While we demonstrate these gains primarily for NCMs, the same principles apply broadly wherever for-loop based assembly or constitutive updates limit performance, establishing a new standard for large-scale, high-fidelity simulations in computational mechanics.
format Preprint
id arxiv_https___arxiv_org_abs_2510_00884
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle COMMET: orders-of-magnitude speed-up in finite element method via batch-vectorized neural constitutive updates
Alheit, Benjamin
Peirlinck, Mathias
Kumar, Siddhant
Computational Engineering, Finance, and Science
Machine Learning
Constitutive evaluations often dominate the computational cost of finite element (FE) simulations whenever material models are complex. Neural constitutive models (NCMs) offer a highly expressive and flexible framework for modeling complex material behavior in solid mechanics. However, their practical adoption in large-scale FE simulations remains limited due to significant computational costs, especially in repeatedly evaluating stress and stiffness. NCMs thus represent an extreme case: their large computational graphs make stress and stiffness evaluations prohibitively expensive, restricting their use to small-scale problems. In this work, we introduce COMMET, an open-source FE framework whose architecture has been redesigned from the ground up to accelerate high-cost constitutive updates. Our framework features a novel assembly algorithm that supports batched and vectorized constitutive evaluations, compute-graph-optimized derivatives that replace automatic differentiation, and distributed-memory parallelism via MPI. These advances dramatically reduce runtime, with speed-ups exceeding three orders of magnitude relative to traditional non-vectorized automatic differentiation-based implementations. While we demonstrate these gains primarily for NCMs, the same principles apply broadly wherever for-loop based assembly or constitutive updates limit performance, establishing a new standard for large-scale, high-fidelity simulations in computational mechanics.
title COMMET: orders-of-magnitude speed-up in finite element method via batch-vectorized neural constitutive updates
topic Computational Engineering, Finance, and Science
Machine Learning
url https://arxiv.org/abs/2510.00884