Contrastive learning with adaptive neighborhoods for brain age prediction on 3D stiffness maps

Träuble, Jakob, Hiscox, Lucy V. ORCID: https://orcid.org/0000-0001-6296-7442, Johnson, Curtis L., Schönlieb, Carola-Bibiane, Kaminski Schierle, Gabriele S. and Aviles-Rivero, Angelica I. 2024. Contrastive learning with adaptive neighborhoods for brain age prediction on 3D stiffness maps. Transactions on Machine Learning Research

Preview

PDF - Published Version Available under License Creative Commons Attribution. Download (1MB) | Preview

Abstract

In the field of neuroimaging, accurate brain age prediction is pivotal for uncovering the complexities of brain aging and pinpointing early indicators of neurodegenerative conditions. Recent advancements in self-supervised learning, particularly in contrastive learning, have demonstrated greater robustness when dealing with complex datasets. However, current approaches often fall short in generalizing across non-uniformly distributed data, prevalent in medical imaging scenarios. To bridge this gap, we introduce a novel contrastive loss that adapts dynamically during the training process, focusing on the localized neighborhoods of samples. Moreover, we expand beyond traditional structural features by incorporating brain stiffness—a mechanical property previously underexplored yet promising due to its sensitivity to age-related changes. This work presents the first application of self-supervised learning to brain mechanical properties, using compiled stiffness maps from various clinical studies to predict brain age. Our approach, featuring dynamic localized loss, consistently outperforms existing state-of-the-art methods, demonstrating superior performance and paving the way for new directions in brain aging research.

Item Type:	Article
Date Type:	Publication
Status:	Published
Schools:	Psychology Cardiff University Brain Research Imaging Centre (CUBRIC)
ISSN:	2835-8856
Date of First Compliant Deposit:	30 October 2024
Date of Acceptance:	28 October 2024
Last Modified:	27 Nov 2024 15:01
URI:	https://orca.cardiff.ac.uk/id/eprint/173524

Actions (repository staff only)

Edit Item