PyTorch-FEA: Autograd-enabled finite element analysis methods with applications for biomechanical analysis of human...

Finite-element analysis (FEA) is widely used as a standard tool for stress and deformation analysis of solid structures, including human tissues and organs. For instance, FEA can be applied at a patient-specific level to assist in medical diagnosis and treatment planning, such as risk assessment of thoracic aortic aneurysm rupture/dissection. These FEA-based biomechanical assessments often involve both forward and inverse mechanics problems. Current commercial FEA software packages (e.g., Abaqus) and inverse methods exhibit performance issues in either accuracy or speed.

In this study, we propose and develop a new library of FEA code and methods, named PyTorch-FEA, by taking advantage of autograd, an automatic differentiation mechanism in PyTorch. We develop a class of PyTorch-FEA functionalities to solve forward and inverse problems with improved loss functions, and we demonstrate the capability of PyTorch-FEA in a series of applications related to human aorta biomechanics. In one of the inverse methods, we combine PyTorch-FEA with deep neural networks (DNNs) to further improve performance.

We applied PyTorch-FEA in four fundamental applications for biomechanical analysis of human aorta. In the forward analysis, PyTorch-FEA achieved a significant reduction in computational time without compromising accuracy compared with Abaqus, a commercial FEA package. Compared to other inverse methods, inverse analysis with PyTorch-FEA achieves better performance in either accuracy or speed, or both if combined with DNNs.

We have presented PyTorch-FEA, a new library of FEA code and methods, representing a new approach to develop FEA methods to forward and inverse problems in solid mechanics. PyTorch-FEA eases the development of new inverse methods and enables a natural integration of FEA and DNNs, which will have numerous potential applications.