Self-Supervised Learning to Improve Topology-Optimized Axon Segmentation and Centerline Detection
- Resource Type
- Conference
- Authors
- Shamsi, Nina I.; Gjesteby, Lars A.; Chavez, David; Snyder, Michael; Eastwood, Brian S.; Fay, Matthew G.; O'Connor, Nathan J.; Glaser, Jack R.; Gerfen, Charles R.; Brattain, Laura J.
- Source
- 2023 IEEE 20th International Symposium on Biomedical Imaging (ISBI) Biomedical Imaging (ISBI), 2023 IEEE 20th International Symposium on. :1-4 Apr, 2023
- Subject
- Bioengineering
Computing and Processing
Photonics and Electrooptics
Signal Processing and Analysis
Training
Axons
Three-dimensional displays
Sensitivity
Annotations
Brain mapping
Self-supervised learning
Axonal Topology
Axon Tracing
Neuron Segmentation
Self-Supervised Learning
U-Net
- Language
- ISSN
- 1945-8452
Large-scale brain mapping requires preserving the topology of axonal structures to understand how neurons connect throughout the brain and intersect different brain regions. The ability to leverage unannotated data for algorithm development would mitigate laborious annotations by subject matter experts. In this work, we applied self-supervised training to a Residual 3D U-Net with an auxiliary classifier to predict the correct order of sliced and shuffled voxels samples of mouse brain data. Pretrained encoder weights from the auxiliary classifier were subsequently used to train a Residual 3D U-Net for axon segmentation and centerline detection with a topology-preserving loss function, soft centerline-Dice. We report improved feature space representation of axonal structures as well as improved evaluation metrics over prior methods. We observed that both the α and k hyperparameters of the topologically-aware loss function show sensitivity to the target task of axon segmentation and centerline detection.