MACS: Multi-Domain Adaptation Enables Accurate Connectomics Segmentation
Abir, A. R.; Saha, A.; Bayzid, M. S.
Show abstract
Connectomics aims to map the brains neural wiring by segmenting cellular structures from high-resolution electron microscopy (EM) images. Manual labeling and proofreading remain a major bottleneck for accurate extraction of microstructures. While computational models have advanced automated segmentation, they typically require training from scratch on each dataset, demanding substantial annotated data. Domain adaptation methods address this by transferring knowledge from a labeled source to a less-annotated target. However, existing approaches are limited to adaptation from a single source domain. This overlooks the potential benefits of integrating information from multiple diverse domains, motivating the development of multidomain adaptation. To address this, we propose MACS, the first known multi-domain adaptation framework that combines knowledge from multiple heterogeneous source domains to learn segmentation in the target domain, and employs active learning to efficiently select the most informative target samples for annotation. MACS uses information-theoretic weighting to combine source domains, and introduces a novel and efficient Bayesian Laplace approximation for uncertainty estimation. Our extensive experiments across nine connectomics datasets demonstrate that MACS consistently and substantially outperforms state-of-the-art models, even under limited annotation budgets, with a mean improvement of 5.89% at the lowest annotation budget and 27.72% at the highest annotation budget. In-depth analyses further reveal that MACS offers mechanistic interpretability by quantifying and explicitly upweighting the most transferable source domains for each target. The preprocessed datasets and the source code of MACS are publicly available at http://github.com/abrarrahmanabir/MACS.
Matching journals
The top 7 journals account for 50% of the predicted probability mass.
Similar papers in this journal
Similar papers in this journal
- RETINA: Reconstruction-based Pre-Trained Enhanced TransUNet for Electron Microscopy Segmentation on the CEM500K Dataset 95%
- Localized semi-nonnegative matrix factorization (LocaNMF) of widefield calcium imaging data 95%
- A Deep Learning approach for time-consistent cell cycle phase prediction from microscopy data 95%
Similar papers in this journal
- A Framework for Falsifiable Explanations of Machine Learning Models with an Application in Computational Pathology 94%
- Clinical Validation of Saliency Maps for Understanding Deep Neural Networks in Ophthalmology 94%
- Track-To-Learn: A general framework for tractography with deep reinforcement learning 94%
Similar papers in this journal
Similar papers in this journal
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.