Deep Active Learning for Cardiac Image Segmentation
Mengyang Li, Senchun Chai, Tongming Wang, Baihai Zhang · 2022 41st Chinese Control Conference (CCC) · 2022
In recent years, the incidence rate of cardiovascular diseases have been increasing. Cardiac cine magnetic resonance imaging (MRI) is an important method to detect cardiovascular diseases. In the diagnosis of cardiovascular diseases, semantic segmentation of left ventricular cavity, left ventricular myocardium and right ventricular cavity of Cardiac MRI data is a very important step. Now many researchers have proposed different heart segmentation methods. However, these methods need a large number of labeled data sets, and the labeling of these data sets is undoubtedly time-consuming and laborious. This paper presents a deep active learning method based on entropy. In each step of active learning, a batch of unlabeled samples with the largest entropy are selected by using a deep supervision network and handed over to human experts for annotation. The model is trained iteratively until it reaches the desired performance. The results of the experiment show that the active learning method we proposed is obviously better than the random sampling method, and only a small amount of labeled data is needed to achieve the segmentation results achieved by training the model with all data sets.