English

Multi-Task Adversarial Variational Autoencoder for Estimating Biological Brain Age with Multimodal Neuroimaging

Computer Vision and Pattern Recognition 2024-11-18 v1 Artificial Intelligence

Abstract

Despite advances in deep learning for estimating brain age from structural MRI data, incorporating functional MRI data is challenging due to its complex structure and the noisy nature of functional connectivity measurements. To address this, we present the Multitask Adversarial Variational Autoencoder, a custom deep learning framework designed to improve brain age predictions through multimodal MRI data integration. This model separates latent variables into generic and unique codes, isolating shared and modality-specific features. By integrating multitask learning with sex classification as an additional task, the model captures sex-specific aging patterns. Evaluated on the OpenBHB dataset, a large multisite brain MRI collection, the model achieves a mean absolute error of 2.77 years, outperforming traditional methods. This success positions M-AVAE as a powerful tool for metaverse-based healthcare applications in brain age estimation.

Keywords

Cite

@article{arxiv.2411.10100,
  title  = {Multi-Task Adversarial Variational Autoencoder for Estimating Biological Brain Age with Multimodal Neuroimaging},
  author = {Muhammad Usman and Azka Rehman and Abdullah Shahid and Abd Ur Rehman and Sung-Min Gho and Aleum Lee and Tariq M. Khan and Imran Razzak},
  journal= {arXiv preprint arXiv:2411.10100},
  year   = {2024}
}
R2 v1 2026-06-28T20:01:04.690Z