TY - JOUR
T1 - Uncertainty-Guided Semi-Supervised (UGSS) mean teacher framework for brain hemorrhage segmentation and volume quantification
AU - Emon, Solayman Hossain
AU - Tseng, Tzu Liang (Bill)
AU - Pokojovy, Michael
AU - Moen, Scott
AU - McCaffrey, Peter
AU - Walser, Eric
AU - Vo, Alexander
AU - Rahman, Md Fashiar
N1 - Publisher Copyright:
© 2024
PY - 2025/4
Y1 - 2025/4
N2 - Traumatic brain injury (TBI) is considered a critical neurological emergency with substantial morbidity and mortality rates across the world. Among significant neuropathological consequences of brain injuries, intracranial hemorrhage (ICH) stands out as a particularly urgent condition necessitating prompt diagnosis to avert life-threatening complications. However, the traditional manual approach to detecting and segmenting brain hemorrhages in CT scans is time-consuming and labor-intensive. This study proposes a fully automated uncertainty-guided framework for intracranial hemorrhage segmentation in brain CT scans. The framework is trained on a semi-supervised scheme that leverages both labeled and unlabeled data. Notably, when trained on 80% of labeled data, the semi-supervised framework yields an average Dice coefficient of 0.613 ± 0.01 and a Jaccard index of 0.441 ± 0.02. These metrics significantly exceed their supervised counterparts, which demonstrates the efficacy of the proposed methodology. Moreover, the proposed approach exhibits an overall accuracy of 89.03% in brain hemorrhage classification with a Cohen's Kappa value of 0.835, which indicates substantial agreement between the model's predictions and the ground truth labels. In addition to its capabilities in intracranial hemorrhage detection and localization, the proposed framework offers a robust estimation of hemorrhage volume and provides a comprehensive 3D volumetric view. The accuracy and reliability of the volume quantification approach are justified through a comprehensive qualitative and quantitative assessment, utilizing visualization techniques and a goodness-of-fit test (R2=0.837). In both instances, the method shows a notable alignment between the predicted hemorrhage volume and the actual hemorrhage volume. Thus, the proposed schemes of uncertainty-guided semi-supervised (UGSS) hemorrhage segmentation and volume quantification enhance model's applicability in clinical practice and research.
AB - Traumatic brain injury (TBI) is considered a critical neurological emergency with substantial morbidity and mortality rates across the world. Among significant neuropathological consequences of brain injuries, intracranial hemorrhage (ICH) stands out as a particularly urgent condition necessitating prompt diagnosis to avert life-threatening complications. However, the traditional manual approach to detecting and segmenting brain hemorrhages in CT scans is time-consuming and labor-intensive. This study proposes a fully automated uncertainty-guided framework for intracranial hemorrhage segmentation in brain CT scans. The framework is trained on a semi-supervised scheme that leverages both labeled and unlabeled data. Notably, when trained on 80% of labeled data, the semi-supervised framework yields an average Dice coefficient of 0.613 ± 0.01 and a Jaccard index of 0.441 ± 0.02. These metrics significantly exceed their supervised counterparts, which demonstrates the efficacy of the proposed methodology. Moreover, the proposed approach exhibits an overall accuracy of 89.03% in brain hemorrhage classification with a Cohen's Kappa value of 0.835, which indicates substantial agreement between the model's predictions and the ground truth labels. In addition to its capabilities in intracranial hemorrhage detection and localization, the proposed framework offers a robust estimation of hemorrhage volume and provides a comprehensive 3D volumetric view. The accuracy and reliability of the volume quantification approach are justified through a comprehensive qualitative and quantitative assessment, utilizing visualization techniques and a goodness-of-fit test (R2=0.837). In both instances, the method shows a notable alignment between the predicted hemorrhage volume and the actual hemorrhage volume. Thus, the proposed schemes of uncertainty-guided semi-supervised (UGSS) hemorrhage segmentation and volume quantification enhance model's applicability in clinical practice and research.
KW - Brain hemorrhage
KW - Deep learning
KW - Semi-supervised learning
KW - Uncertainty quantification
UR - http://www.scopus.com/inward/record.url?scp=85213006655&partnerID=8YFLogxK
UR - http://www.scopus.com/inward/citedby.url?scp=85213006655&partnerID=8YFLogxK
U2 - 10.1016/j.bspc.2024.107386
DO - 10.1016/j.bspc.2024.107386
M3 - Article
AN - SCOPUS:85213006655
SN - 1746-8094
VL - 102
JO - Biomedical Signal Processing and Control
JF - Biomedical Signal Processing and Control
M1 - 107386
ER -