Summary: Linking brain activity to behavior requires identifying and tracking individual neurons in real time — a task made challenging when the subject is a moving, deforming animal such as a wriggling roundworm or a flexible jellyfish. Researchers at MIT developed three AI-enhanced tools that overcome this alignment-and-annotation bottleneck, automating what used to take months of manual work.
The three tools — BrainAlignNet, AutoCellLabeler, and CellDiscoveryNet — automatically locate, follow, and classify fluorescently labeled neurons across long imaging sequences with extremely high accuracy (up to 99.6% for tracking). By replacing labor-intensive manual annotation with near-instant automated analysis, these methods provide a scalable approach for decoding nervous-system activity in living, behaving animals.
Key Facts
- The “wiggle” problem: In freely moving or deforming animals, every part of the body can shift independently, so neurons that appear at one position in a frame may be elsewhere a moment later. That non-rigid motion makes precise tracking and identification difficult.
- BrainAlignNet: A registration network that tracks cells across long video series with single-pixel accuracy and 99.6% success in linking neurons over time. It runs roughly 600 times faster than the lab’s previous manual or semi-manual methods.
- AutoCellLabeler: A supervised classifier that assigns cell-type identities from multi-spectral fluorescence labels. With full color barcoding (NeuroPAL) it reaches about 98% accuracy and remains robust when fewer color channels are available.
- CellDiscoveryNet: An unsupervised approach that clusters and identifies conserved cell types across different animals without any human labels or training, achieving performance comparable to expert human annotators.
- Broad applicability: Although tuned for the model organism Caenorhabditis elegans, the same methods successfully registered neurons in the morphologically different jellyfish Clytia hemisphaerica, showing the tools’ potential for varied biological imaging problems.
Source: Picower Institute at MIT
Why this matters
Understanding how patterns of neuronal activity drive behavior depends on reliably mapping each signal to a specific cell over time. Many neuroscience labs use transparent model animals so they can image nearly all neurons at once as animals behave. But visibility alone is not sufficient: researchers also need automated ways to align images, follow cells as bodies deform, and label cell identities across large datasets. These three deep-learning tools were developed to meet that need directly.

The three components of the pipeline address different parts of the alignment-and-annotation challenge. BrainAlignNet performs non-rigid registration to determine where the same cell has moved between frames. AutoCellLabeler focuses on assigning a biological identity to each detected cell when provided with human-labeled training examples. CellDiscoveryNet discovers and groups consistent cell types across many animals without supervision, enabling cross-subject comparisons and large-scale cell-type discovery.
These networks are built on existing neural-network architectures that the authors adapted and optimized for image registration and multi-spectral cell classification. Importantly, the models learn the relevant image features automatically — color, position, or shape — rather than relying on hand-engineered rules. That flexibility helps the systems generalize to different organisms and imaging setups.
From months to minutes
Before these AI tools, annotating a single worm video could require a trained researcher several hours of careful manual labeling; scaling up experiments created an enormous bottleneck. In one example from the lab, outsourcing manual annotation was estimated to cost six figures. The automated tools dramatically reduce those time and cost barriers while improving consistency and accuracy across large datasets.
These advances also translate to other model organisms. In collaboration with a colleague’s lab, BrainAlignNet was adapted to register neurons in the deforming, highly motile jellyfish Clytia hemisphaerica. That lab reported the tool enabled extraction of neural activity from behavioral videos that would have been otherwise extremely difficult to analyze.
The research team notes there is still work to do: for example, expanding labeled markers in the jellyfish to cover all cell types and developing imaging systems capable of recording freely swimming animals at the required resolution. Nonetheless, these methods provide an immediate path forward for many labs that generate large imaging datasets and need scalable, accurate cell registration and annotation.
In addition to the senior and lead authors, the study lists multiple contributors across imaging, model development, and experimental analysis. Funding came from agencies and foundations including the National Institutes of Health, the National Science Foundation, the McKnight Foundation, the Alfred P. Sloan Foundation, the Howard Hughes Medical Institute, and the Freedom Together Foundation.
Key Questions Answered:
A: Tracking a single neuron in a deforming, moving animal is extremely difficult — like trying to follow a grape in a bowl of Jell-O while it’s being shaken. Manual labeling of a single video can take hours for an expert. These AI models do the same job in seconds to minutes with far greater scalability.
A: The tools were developed for transparent invertebrate systems, but the underlying deep-learning approaches are general and can be adapted to other large series of microscopy images, including studies of human tissues or other organisms, provided appropriate imaging and training data are available.
A: Yes. CellDiscoveryNet demonstrates unsupervised learning: it compares multi-spectral imaging across many animals and automatically clusters recurring cell types without human labels, matching the performance of trained human annotators.
Editorial Notes:
- This article was edited by a Neuroscience News editor.
- The peer-reviewed journal paper was examined in full by editorial staff.
- Additional explanatory context was provided by the reporting team.
About this AI and neuroscience research news
Author: David Orenstein
Source: Picower Institute at MIT
Contact: David Orenstein – Picower Institute at MIT
Image: The image is credited to Neuroscience News
Original Research: Open access.
Title: Deep neural networks to register and annotate cells in moving and deforming nervous systems.
Authors: Adam A. Atanas, Alicia Kun-Yang Lu, Brian Goodell, Jungsoo Kim, Saba N. Baskoylu, Di Kang, Talya S. Kramer, Eric Bueno, Flossie K. Wan, Karen L. Cunningham, Brady Weissbourd, and Steven W. Flavell.
Journal: eLife
DOI: 10.7554/eLife.108159.2
Abstract
Deep neural networks to register and annotate cells in moving and deforming nervous systems
Aligning and annotating the many heterogeneous cell types present in complex tissues remains a major challenge for biomedical image analysis. This work presents a suite of deep neural networks designed for automatic, non-rigid registration and cell identification in freely moving and deforming invertebrate nervous systems.
Using a semi-supervised approach, the authors trained a registration network (BrainAlignNet) to align pairs of images of bending Caenorhabditis elegans heads with single-pixel accuracy, enabling neuron linking over time with 99.6% accuracy. The same registration approach was adapted to the jellyfish Clytia hemisphaerica, an organism with a distinct body plan and motion patterns.
A second network (AutoCellLabeler) was trained to annotate more than 100 neuronal cell types in the C. elegans head from multi-spectral fluorescent markers, reaching roughly 98% accuracy and outperforming individual human labelers by aggregating training data across many labeled datasets.
Finally, an unsupervised model (CellDiscoveryNet) learned to discover and annotate over 100 cell types by comparing multi-spectral imaging from many animals, matching the performance of trained human annotators without using manual labels. These tools are immediately useful for a broad range of biological imaging tasks and can be generalized to other contexts that require alignment and annotation of dense, heterogeneous cell types in complex tissues.