Fine-tuning TrailMap: The utility of transfer learning to improve the performance of deep learning in axon segmentation of light-sheet microscopy images.

TitleFine-tuning TrailMap: The utility of transfer learning to improve the performance of deep learning in axon segmentation of light-sheet microscopy images.
Publication TypeJournal Article
Year of Publication2024
AuthorsOostrom M, Muniak MA, West RMEichler, Akers S, Pande P, Obiri M, Wang W, Bowyer K, Wu Z, Bramer LM, Mao T, Webb-Robertson BJo M
JournalPLoS One
Volume19
Issue3
Paginatione0293856
Date Published2024
ISSN1932-6203
KeywordsAnimals, Axons, Brain, Deep Learning, Machine Learning, Mice, Microscopy
Abstract

Light-sheet microscopy has made possible the 3D imaging of both fixed and live biological tissue, with samples as large as the entire mouse brain. However, segmentation and quantification of that data remains a time-consuming manual undertaking. Machine learning methods promise the possibility of automating this process. This study seeks to advance the performance of prior models through optimizing transfer learning. We fine-tuned the existing TrailMap model using expert-labeled data from noradrenergic axonal structures in the mouse brain. By changing the cross-entropy weights and using augmentation, we demonstrate a generally improved adjusted F1-score over using the originally trained TrailMap model within our test datasets.

DOI10.1371/journal.pone.0293856
Alternate JournalPLoS One
PubMed ID38551935
PubMed Central IDPMC10980229
Grant ListR01 NS081071 / NS / NINDS NIH HHS / United States
R01 NS104944 / NS / NINDS NIH HHS / United States
RF1 MH120119 / MH / NIMH NIH HHS / United States
RF1 MH128969 / MH / NIMH NIH HHS / United States