학술논문

Label Propagation Via Random Walk for Training Robust Thalamus Nuclei Parcellation Model from Noisy Annotations
Document Type
Conference
Source
2023 IEEE 20th International Symposium on Biomedical Imaging (ISBI) Biomedical Imaging (ISBI), 2023 IEEE 20th International Symposium on. :1-5 Apr, 2023
Subject
Bioengineering
Computing and Processing
Photonics and Electrooptics
Signal Processing and Analysis
Training
Three-dimensional displays
Annotations
Manuals
Brain modeling
Thalamus
Noise measurement
Thalamic nuclei
MRI
3D Unet
Label propagation
Random walk
Language
ISSN
1945-8452
Abstract
Data-driven thalamic nuclei parcellation depends on highquality manual annotations. However, the small size and low contrast changes among thalamic nuclei, yield annotations that are often incomplete, noisy, or ambiguously labelled. To train a robust thalamic nuclei parcellation model with noisy annotations, we propose a label propagation algorithm based on random walker to refine the annotations before model training. A two-step model was trained to generate first the whole thalamus and then the nuclei masks. We conducted experiments on a mild traumatic brain injury (mTBI) dataset with noisy thalamic nuclei annotations. Our model outperforms current state-of-the-art thalamic nuclei parcellations by a clear margin. We believe our method can also facilitate the training of other parcellation models with noisy labels.