Dual Graph Attention based Disentanglement Multiple Instance Learning for Brain Age Estimation

• URL: http://arxiv.org/abs/2403.01246v1
• Date: Sat, 2 Mar 2024 16:13:06 GMT
• Title: Dual Graph Attention based Disentanglement Multiple Instance Learning for Brain Age Estimation
• Authors: Fanzhe Yan, Gang Yang, Yu Li, Aiping Liu, Xun Chen
• Abstract summary: We propose a Dual Graph Attention based Disentanglement Multi-instance Learning (DGA-DMIL) framework for improving brain age estimation. A dual graph attention aggregator is then proposed to learn the backbone features by exploiting the intra- and inter-instance relationships. Our proposed model demonstrates exceptional accuracy in estimating brain age, achieving a remarkable mean absolute error of 2.12 years in the UK Biobank.
• Score: 26.33669583527592
• License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
• Abstract: Deep learning techniques have demonstrated great potential for accurately estimating brain age by analyzing Magnetic Resonance Imaging (MRI) data from healthy individuals. However, current methods for brain age estimation often directly utilize whole input images, overlooking two important considerations: 1) the heterogeneous nature of brain aging, where different brain regions may degenerate at different rates, and 2) the existence of age-independent redundancies in brain structure. To overcome these limitations, we propose a Dual Graph Attention based Disentanglement Multi-instance Learning (DGA-DMIL) framework for improving brain age estimation. Specifically, the 3D MRI data, treated as a bag of instances, is fed into a 2D convolutional neural network backbone, to capture the unique aging patterns in MRI. A dual graph attention aggregator is then proposed to learn the backbone features by exploiting the intra- and inter-instance relationships. Furthermore, a disentanglement branch is introduced to separate age-related features from age-independent structural representations to ameliorate the interference of redundant information on age prediction. To verify the effectiveness of the proposed framework, we evaluate it on two datasets, UK Biobank and ADNI, containing a total of 35,388 healthy individuals. Our proposed model demonstrates exceptional accuracy in estimating brain age, achieving a remarkable mean absolute error of 2.12 years in the UK Biobank. The results establish our approach as state-of-the-art compared to other competing brain age estimation models. In addition, the instance contribution scores identify the varied importance of brain areas for aging prediction, which provides deeper insights into the understanding of brain aging.

Related papers

• Knowledge-Guided Prompt Learning for Lifespan Brain MR Image Segmentation [53.70131202548981]
  We present a two-step segmentation framework employing Knowledge-Guided Prompt Learning (KGPL) for brain MRI. Specifically, we first pre-train segmentation models on large-scale datasets with sub-optimal labels. The introduction of knowledge-wise prompts captures semantic relationships between anatomical variability and biological processes.
  arXiv  Detail & Related papers  (2024-07-31T04:32:43Z)
• Ordinal Classification with Distance Regularization for Robust Brain Age Prediction [25.555190119033615]
  Age is one of the major known risk factors for Alzheimer's Disease (AD) Brain age, a measure derived from brain imaging reflecting structural changes due to aging, may have the potential to identify AD onset, assess disease risk, and plan targeted interventions. Deep learning-based regression techniques to predict brain age from magnetic resonance imaging (MRI) scans have shown great accuracy recently. These methods are subject to an inherent regression to the mean effect, which causes a systematic bias resulting in an overestimation of brain age in young subjects and underestimation in old subjects.
  arXiv  Detail & Related papers  (2023-10-25T20:39:07Z)
• Robust Brain Age Estimation via Regression Models and MRI-derived Features [2.028990630951476]
  We present a novel brain age estimation framework using the Open Big Healthy Brain (OpenBHB) dataset. Our approach integrates three different MRI-derived region-wise features and different regression models, resulting in a highly accurate brain age estimation.
  arXiv  Detail & Related papers  (2023-06-08T19:07:22Z)
• Explainable Brain Age Prediction using coVariance Neural Networks [94.81523881951397]
  We propose an explanation-driven and anatomically interpretable framework for brain age prediction using cortical thickness features. Specifically, our brain age prediction framework extends beyond the coarse metric of brain age gap in Alzheimer's disease (AD) We make two important observations: VNNs can assign anatomical interpretability to elevated brain age gap in AD by identifying contributing brain regions.
  arXiv  Detail & Related papers  (2023-05-27T22:28:25Z)
• Brain Structure Ages -- A new biomarker for multi-disease classification [0.0]
  We propose to extend the notion of global brain age by estimating brain structure ages using structural magnetic resonance imaging. Brain structure ages can be used to compute the deviation from the normal aging process of each brain structure.
  arXiv  Detail & Related papers  (2023-04-13T14:56:51Z)
• Unsupervised Anomaly Detection in 3D Brain MRI using Deep Learning with Multi-Task Brain Age Prediction [53.122045119395594]
  Unsupervised anomaly detection (UAD) in brain MRI with deep learning has shown promising results. We propose deep learning for UAD in 3D brain MRI considering additional age information. Based on our analysis, we propose a novel deep learning approach for UAD with multi-task age prediction.
  arXiv  Detail & Related papers  (2022-01-31T09:39:52Z)
• Voxel-level Importance Maps for Interpretable Brain Age Estimation [70.5330922395729]
  We focus on the task of brain age regression from 3D brain Magnetic Resonance (MR) images using a Convolutional Neural Network, termed prediction model. We implement a noise model which aims to add as much noise as possible to the input without harming the performance of the prediction model. We test our method on 13,750 3D brain MR images from the UK Biobank, and our findings are consistent with the existing neuropathology literature.
  arXiv  Detail & Related papers  (2021-08-11T18:08:09Z)
• Brain Age Estimation From MRI Using Cascade Networks with Ranking Loss [75.03117866578913]
  A novel 3D convolutional network, called two-stage-age-network (TSAN), is proposed to estimate brain age from T1-weighted MRI data. Experiments with $6586$ MRIs showed that TSAN could provide accurate brain age estimation.
  arXiv  Detail & Related papers  (2021-06-06T07:11:25Z)
• Patch-based Brain Age Estimation from MR Images [64.66978138243083]
  Brain age estimation from Magnetic Resonance Images (MRI) derives the difference between a subject's biological brain age and their chronological age. Early detection of neurodegeneration manifesting as a higher brain age can potentially facilitate better medical care and planning for affected individuals. We develop a new deep learning approach that uses 3D patches of the brain as well as convolutional neural networks (CNNs) to develop a localised brain age estimator.
  arXiv  Detail & Related papers  (2020-08-29T11:50:37Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.