RESUMO
Uncertainty estimations through approximate Bayesian inference provide interesting insights to deep neural networks' behavior. In unsupervised learning tasks, where expert labels are unavailable, it becomes ever more important to critique the model through uncertainties. This paper presents a proof-of-concept for generalizing the aleatoric and epistemic uncertainties in unsupervised MR-CT synthesis of scoliotic spines. A novel adaptation of the cycle-consistency constraint in CycleGAN is proposed such that the model predicts the aleatoric uncertainty maps in addition to the standard volume-to-volume translation between Magnetic Resonance (MR) and Computed Tomography (CT) data. Ablation experiments were performed to understand uncertainty estimation as an implicit regularizer and a measure of the model's confidence. The aleatoric uncertainty helps in distinguishing between the bone and soft-tissue regions in CT and MR data during translation, while the epistemic uncertainty provides interpretable information to the user for downstream tasks.
RESUMO
Purpose: To develop a deep learning tool for the automatic segmentation of T2-weighted intramedullary lesions in spinal cord injury (SCI). Material and Methods: This retrospective study included a cohort of SCI patients from three sites enrolled between July 2002 and February 2023. A deep learning model, SCIseg, was trained in a three-phase process involving active learning for the automatic segmentation of intramedullary SCI lesions and the spinal cord. The data consisted of T2-weighted MRI acquired using different scanner manufacturers with heterogeneous image resolutions (isotropic/anisotropic), orientations (axial/sagittal), lesion etiologies (traumatic/ischemic/hemorrhagic) and lesions spread across the cervical, thoracic and lumbar spine. The segmentations from the proposed model were visually and quantitatively compared with other open-source baselines. Wilcoxon signed-rank test was used to compare quantitative MRI biomarkers (lesion volume, lesion length, and maximal axial damage ratio) computed from manual lesion masks and those obtained automatically with SCIseg predictions. Results: MRI data from 191 SCI patients (mean age, 48.1 years ± 17.9 [SD]; 142 males) were used for model training and evaluation. SCIseg achieved the best segmentation performance for both the cord and lesions. There was no statistically significant difference between lesion length and maximal axial damage ratio computed from manually annotated lesions and those obtained using SCIseg. Conclusion: Automatic segmentation of intramedullary lesions commonly seen in SCI replaces the tedious manual annotation process and enables the extraction of relevant lesion morphometrics in large cohorts. The proposed model segments lesions across different etiologies, scanner manufacturers, and heterogeneous image resolutions. SCIseg is open-source and accessible through the Spinal Cord Toolbox.