Reproduce “FGNet: Leveraging Feature-Guided Attention to Refine SAM2 for 3D EM Neuron Segmentation” on OpenResearch

Turn your favorite coding agents into research agents that review literature, develop hypotheses, and run experiments.

Universal app for Apple silicon and Intel. Requires macOS 11 or later.

About this paper

Zhenghua Li, Hang Chen, Zihao Sun, Kai Li, Xiaolin Hu

Researchers from Tsinghua University developed FGNet, a framework that leverages the Segment Anything 2 (SAM2) foundation model for 3D EM neuron segmentation by integrating a Feature-Guided Attention module and a lightweight Fine-Grained Encoder. This approach achieved state-of-the-art results on challenging datasets like AC3/AC4, CREMI, and Wafer4, improving Variation of Information (VOI) by 12.5% on AC3/AC4 compared to prior methods.

Key insights

  • Effectively bridging the domain gap between natural images and specialized EM data can be achieved by integrating general-purpose features from a large foundation model with domain-adaptive, fine-grained refinement through an attention mechanism.
  • A lightweight and shallow Fine-Grained Encoder, when guided by features from a foundation model, can efficiently capture intricate details specific to EM images without requiring a deep architecture.
  • Hierarchical feature-guided attention allows for progressive refinement, shifting focus from broader structural regions to increasingly fine-grained details, which is crucial for accurately segmenting complex neural morphologies.

Summary by alphaXiv.

Tip: change arxiv.org to autoarxiv.org in any arXiv abstract link to open the paper here.