Semi-Supervised Diffusion Model for Brain Age Prediction
Authors:
Ayodeji Ijishakin,
Sophie Martin,
Florence Townend,
Federica Agosta,
Edoardo Gioele Spinelli,
Silvia Basaia,
Paride Schito,
Yuri Falzone,
Massimo Filippi,
James Cole,
Andrea Malaspina
Abstract:
Brain age prediction models have succeeded in predicting clinical outcomes in neurodegenerative diseases, but can struggle with tasks involving faster progressing diseases and low quality data. To enhance their performance, we employ a semi-supervised diffusion model, obtaining a 0.83(p<0.01) correlation between chronological and predicted age on low quality T1w MR images. This was competitive wit…
▽ More
Brain age prediction models have succeeded in predicting clinical outcomes in neurodegenerative diseases, but can struggle with tasks involving faster progressing diseases and low quality data. To enhance their performance, we employ a semi-supervised diffusion model, obtaining a 0.83(p<0.01) correlation between chronological and predicted age on low quality T1w MR images. This was competitive with state-of-the-art non-generative methods. Furthermore, the predictions produced by our model were significantly associated with survival length (r=0.24, p<0.05) in Amyotrophic Lateral Sclerosis. Thus, our approach demonstrates the value of diffusion-based architectures for the task of brain age prediction.
△ Less
Submitted 14 February, 2024;
originally announced February 2024.
Interpretable Alzheimer's Disease Classification Via a Contrastive Diffusion Autoencoder
Authors:
Ayodeji Ijishakin,
Ahmed Abdulaal,
Adamos Hadjivasiliou,
Sophie Martin,
James Cole
Abstract:
In visual object classification, humans often justify their choices by comparing objects to prototypical examples within that class. We may therefore increase the interpretability of deep learning models by imbuing them with a similar style of reasoning. In this work, we apply this principle by classifying Alzheimer's Disease based on the similarity of images to training examples within the latent…
▽ More
In visual object classification, humans often justify their choices by comparing objects to prototypical examples within that class. We may therefore increase the interpretability of deep learning models by imbuing them with a similar style of reasoning. In this work, we apply this principle by classifying Alzheimer's Disease based on the similarity of images to training examples within the latent space. We use a contrastive loss combined with a diffusion autoencoder backbone, to produce a semantically meaningful latent space, such that neighbouring latents have similar image-level features. We achieve a classification accuracy comparable to black box approaches on a dataset of 2D MRI images, whilst producing human interpretable model explanations. Therefore, this work stands as a contribution to the pertinent development of accurate and interpretable deep learning within medical imaging.
△ Less
Submitted 25 October, 2023; v1 submitted 5 June, 2023;
originally announced June 2023.