Description:
This project collects 6 different subprojects from the Division of Biomedical Imaging at KTH that require the use of GPUs for AI. This is a brief description of them:
1. Deep learning-based breast image analysis (PhD student Zhikai Yang). Goal: A) use diffusion and ViTs for the detection, segmentation, and classification of breast cancer images. B) Generation of mammograms from tomosynthesis data with AI. C) Perform image reconstruction of digital tomosynthesis with primal-dual neural networks.
2. Transformers-based Indolent Lymphoma Classification (PhD student Simone Bendazzoli). Goal: use ViT models to predict, from an initial estimate, the 3D bounding boxes of the Lymphoma lesions in PET-CT data.
3. Synthetic generation of longitudinal MRI using aging templates (PhD student Jingru Fu). Goal: use deep learning-based diffeomorphic registration to generate age-specific templates of subjects that can be used to predict future MRI scans of subjects.
4. Reinforcement learning-based tractography (PhD student Fabian Sinzinger). Goal: combine equivariant graph neural networks and reinforcement learning to create streamlines of neural fibers from diffusion MRI data.
5. Machine learning methods for brain magnetic resonance elastography (MRE) (PhD student Hampus Möller). Goal: combine physics-informed neural networks, neural operators and system identification to estimate the mechanical properties of the tissues from raw MRE data.
6. Generative models for diffusion MRI (PhD student Sanna Persson). Goal: use generative models to estimate diffusion MRI data with a given b-value (a scanning parameter) from images acquired with a different b-value. These images will be used to predict the mechanical properties of brain tissue.
All subprojects use anonymized data without sensitive information or synthetic data.
Impact:
All sub-projects aim to improve the quality of the images, increase accuracy, or add extra information that can potentially be used by clinicians in the future, which will potentially benefit the healthcare system and society in general.