Tiny-objective segmentation for spot signs on multi-phase CT angiography via contrastive learning with dynamic-updated positive-negative memory banks.
Authors
Affiliations (10)
Affiliations (10)
- School of Life Science and Technology Huazhong University of Science and Technology, China; Department of Clinical Neurosciences, Cumming School of Medicine, University of Calgary, Canada. Electronic address: [email protected].
- Department of Clinical Neurosciences, Cumming School of Medicine, University of Calgary, Canada. Electronic address: [email protected].
- Department of Clinical Neurosciences, Cumming School of Medicine, University of Calgary, Canada. Electronic address: [email protected].
- Diagnostic and Interventional Neuroradiology Department, University Hospital of Tours, France. Electronic address: [email protected].
- Department of Internal Medicine, Neurology division, Rady Faculty of Health Sciences, University of Manitoba, Canada. Electronic address: [email protected].
- Department of Radiology, Nuclear Medicine, Maastricht University Medical Center (MUMC+), Netherlands. Electronic address: [email protected].
- Department of Clinical Neurosciences, Cumming School of Medicine, University of Calgary, Canada. Electronic address: [email protected].
- Department of Clinical Neurosciences, Cumming School of Medicine, University of Calgary, Canada. Electronic address: [email protected].
- Department of Clinical Neurosciences, Cumming School of Medicine, University of Calgary, Canada. Electronic address: [email protected].
- School of Life Science and Technology Huazhong University of Science and Technology, China; Advanced Biomedical Imaging Facility, Hubei, China; Faculty of Applied Sciences, Macao Polytechnic University, Macao Special Administrative Region of China. Electronic address: [email protected].
Abstract
Presence of spot sign on CT Angiography (CTA) is associated with hematoma growth in patients with intracerebral hemorrhage. Measuring spot sign volume over time may aid to predict hematoma expansion. Due to the difficulties that imaging characteristics of spot sign are similar with vein and calcification and spot signs are tiny appeared in CTA images to detect, our aim is to develop an automated method to pick up spot signs accurately. We proposed a novel collaborative architecture of network based on a student-teacher model by efficiently exploiting additional negative samples with contrastive learning. In particular, a set of dynamic-updated memory banks is proposed to learn more distinctive features from the extremely imbalanced positive and negative samples. Alongside, a two-steam network with an additional contextual-decoder is designed for learning more contextual information at different scales in a collaborative way. Besides, to better inhibit the false positive detection rate, a region restriction loss function is further designed to confine the spot sign segmentation within the hemorrhage. Quantitative evaluations using dice, volume correlation, sensitivity, specificity, area under the curve show that the proposed method is able to segment and detect spot signs accurately. Our proposed contractive learning framework obtained the best segmentation performance regarding a mean Dice of 0.638 ± 0211, a mean VC of 0.871 and a mean VDP of 0.348 ± 0.237 and detection performance regarding sensitivity of 0.956 with CI(0.895,1.000), specificity of 0.833 with CI(0.766,0.900), and AUC of 0.892 with CI(0.888,0.896), outperforming nnuNet, cascade-nnuNet, nnuNet++, SegRegNet, UNETR and SwinUNETR. This paper proposed a novel segmentation approach that leverages contrastive learning to explore additional negative samples concurrently for the automatic segmentation of spot signs on mCTA images. The experimental results demonstrate the effectiveness of our method and highlight its potential applicability in clinical settings for measuring spot sign volumes.