Connected machine-learning models with joint training for lesion detection

Abstract

Embodiments disclosed herein generally relate to connected machine learning models with joint training for lesion detection. Particularly, aspects of the present disclosure are directed to accessing a three-dimensional magnetic resonance imaging (MRI) image, wherein the three-dimensional MRI image depicts a region of a brain of a subject, wherein the region of the brain includes at least a first type of lesions and a second type of lesions; inputting the three-dimensional MRI image into a machine-learning model comprising a first convolutional neural network and a second convolutional neural network; generating a first segmentation mask for the first type of lesions using the first convolutional neural network that takes as input the three-dimensional MRI image; generating a second segmentation mask for the second type of lesions using the second convolutional neural network that takes as input the three-dimensional MRI image; and outputting the first segmentation mask and the second segmentation mask.