A Registration- and Uncertainty-based Framework for White Matter Tract Segmentation With Only One Annotated Subject
Not on the map. Why?
Abstract (arXiv)
White matter (WM) tract segmentation based on diffusion magnetic resonance imaging (dMRI) plays an important role in the analysis of human health and brain diseases. However, the annotation of WM tracts is time-consuming and needs experienced neuroanatomists. In this study, to explore tract segmentation in the challenging setting of minimal annotations, we propose a novel framework utilizing only one annotated subject (subject-level one-shot) for tract segmentation. Our method is constructed by proposed registration-based peak augmentation (RPA) and uncertainty-based refining (URe) modules. RPA module synthesizes pseudo subjects and their corresponding labels to improve the tract segmentation performance. The proposed URe module alleviates the negative influence of the low-confidence voxels on pseudo subjects. Experimental results show that our method outperforms other state-of-the-art methods by a large margin, and our proposed modules are effective. Overall, our method achieves accurate whole-brain tract segmentation with only one annotated subject. Our code is available at https://github.com/HaoXu0507/ISBI2023-One-Shot-WM-Tract-Segmentation.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Builds on 8 of its 19 references that are in the map. First posted on arXiv 25 Mar 2023.
Builds on
8 of 19 references.
References · 8
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2018VoxelMorph: A Learning Framework for Deformable Medical Image Registrationcited by 657
- 2019Data augmentation using learned transformations for one-shot medical image segmentationcited by 138
- 2018TractSeg - Fast and accurate white matter tract segmentationcited by 50
- 2022Superficial White Matter Analysis: An Efficient Point-cloud-based Deep Learning Framework with Supervised Contrastive Learning for Consistent Tractography Parcellation across Populations and dMRI Acquisitionscited by 19
- 2022SupWMA: Consistent and Efficient Tractography Parcellation of Superficial White Matter with Deep Learningcited by 13
- 2020Neuro4Neuro: A neural network approach for neural tract segmentation using large-scale population-based diffusion imagingcited by 6
- 2023One-Shot Segmentation of Novel White Matter Tracts via Extensive Data Augmentationcited by 5
Built on by
Citing papers per year: 2023: 1; 2024: 2; 2025: 1.
4 papers.
- 2024Diffusion MRI with Machine Learningcited by 11
- 2023TractCloud: Registration-free tractography parcellation with a novel local-global streamline point cloud representationcited by 10
- 2024ORID: Organ-Regional Information Driven Framework for Radiology Report Generationcited by 3
- 2025MultiCo3D: Multi-Label Voxel Contrast for One-Shot Incremental Segmentation of 3D Neuroimagescited by 2
- 2025MultiCo3D: Multi-Label Voxel Contrast for One-Shot Incremental Segmentation of 3D Neuroimages9 Mar 2025
- 2024ORID: Organ-Regional Information Driven Framework for Radiology Report Generation20 Nov 2024
- 2024Diffusion MRI with Machine Learning1 Jan 2024
- 2023TractCloud: Registration-free tractography parcellation with a novel local-global streamline point cloud representation18 Jul 2023