AxonSynth: Domain-Randomized Synthetic Data for Zero-Shot 3D Axon Segmentation in Light-Sheet Microscopy
Organizations: Department of Computer Science, University of Massachusetts Boston, Boston, MA, USA · Athinoula A. Martinos Center for Biomedical Imaging, Massachusetts General Hospital and Harvard Medical School, Charlestown, MA, USA · Universidad San Sebastián, Chile · Université Claude Bernard Lyon 1, Université de Lyon, Lyon, France · Department of Imaging Sciences, St. Jude Children’s Research Hospital, Memphis, TN, USA · Department of Biomedical Engineering, Columbia University in the City of New York, New York, NY, USA · Department of Experimental Psychology, University College London, London, United Kingdom
Abstract
Accurate segmentation of axons in 3D microscopy data is important for analyzing white-matter organization, but dense ground truth labels are expensive to obtain. Existing supervised axon segmentation methods rely on target-domain annotations and can be brittle when tissue type, species, modality, or acquisition conditions change. We present AxonSynth, a domain-randomized synthetic-data framework for training 3D axon segmentation models without manually annotated real training volumes. AxonSynth generates dense synthetic axon labels with orientation priors that reflect realistic fiber configurations and renders them with randomized density, contrast, bias fields, blur, and noise. A three-class 3D U-Net is trained to predict background, axon sheath and intra-axonal space. We evaluate zero-shot transfer on 10 held-out light-sheet microscopy (LSM) patches from macaque and human brain samples labeled with one of three axonal markers, comparing against calibrated thresholding and Frangi filtering using overlap, corrected detection, false-positive, and topology metrics. On macaque samples, AxonSynth achieved the best corrected Dice and corrected precision (0.826 and 0.851), compared with 0.765 and 0.754 for thresholding and 0.685 and 0.762 for Frangi. On human samples, corrected Dice was comparable to thresholding (0.857 vs. 0.868), while component-count error decreased from 22,504 to 3,377. Across all held-out patches, AxonSynth reduced component-count error in 10/10 patches and Euler-characteristic error in 8/10. These results show that synthetic-label domain randomization can reduce dependence on manual axon annotation while supporting synthetic-to-real 3D segmentation.
Figures & tables
| Component | Setting | Component | Setting |
|---|---|---|---|
| Label count | 500 volumes; 400 train, 100 validation | Patch and voxel | voxels; 0.8 m |
| Axon-tree count | trees/patch ( – ) | Orientation prior | Fixed -axis Bingham; one component |
| Orientation variance | Radius | mm; clip mm | |
| Radius change | Branch radius ratio | ||
| Branch count | Tortuosity | ||
| Levels per tree | Max. 2 (zero or one branch point) | Rasterization | B-spline curves; cosine partial volume |
| Method | Dice | Corr. Dice | cPrec. | cRec. | cFPR | ||
|---|---|---|---|---|---|---|---|
| Human | |||||||
| Thresholding | 0.576 | 0.868 | 0.910 | 0.837 | 0.0088 | 22,504 | 19,499 |
| AxonSynth | 0.465 | 0.857 | 0.878 | 0.839 | 0.0109 | 3,377 | 3,692 |
| Frangi | 0.512 | 0.769 | 0.970 | 0.639 | 0.0017 | 1,458 | 1,006 |
| Macaque | |||||||
| Thresholding | 0.550 | 0.765 | 0.754 | 0.852 | 0.0409 | 14,081 | 9,914 |