Quality-driven deep active learning method for 3D brain MRI segmentation
| Autor: | Chunna Tian, Jie Li, Xinbo Gao, Zhusi Zhong, Zhicheng Jiao, Zhenxi Zhang |
|---|---|
| Rok vydání: | 2021 |
| Předmět: |
0209 industrial biotechnology
Active learning (machine learning) business.industry Computer science Cognitive Neuroscience media_common.quotation_subject Deep learning ComputingMethodologies_IMAGEPROCESSINGANDCOMPUTERVISION Pattern recognition 02 engineering and technology Computer Science Applications Task (project management) Annotation 020901 industrial engineering & automation Artificial Intelligence 0202 electrical engineering electronic engineering information engineering Brain mri Brain segmentation 020201 artificial intelligence & image processing Segmentation Quality (business) Artificial intelligence business media_common |
| Zdroj: | Neurocomputing. 446:106-117 |
| ISSN: | 0925-2312 |
| DOI: | 10.1016/j.neucom.2021.03.050 |
| Popis: | Automatic segmentation of the brain Magnetic Resonance Imaging (MRI) plays a crucial role in many brain MRI processing algorithms, which is effective for the prevention, detection, monitoring, and treatment planning of brain disease. Currently, deep learning algorithms have shown outstanding performance in brain segmentation. Most algorithms train models with fully annotated brain MRI datasets. However, full annotation of 3D brain MRI is laborious and time-consuming. Using sparsely annotated datasets to train models may be one appropriate solution to reduce annotation cost. However, the approach of 3D dense segmentation with sparse annotation has not been fully studied. In this paper, we develop a segmentation framework combined with the quality-driven active learning (QDAL) module for suggestive annotation. In the proposed Active Learning module, attention mechanism and deep supervision mode are used to improve the segmentation accuracy and feedback segmentation quality information. Meanwhile, we observe a high correlation coefficient between the proposed two surrogate metrics and the real segmentation accuracy of per slice in one scan. We validate our framework on two public brain MRI datasets for brain region extraction and brain tissue segmentation. The comparative experiments demonstrate that the QDAL method outperforms the other four popular sampling strategies. The segmentation network with the guidance of the QDAL method only needs 15–20% annotated slices in brain extraction task, and 30–40% annotated slices in tissue segmentation task to achieve competitive results compared with training with full supervision. |
| Databáze: | OpenAIRE |
| Externí odkaz: |
|
Full Text from ScienceDirect