BDAL : Balanced Distribution Active Learning for MRI Cardiac Multistructures Segmentation

Shu, Xiu and Yang, Yunyun and Liu, Jun and Chang, Xiaojun and Wu, Boying (2024) BDAL : Balanced Distribution Active Learning for MRI Cardiac Multistructures Segmentation. IEEE Transactions on Industrial Informatics, 20 (4). pp. 6099-6108. ISSN 1551-3203

Abstract

Various kinds of heart diseases pose a serious threat to human health. To effectively treat and prevent these diseases, accurate segmentation of the entire heart structure is crucial for medical research and application. At present, the solution to this problem still needs to rely on a lot of manpower. Not only is this time-consuming, but accuracy is sometimes difficult to guarantee. In the deep learning methods for medical image segmentation, large labeled images are difficult to obtain. Typically, the large databases have several thousand images, of which only a few hundred have been annotated, and the number of individual patients is even smaller. In this article, we focus on a small part of the dataset to minimize the cost of manual labeling and maximize the accurate segmentation results. The small part of the dataset contains more representative and informative images, avoiding doctors to repeatedly label images with similar information. We proposed a balanced distribution active learning (BDAL) framework for MRI cardiac multistructures segmentation based on reinforcement learning. The deep Q-network framework can learn an effective policy to select some informative and representative images to be labeled from a large number of the unlabeled dataset. We consider the shape features of images and the balance of different class distributions to build new state and action representation, which can help the agent to identify informative and representative images for annotation. Our BDAL method provides an agent to improve the ability of AL to select images to improve the accuracy of segmentation. Moreover, experiments and results show that our BDAL method significantly outperforms all baselines and other AL-based methods under the same amount of annotation budget on MRI cardiac multistructures segmentation in datasets ACDC and M&Ms .