Exploiting Generative Models for Downstream Classification Tasks on Latent Spaces Using 3D Brain MRI Scans: A Down Syndrome Case Study
摘要
Generative models have emerged as powerful tools in medical imaging, enabling tasks such as segmentation, anomaly detection, and high-quality synthetic data generation. These models typically rely on learning meaningful latent representations, which are particularly valuable given the high-dimensional nature of 3D medical images like brain magnetic resonance imaging (MRI) scans. Despite their potential, latent representations remain underexplored in terms of their structure, information content, and applicability to downstream clinical tasks. Investigating these representations is crucial for advancing the use of generative models in neuroimaging research and clinical decision-making. In this work, we develop a variational autoencoder (VAE) to encode 3D brain MRI scans into a compact latent space for generative and predictive applications. We systematically evaluate the effectiveness of the learned representations through three key analyses: (i) a qualitative assessment of MRI reconstruction quality, (ii) a visualization of the latent space structure using Principal Component Analysis, and (iii) different downstream classification tasks on a proprietary dataset of euploid and Down syndrome individuals brain MRI scans. Our results demonstrate that the VAE successfully captures essential brain features while maintaining high reconstruction fidelity. The latent space exhibits clear clustering patterns, particularly in distinguishing euploid subjects from persons with Down syndrome. Furthermore, classification experiments on this latent space reveal the potential of generative models for encoding biologically relevant brain anatomical features, facilitating research on disorders with associated neuroanatomical alterations.