Biomechanical modelling of brain atrophy through deep learning

Mariana da Silva,Kara Garcia,Carole H. Sudre,Cher Bass,M. Jorge Cardoso,Emma Robinson
DOI: https://doi.org/10.48550/arXiv.2012.07596
2020-12-14
Abstract:We present a proof-of-concept, deep learning (DL) based, differentiable biomechanical model of realistic brain deformations. Using prescribed maps of local atrophy and growth as input, the network learns to deform images according to a Neo-Hookean model of tissue deformation. The tool is validated using longitudinal brain atrophy data from the Alzheimer's Disease Neuroimaging Initiative (ADNI) dataset, and we demonstrate that the trained model is capable of rapidly simulating new brain deformations with minimal residuals. This method has the potential to be used in data augmentation or for the exploration of different causal hypotheses reflecting brain growth and atrophy.
Machine Learning,Computer Vision and Pattern Recognition,Image and Video Processing,Tissues and Organs
What problem does this paper attempt to address?
The problem that this paper attempts to solve is to construct a differentiable biomechanical model through deep learning (DL) to simulate the process of brain atrophy and deformation. Specifically, the author hopes to develop a method that can quickly and accurately simulate the deformation of the brain under different atrophy patterns, thereby providing a powerful tool for studying the mechanisms of neurodegenerative diseases such as Alzheimer's disease. ### Core of the problem 1. **Limitations of traditional methods**: - Traditional biomechanical models are usually based on the finite element method (FEM) or the finite difference method (FDM). Although these methods are accurate, they are computationally costly and slow. - These methods need to re - solve the system equations for each new image or condition, resulting in low efficiency. 2. **Advantages of deep learning**: - Deep - learning - driven deformable image registration methods (such as VoxelMorph) can significantly increase the speed during inference while maintaining high performance. - Deep - learning models can be trained with a large amount of data, so as to better capture complex non - linear relationships. ### Specific objectives of the paper - **Construct a deep - learning framework**: This framework can estimate the deformation field of brain tissue according to the given local atrophy and growth maps. - **Verify the effectiveness of the model**: Use the longitudinal brain atrophy data from the Alzheimer's Disease Neuroimaging Initiative (ADNI) dataset for verification, and prove that the model can quickly generate new brain deformations with small residuals. - **Explore potential applications**: This method can be used not only for data augmentation, but also for exploring different causal hypotheses, reflecting the growth and atrophy processes of brain tissue. ### Key technical points - **Neo - Hookean material model**: Model brain tissue as Neo - Hookean material, and describe elastic deformation by minimizing the strain energy density \( W \). \[ W=\sum\frac{\mu}{2}[\text{Tr}(F_K F_K^T)-a - b]+\frac{K}{2}(J - 1)^2 \] where \( J=\det(F_K) \), \(\mu\) is the shear modulus, \( K = 100\mu\) is the volume modulus, and \( a\) and \( b\) are dimension - related parameters. - **Loss function design**: In order to ensure biomechanically meaningful deformation, additional loss terms are added to encourage no displacement in the background area and fix the center point of the brain. \[ L = W+\lambda_1\sum\|u_{\text{background}}\|^2+\lambda_2\|u_{\text{center}}\|^2 \] Through these technical means, the paper proposes a fast, efficient and accurate method for brain atrophy simulation, providing new tools and perspectives for the study of neurodegenerative diseases.