Neural network surrogate and projected gradient descent for fast and reliable finite element model calibration: A case study on an intervertebral disc.

Journal: Computers in biology and medicine
PMID:

Abstract

Accurate calibration of finite element (FE) models is essential across various biomechanical applications, including human intervertebral discs (IVDs), to ensure their reliability and use in diagnosing and planning treatments. However, traditional calibration methods are computationally intensive, requiring iterative, derivative-free optimization algorithms that often take days to converge. This study addresses these challenges by introducing a novel, efficient, and effective calibration method demonstrated on a human L4-L5 IVD FE model as a case study using a neural network (NN) surrogate. The NN surrogate predicts simulation outcomes with high accuracy, outperforming other machine learning models, and significantly reduces the computational cost associated with traditional FE simulations. Next, a Projected Gradient Descent (PGD) approach guided by gradients of the NN surrogate is proposed to efficiently calibrate FE models. Our method explicitly enforces feasibility with a projection step, thus maintaining material bounds throughout the optimization process. The proposed method is evaluated against state-of-the-art Genetic Algorithm (GA) and inverse model baselines on synthetic and in vitro experimental datasets. Our approach demonstrates superior performance on synthetic data, achieving a Mean Absolute Error (MAE) of 0.06 compared to the baselines' MAE of 0.18 and 0.54, respectively. On experimental specimens, our method outperforms the baseline in 5 out of 6 cases. While our approach requires initial dataset generation and surrogate training, these steps are performed only once, and the actual calibration takes under three seconds. In contrast, traditional calibration time scales linearly with the number of specimens, taking up to 8 days in the worst-case. Such efficiency paves the way for applying more complex FE models, potentially extending beyond IVDs, and enabling accurate patient-specific simulations.

Authors

  • Matan Atad
    Institute for Neuroradiology, TUM University Hospital, School of Medicine and Health, Technical University of Munich (TUM), Munich, Germany; Chair for AI in Healthcare and Medicine, Technical University of Munich (TUM) and TUM University Hospital, Munich, Germany. Electronic address: matan.atad@tum.de.
  • Gabriel Gruber
    Institute for Neuroradiology, TUM University Hospital, School of Medicine and Health, Technical University of Munich (TUM), Munich, Germany.
  • Marx Ribeiro
    Department of Trauma and Reconstructive Surgery, University Hospital Halle, Martin-Luther-University Halle-Wittenberg, Halle (Saale), Germany; Department of Mechanical Engineering, Federal University of Santa Catarina, Florianópolis, Brazil.
  • Luis Fernando Nicolini
    Department of Mechanical Engineering, Federal University of Santa Maria, Av. Santa Maria, Brazil.
  • Robert Graf
    Institute for Neuroradiology, TUM University Hospital, School of Medicine and Health, Technical University of Munich (TUM), Munich, Germany; Chair for AI in Healthcare and Medicine, Technical University of Munich (TUM) and TUM University Hospital, Munich, Germany.
  • Hendrik Möller
    Institute for Neuroradiology, TUM University Hospital, School of Medicine and Health, Technical University of Munich (TUM), Munich, Germany; Chair for AI in Healthcare and Medicine, Technical University of Munich (TUM) and TUM University Hospital, Munich, Germany.
  • Kati Nispel
    Institute for Neuroradiology, TUM University Hospital, School of Medicine and Health, Technical University of Munich (TUM), Munich, Germany; Associate Professorship of Sport Equipment and Sport Materials, School of Engineering and Design, Technical University of Munich, Munich, Germany.
  • Ivan Ezhov
    Technical University of Munich, Department of Computer Science, Munich, Germany.
  • Daniel Rueckert
    Biomedical Image Analysis (BioMedIA) Group, Department of Computing, Imperial College London, UK. Electronic address: d.rueckert@imperial.ac.uk.
  • Jan S Kirschke
    Neuroradiology, School of Medicine, Technical University of Munich, Munich, Germany.