NeuroRefiner: Morphology-Aware Multi-Agent Refinement for 3D Fluorescence Microscopy Neuron Segmentation
Analyze & explore related research in Conceptio
Connect this record directly into your research workflow or AI agent. Explore citations across open repositories, query full text, and export structured proof bundles.
NeuroRefiner: Morphology-Aware Multi-Agent Refinement for 3D Fluorescence Microscopy Neuron Segmentation Haiyang Yan1,2 , Jinyue Guo1,3 , Yanchao Zhang1,2 , Bingqing Wang1,3 , Zhenchen Li4 , Jing Liu1 , Jiazheng Liu1 , Linlin Li1 , and Hua Han1,2(B) State Key Laboratory of Brain Cognition and Brain-inspired Intelligence Technology, Institute of Automation, Chinese Academy of Sciences {yanhaiyang2022, hua.han}@ia.ac.cn 2 School of Future Technology, University of Chinese Academy of Sciences 3 School of Artificial Intelligence, University of Chinese Academy of Sciences 4 The State Key Laboratory of Cognitive Neuroscience and Learning, Beijing Normal University, Beijing, China
arXiv:2608.09636v1 [cs.CV] 10 Aug 2026
1
Abstract. Accurate 3D neuron segmentation in fluorescence microscopy is critical for neuroscience. However, the sparse and elongated morphology of neurons poses significant challenges to existing segmentation methods. These methods struggle to preserve both local details and global topology, leading to fragmented results. To address this, we propose NeuroRefiner, a multi-agent system that formalizes the human expert workflow involving iterative global observation and local editing. Specifically, NeuroRefiner comprises three collaborative agents dedicated to diagnosing topological errors, generating correction instructions, and validating refinement quality. To facilitate agent instruction-guided segmentation refinement, we propose TopoRefineNet, a dedicated 3D U-Netbased tool that leverages cross-modality feature fusion to generate refined masks. Through multi-round agent reasoning and voxel-level editing, NeuroRefiner produces topologically more accurate segmentations with enhanced interpretability. Experiments on the BigNeuron, CWMBS, and ZBFWB datasets demonstrate that NeuroRefiner outperforms state-ofthe-art methods, notably achieving a 3.02% improvement in F1 score on the challenging ZBFWB dataset. Keywords: Neuron reconstruction · Multi-agent system · Segmentation refinement · Fluorescence Microscopy
1
Introduction
Neuronal reconstruction aims to derive neuronal skeletons from 3D fluorescence microscopy volumes, thereby enabling quantitative analysis in neuroscience [10, 47]. Accurate reconstruction requires not only recovering fine-grained structural details but also preserving topological correctness. Unlike electron microscopy (EM) neuron reconstruction [14,34,49], which typically deals with dense
2
H. Yan et al. (a) Sliding window
.. Single patch
Input image
..
Segmentation network
Merge Single patch Imperfect mask
Prediction
Input
Interpretability
Global receptive field
(b)
Modify the incorrect result
Global Inspector Input image
Refined mask
and
Segmentation Imperfect network mask
Topological anomaly region detection
Blocks to refine
Merge sub-blocks
Multi-agent Interaction Loop
Refinement Advisor
Change Validator
···
··· TopoRefineNet
and
Generation of refinement instruction Global receptive field
instructions
Segmentation Refinement tool Transparent reasoning
vs
Assess refinement quality Iterative refinement
Fig. 1: (a) Mainstream neuron segmentation paradigms. Sliding window-based methods fail to preserve holistic neuronal morphology. Moreover, a single forward pass cannot rectify segmentation errors and lacks interpretability. (b) Overview of NeuroRefiner. By combining global topological observation with local fine-grained refinement, it generates continuous segmentation results consistent with topological priors. The explicit and transparent reasoning process of the agents offers enhanced interpretability.
cellular structures, fluorescence microscopy poses distinct challenges. These challenges stem from extreme signal sparsity and filamentous morphology within high-resolution 3D volumes that are frequently corrupted by substantial noise [4, 23]. Consequently, robust automated reconstruction requires multi-scale features to preserve both structural details and global connectivity. Neuron segmentation plays a pivotal role in neuronal reconstruction by enhancing weak signals and suppressing noise [20,22]. As shown in Fig. 1 (a), given the prohibitive size of raw image volumes, existing segmentation approaches [37, 38, 41, 44] typically rely on sliding-window strategies for inference. Nevertheless, the limited receptive field in each local window precludes the modeling of long-range dependencies essential for preserving global neuronal morphology [42], resulting in fragmented segmentation. Furthermore, these end-to-end approaches typically operate as black boxes, lacking interpretability and offering no mechanism for the correction of significant topological errors in segmentation. Such errors are challenging to localize and rectify, ultimately constituting a critical bottleneck for the automation of neuronal reconstruction. Agent-based biomedical image segmentation methods [15,16,45] have demonstrated enhanced performance and interpretability via the reasoning and tool in-
NeuroRefiner
3
vocation capabilities of large language models (LLMs). However, existing agents and their associated tool designs remain ill-suited for neuron segmentation. First, they depend on foundation models (e.g., MedSAM2 [25]) as tools, which process 3D volumes slice-by-slice using 2D encoders. Yet, this proves inadequate for neurons, as their structural sparsity and ambiguous boundaries hinder effective feature extraction within isolated 2D planes [51]. Furthermore, these agents confine the workflow to single tool invocations, which lack the flexibility to iteratively refine the segmentation via agent instructions. These limitations underscore the critical need for a dedicated agent framework equipped with domain-specific tools tailored for neuron segmentation. Motivated by the neuroscientist’s cognitive paradigm of global observation and local refinement, we propose NeuroRefiner, a novel multi-agent framework designed for iterative, topology-aware neuron segmentation. As depicted in Fig. 1 (b), our approach comprises three specialized agents: the Global Inspector, which leverages initial segmentation masks and their corresponding Betti numbers to pinpoint sub-regions requiring correction; the Refinement Advisor, which generates detailed refinement instructions for each sub-region; and the Change Validator, which assesses refinement validity and determines whether to accept or regenerate the refinement instructions. Complementing these reasoning agents is a dedicated execution tool, TopoRefineNet, which performs instruction-guided editing of the segmentation mask. This closed-loop pipeline effectively leverages multi-scale features and operates iteratively until the segmentation satisfies the structural completeness of neurons. In summary, our contributions are threefold: – To overcome the limitations of existing neuron segmentation approaches, we present NeuroRefiner—the first LLM-based multi-agent system that achieves topology-aware iterative optimization of segmentation results. – To enable efficient collaboration between reasoning agents and specialized segmentation models, we design TopoRefineNet, a dedicated segmentation editing tool, together with a tailored two-stage training strategy that aligns model behavior with structural refinement instructions. – Experiments on three benchmarks demonstrate that our method outperforms existing neuron segmentation and segmentation refinement approaches.
2
Related Work
Neuron Segmentation. Recent advancements have prioritized the design of specialized architectures to extract multi-scale features, aiming to simultaneously model global neuronal morphology and fine-grained structural details. Architecturally, this is typically realized through the integration of dedicated convolution [41, 43] or graph-based reasoning modules [35]. In terms of morphological constraints, methods such as SGSNet [44] and Tubular [37] augment feature representation via multi-task learning that predicts auxiliary geometric cues. To capture intrinsic morphological priors, MP-NRGAN [5] employs adversarial training utilizing weak supervision signals derived from reconstructed neurons. Furthermore, to mitigate the limitations of restricted receptive fields inherent in
4
H. Yan et al.
patch-based inference, GBP-Net [42] encodes long-range contextual information and fuses it with intra-patch features. Despite these advancements, prevailing models remain hindered by singlepass inference paradigms, which preclude iterative error correction. While certain approaches [3, 52] utilize point cloud networks to refine segmentation outputs, they fundamentally fail to recall omitted structural segments (false negatives) missed during the initial prediction. Conversely, our framework introduces a multi-round refinement mechanism guided by multi-scale features, effectively mitigating the errors in single-pass models. Agent for Biomedical Image Segmentation. Existing agent-based approaches generally fall into two distinct paradigms. The first paradigm relies on conventional multi-agent reinforcement learning (MARL) for low-level, pixel-wise decision-making. Methods such as IteR-MRL [19] and BS-IRIS [24] employ multiagent reinforcement learning for interactive 3D segmentation, treating voxels as collaborative agents to iteratively refine results. Similarly, dbMiM [6] employs MARL to optimize masking strategies for self-supervised neuron segmentation. However, these approaches are predominantly tailored for electron microscopy or general 3D modalities (e.g., MRI/CT), rendering them ineffective for handling neurons in fluorescence microscopy. The second paradigm represents a more advanced, LLM-driven approach. These agents are LLM-centric and leverage planning-execution loops to invoke external segmentation tools. For instance, Ophiuchus [16] adopts a three-stage training strategy to coordinate vision tools like SAM2 [31] and BiomedParse [53] for enhanced lesion segmentation. IBISAgent [15] generates textual click instructions to invoke interactive segmentation tools for progressive mask refinement, while GenCellAgent [45] introduces a multi-agent framework that achieves intelligent tool routing via a planner-executor-evaluator loop. Although these methods have improved accuracy, they lack topology-aware agent design and tools specifically tailored for sparse neurons. Segmentation Refinement. High-resolution image segmentation conventionally necessitates downsampling operations or patch-based inference strategies, which incur boundary artifacts or information loss. To mitigate these issues, a series of refinement methods has been proposed. Mask Transfiner [17] identifies incoherent regions to perform fine-grained label correction. SegRefiner [36] introduced a generic, iterative framework leveraging discrete diffusion processes to correct diverse segmentation errors. To achieve accurate segmentation of EM neurons, FGNet [18] refines the semantic features extracted by SAM2 [30] by training a lightweight fine-grained encoder. While effective for boundary regularization, existing methods neglect global topology, which is paramount for preserving filamentous neuronal structures. Hence, topology-aware refinement for neuronal structures remains a critical, underexplored frontier.
NeuroRefiner Step1 : Check the abnormal sub-blocks and output their types.
Step 2: Propose correction suggestions based on the image and the segmentation
Step 3: Refine the segmentation based on the revised suggestions
Block 1
𝑀init
MIP
yz
𝑀1
… result removed the noise according to the instruction … Instruction adherence
𝐼
+
xy
𝑀1
Step 4: Compare the results to Step n : The result is determine whether to accept . …… complete and can be output.
Change Validator
TopoRefineNet 𝐼1
Refinement Advisor
…
𝑀1 , 𝐼1
𝓑fix :{1:’FP’, 5:’FN’, … }
𝐼5
xy
𝑀5
TopoRefineNet yz
𝑀5
3D view
… result removed the noise according to the instruction
Refined Segmentation …
Multi round
Instruction adherence
Refinement Advisor 𝑀5 , 𝐼5
𝒕5 : Connect the neurons
located in the middle upper posterior region
Merge
… after refinement has less irrelevant structure and should be accepted.
Change Validator
Block 5
Global inspector
Merge
Topological improvement
Text Encoder
Connected components: [8,4,3,5,7,3]
…, from the overall image, … break in the middle…. Rechecking the id 5, … have a false negative, … … examined the block with the high connected component … id 1, found noise …
𝑀1refined
𝒕1 : Remove the isolated noise located in the left upper front
xy
𝑀𝑖
5
Text Encoder
𝑀5refined
… after refinement has better continuity and should be accepted. Topological improvement
𝑀t
Fig. 2: Collaborative NeuroRefiner workflow. The system performs multi-round refinement via the interaction of agents and TopoRefineNet to obtain topologically complete segmentation results.
3
Method
Sec. 3.1 describes the collaborative mechanism among agents in NeuroRefiner. Sec. 3.2 details the design of each individual agent and Sec. 3.3 presents the architecture and training strategy of our proposed tool, TopoRefineNet. 3.1
NeuroRefiner
Manual refinement of neuron segmentation follows a hierarchical workflow in which experts iteratively alternate between global morphological assessment and fine-grained local editing. Motivated by this paradigm, we propose a multi-agent collaborative framework, as outlined in Algorithm 1. The segmentation mask M^{(t)} is initialized using an off-the-shelf segmentation model. At each iteration, the Global Inspector conducts a holistic assessment of M^{(t)} to detect sub-regions that violate topological priors. Subsequently, the Refinement Advisor generates a set of structure-aware refinement instructions {ti }, each tailored to a specific anomalous sub-region. These instructions guide TopoRefineNet to produce refined sub-region masks, denoted as {Mirefined }. As a validation mechanism, the Change Validator acccepts only those refinements that yield an improvement over the original segmentation, as M^{(t+1)} = \text {Merge}\left ( M^{(t)}, \left \{ M_i^{\text {refined}} \mid \text {Validator}(M_i^{\text {refined}}) = \text {True} \right \} \right ). \label {eq:important}
(1)
As illustrated in Fig. 2, the process terminates when the Global Inspector determines that the neuronal morphology is complete or upon reaching T_{\text {max}} iterations. This iterative protocol ensures transparency, as the explicit diagnostic decisions and instructions generated at each step are auditable.
6
H. Yan et al.
Algorithm 1 NeuroRefiner: Morphology-Aware Multi-Agent Refinement Require: 3D image I, initial mask M (0) , max iterations Tmax , iteration t ← 0 1: while t < Tmax do 2: Compute z-axis MIP M xy , sub-blocks {bi }, and Betti numbers {β0 (bi )} 3: Bf ix , {τi } ← πins (M xy , {Mixy }, {β0 (bi )}) // Identify blocks to fix 4: if Bf ix = ∅ then break //morphologically converged 5: for each block bi ∈ Bf ix do 6: Initialize retry counter k ← 0 7: while k < Tmax do 8: ti ← πadv (Iixy , Iiyz , Mixy , Miyz , τi ) // Generate refinement instruction 9: Miref ined ← TopoRefineNet(Ii , Mi , ti ) // Instruction-guided editing 10: ri ← πval (Mixy , Mirefined , ti , τi ) // Validate refinement quality 11: if ri is Accept then 12: M [bi ] ← Mirefined , break //Update local mask 13: else 14: k ← k + 1 //Retry 15: end if 16: end while 17: end for 18: t←t+1 19: end while 20: return M final ← M (t)
3.2
Design of Agents in NeuroRefiner
Global Inspector leverages the high discriminability of global morphological anomalies to identify sub-regions violating topological priors. Given an initial 3D segmentation mask Minit and the corresponding image I, we first apply Maximum Intensity Projection (MIP) along the z-axis to obtain projection M xy and I xy . Performing MIP on sparse volumes facilitates their input into VLMs while inducing minimal signal overlap. The projected volume is then partitioned into a set of non-overlapping 2D sub-blocks B = {bi }N i=1 , yielding corresponding image and mask patches {Iixy } and {Mixy }. As a crucial metric for neuron segmentation, topological correctness serves as an effective guide for identifying erroneous regions. For each sub-block, the 0thorder Betti number is computed to quantify topological irregularities as β0 (bi ) = CC(Mi ), where CC(·) denotes the connected component counting function in the corresponding 3D region. An elevated β0 value, which indicates a greater number of isolated fragments within a sub-block, suggests a higher likelihood of topological errors. Based on this topological metric and the mask, the inspection policy πins identifies abnormal regions and categorizes their error types: \operatorname {B}_{\mathrm {fix}},{\tau _i}=\pi _{\mathrm {ins}}(M^{xy},\{M_i^{\mathrm {xy}}\},{\beta _0(b_i)}),
(2)
where Bf ix ⊆ B denotes the subset of blocks requiring refinement and τi labels each block as either false-positive or false-negative. They provide guidance for the Refinement Advisor to generate detailed instructions.
NeuroRefiner
7
Refinement Advisor generates structured, morphology-aware instructions to guide the correction of identified sub-blocks. To precisely delineate the abnormal region along the z-axis, the agent first applies MIP to the 3D region corresponding to bi along the x-axis and then localizes the defect in the yz-plane. Upon obtaining the xy and yz plane projections of the target region, it jointly analyzes the image and mask from orthogonal views to uncover connectivity cues and formulate a refinement instruction, as \mathbf {t}_i = \pi _{\mathrm {adv}}\big (I_i^{xy}, I_i^{yz}, M_i^{xy}, M_i^{yz}, \tau _i\big ).
(3)
Each instruction ti explicitly encodes both a spatial location (e.g., “upper-leftposterior”) and an editing operation (e.g., “connect” or “remove”). The editing module TopoRefineNet subsequently performs refinement based on the instruction and outputs Mirefined . This design decouples reasoning from voxel-level operations by using natural language as an intermediary, thereby fully unleashing the agent’s reasoning and expressive capabilities. Change Validator ensures the reliability of iterative refinement by evaluating the output of TopoRefineNet and preventing erroneous edits from being merged. Specifically, it compares the initial mask Mi against the refined result Mirefined , conditioned on the instruction ti and the error type τi to assess whether the edit is beneficial to morphology. This evaluation is formalized as r_i = \pi _{\mathrm {val}}\big (M_i^{xy}, M_i^{\mathrm {refined}}, \mathbf {t}_i, \tau _i\big ).
(4)
The Change Validator only approves an edit when two criteria are satisfied: (1) instruction adherence—the modification correctly implements the spatial and operational intent of ti , and (2) topological improvement—the refined mask exhibits reduced fragmentation or noise without introducing new artifacts. If either condition fails, it triggers the refinement advisor to generate a new instruction, thereby closing the loop with corrective feedback. For cases that remain unresolved after multiple refinement cycles, the system escalates the sub-block to human experts for manual intervention. 3.3
TopoRefineNet
Architecture. Serving as the critical bridge between high-level agent reasoning and low-level pixel operations, TopoRefineNet executes the conversion of topological instructions into voxel-level mask modifications. Inspired by recent advances in multimodal conditional generation [8, 33, 48], TopoRefineNet formulates segmentation refinement as an instruction-guided image editing task. As illustrated in Fig. 3 (a), it takes the image Ii and the noisy mask Mi ∈ RH×W ×D as input, which are concatenated and encoded by a visual encoder Eθ to produce a hierarchical feature pyramid, as {Fvi }L i=1 = Eθ ([Ii ; Mi ]). The instruction ti is embedded via a frozen pre-trained text encoder, as Ft = Tϕ (t) ∈ Rd . The interaction is achieved via cross-attention between the deepest visual features FvL and Ft , formulated as
8
H. Yan et al.
(a) Instruction Refinement Advisor
Text Encoder
TopoRefineNet
Cross Attention Initial mask
Image
Image Encoder
Image Prediction Decoder
(b) Stage 1
Stage 2
Synthetic instruction
Instruction from VLM
Prediction
Image
Synthetic defect
Data Augmentation
Prediction
Image
Label
Real defect
Label
Fig. 3: (a) Architecture of TopoRefineNet. The image and instruction are encoded by their respective encoders, fused via cross-attention, and decoded to generate the corrected mask. (b) The proposed two-stage training paradigm. The first stage is trained on synthetic defects, while the second stage utilizes real masks with defects.
\mathbf {F}_L^{\prime v}= \text {CrossAttn}(\mathbf {F}_L^v, \mathbf {F}^t) = \text {softmax}\left (\frac {\mathbf {W}_q \mathbf {F}_L^v (\mathbf {W}_k \mathbf {F}^t)^\top }{\sqrt {d_k}}\right ) \mathbf {W}_v \mathbf {F}^t,
(5)
where Wq , Wk , Wv are learnable projection matrices and dk denotes the feature dimension. This mechanism dynamically recalibrates visual features by attending to regions and editing operations specified in the instruction. The fused feature F′v L is passed to the decoder D, which integrates it with shallow features via skip connections and upsamples to yield the refined mask Mrefined . The entire process is concisely expressed as a conditional generation function, as M_{\mathrm {refined}} = \mathcal {F}_\theta (I_i, M_{\mathrm {i}}, \mathbf {t_i}) = \mathcal {D} \circ \text {CrossAttn}\big (\mathcal {E}_\theta ([I_i; M_{\mathrm {i}}]), \mathcal {T}_\phi (\mathbf {t_i})\big ).
(6)
This design addresses the limitations of general tools in editing sparse neuronal structures, empowering the system with essential capabilities for fine-grained morphological refinement. Training Strategy. To endow TopoRefineNet with precise instruction-to-voxel editing capability, we devise a two-stage training strategy as shown in Fig. 3 (b). The initial stage prioritizes cross-modal alignment using synthetic data with morphological defects. Specifically, ground-truth masks M_{\mathrm {gt}} are perturbed via data augmentation to yield corrupted variants (Sec. 4.1), with corresponding refinement instructions generated based on the location and type of perturbation. The second stage subsequently shifts focus toward real-world generalization. We curate segmentation outputs from multiple networks trained on diverse neuronal datasets and specifically select instances that exhibit representative topological
NeuroRefiner
9
errors. For each identified error, a refinement instruction is synthesized using Qwen3-VL [1]. By progressively increasing defect complexity from synthetic scenarios to complex real-world cases, this scheme stabilizes optimization and enhances robustness against the diverse instructions encountered during inference.
4
Experiments
4.1
Experiments Setup
Dataset. To evaluate NeuroRefiner across diverse neuronal morphologies and imaging conditions, we conducted experiments on three representative benchmarks, adhering to the training–testing splits established in prior work [22,42,44]. The BigNeuron [28] dataset comprises expert-annotated neurons spanning multiple species. Images were acquired across different laboratories and imaging platforms, exhibiting substantial variations in appearance, resolution, and scale. The CWMBS [22] dataset is derived from whole-brain mouse imaging and contains volumes of size 256 × 256 × 256 voxels (physical resolution: 0.2µm × 0.2µm × 1µm), of which 83 exhibit strong background noise and 162 feature thin filamentary structures with weak signals. The ZBFWB dataset [9] consists of confocal microscopy images from 6-day-old zebrafish larvae (1000 × 2000 × 250 voxels; 0.5µm × 0.5µm × 1µm), presenting challenges due to its low contrast, long-range axonal projections, intricate network topologies, and low contrast. Metrics. To ensure consistency with prior work [22, 42, 44], we use reconstructions generated by APP2 [39] to assess segmentation quality from point-level accuracy to topological completeness. We compute the precision, recall, and F1score to evaluate voxel-level accuracy and further employ the distance-based metric Spatial Distance (SD) to measure the average bidirectional distance between prediction and ground-truth structures. Its variant SSD [29], excludes matched points within 2 voxels to focus on significant topological errors. Finally, MES [40] evaluates the overall morphological fidelity of topology by quantifying the lengths of missing and extra structures. Lower SD and SSD values, along with higher values for all other metrics, indicate superior reconstruction. Details. In our experiments, all agents use Qwen3-VL-8B [1] as the foundation model, and text embeddings are obtained from Qwen3 Embedding 0.6B [50]. The Global Inspector performs error detection using a region of 128 × 128 pixels as its fundamental unit. The maximum number of iterations T_{\text {max}} is set to 5. The prompts for all agents and the template of instructions for TopoRefineNet are provided in the supplementary material. Both training stages of TopoRefineNet employ cross-entropy loss and dice loss [26]. In the first stage, to synthesize FN segmentations, we randomly select a positive sample as the center and perform an erosion operation with a random kernel size k ∈ [5, 20] to generate breaks of varying lengths. For FP samples,
10
H. Yan et al.
Table 1: Quantitative results (%) on the CWMBS dataset. The best results are bolded and the second-best results are underlined in the following table.
Models
Weak Signal
Strong Noise
SD↓ SSD↓ PRE↑ REC↑ F1↑ MES↑ SD↓ SSD↓ PRE↑ REC↑ F1↑ MES↑
3D Segmentation Baselines 3D U-Net SwinU nnUNet
28.35 33.71 93.85 62.37 74.94 46.21 24.98 31.61 94.16 69.57 79.23 54.30 24.94 31.14 94.07 65.07 76.93 56.45 19.43 27.17 93.71 74.40 80.98 56.30 25.97 31.66 95.03 68.82 79.83 54.76 33.65 39.89 90.38 61.43 73.14 52.19
Neuron-tailored architectures 23.80 29.97 93.58 71.12 80.82 55.47 34.96 40.17 88.16 56.34 68.75 46.95 35.70 40.82 92.44 55.63 69.46 50.29 19.59 26.79 94.79 73.13 82.56 58.25 GBP-Net 11.85 19.27 95.06 83.50 88.91 67.09 17.15 24.94 93.41 75.93 83.77 59.41 ADTL-Net 25.96 31.65 93.49 62.72 75.07 53.51 28.11 35.18 95.24 60.68 74.13 52.79 V-Net
SGSNet
Refinement approaches on 3D UNet Transfiner 27.33 32.14
94.10 62.28 74.95 46.22 24.53 31.52 94.33 70.22 80.51 56.28 26.74 32.09 93.00 64.72 76.32 48.93 23.22 30.18 95.39 70.11 80.82 57.52 SegRefiner 17.61 23.79 94.92 77.84 85.54 62.31 21.64 27.80 92.19 72.84 81.38 57.91 Ours 9.17 15.21 95.72 85.05 90.07 70.32 13.29 19.52 94.31 79.66 86.37 64.39 SegFix
Refinement approaches on nnUNet Transfiner 25.34 31.73
95.63 66.29 78.30 53.16 33.24 39.70 91.69 63.79 75.24 53.27 24.82 30.57 95.47 69.69 80.57 56.79 30.09 35.35 92.74 61.02 73.61 55.38 SegRefiner 15.98 22.60 96.71 73.46 83.50 61.41 22.53 27.83 94.35 66.37 77.92 54.93 Ours 7.26 11.86 96.79 87.39 91.85 72.60 15.96 20.17 93.76 77.47 84.84 61.82 SegFix
we either randomly copy-paste annotations of other neurons or introduce Gaussian noise within local regions. In the second stage, we trained several common architectures—namely, 3D U-Net [7], V-Net [21], UNETR [12], nnFormer [54], and SwinUNETR [11]—on a subset of the training set, followed by inference on the complete training set. Based on connected component and F1 scores, we selected a total of 5,310 volumes of 128 × 128 × 64 voxels exhibiting significant segmentation errors for training. 4.2
Quantitative Results
We comprehensively evaluated NeuroRefiner against a diverse set of state-ofthe-art (SOTA) methods for fluorescence microscopy neuron segmentation. The baselines include 3D U-Net [7], SwinUNETR [11], and nnUNet [13]; neurontailored architectures SGSNet [44], GBP-Net [42], Improved V-Net [21], and ADTL-Net [22]; refinement-based methods Mask Transfiner [17], SegFix [46], and SegRefiner [36]. We reproduced the corresponding 3D versions of the 2Dbased refinement-based methods using the same training data as TopoRefineNet.
NeuroRefiner
11
Table 2: Quantitative results (%) on the BigNeuron and ZBFWB datasets.
ZBFWB
BigNeuron
Method
SD↓ SSD↓ PRE↑ REC↑ F1↑ MES↑
3D U-Net [7] 7.23 nnUNet [13] 5.16 Improved V-Net [21] 7.40 SGSNet [44] 5.90 GBP-Net [42] 4.89 ADTL-Net [22] 5.12 Mask Transfiner [17] 6.82 SegFix [46] 7.18 SegRefiner [36] 5.79 Ours 3.39
11.27 8.68 11.16 9.34 8.41 8.39 10.57 11.35 9.94 6.94
83.15 87.34 82.50 86.22 88.83 88.25 86.14 83.92 87.18 89.74
89.76 86.33 66.82 86.87 87.10 68.72 90.41 86.27 67.16 90.28 88.20 69.90 89.23 89.03 70.77 87.14 87.69 69.32 88.89 87.49 69.01 89.19 86.47 66.32 89.23 88.19 69.27 90.37 90.05 72.02
3D U-Net [7] 38.65 45.41 86.33 nnUNet [13] 27.94 34.40 86.40 Improved V-Net [21] 42.43 49.06 84.27 SGSNet [44] 44.34 50.29 82.54 GBP-Net [42] 20.35 28.44 91.13 ADTL-Net [22] 46.74 54.87 82.53 Mask Transfiner [17] 36.23 43.12 88.26 SegFix [46] 36.42 42.97 87.21 SegRefiner [36] 30.18 36.25 88.25 Ours 16.31 23.66 92.77
63.15 72.94 53.63 67.60 75.85 61.78 58.16 68.82 51.45 55.36 66.27 48.14 76.39 83.11 71.41 52.77 64.38 48.79 63.92 74.14 54.87 62.94 73.11 55.96 65.19 74.99 60.32 80.38 86.13 74.25
Tab. 1 reports quantitative results on the CWMBS subsets. CNN-based models (3D U-Net, SGSNet) are hampered by local receptive fields, limiting generalization under distribution shifts, whereas multiscale methods (GBP-Net, SwinUNETR) demonstrate superior accuracy. Due to a lack of topology-centric refinement, Transfiner and SegFix yield only marginal gains on the masks of 3D U-Net and nnUNet. In contrast, our approach leverages agent-based reasoning and topology-centric refinement and improves F1 scores of various segmentation masks by over 10% on the weak-signal subset, establishing SOTA results across both subsets. Tab. 2 summarizes results on the BigNeuron and ZBFWB datasets, where all refinement approaches utilize 3D U-Net segmentation outputs as initialization. Benefiting from multi-step refinement, both SegRefiner and our approach boost F1 scores on the BigNeuron dataset. Specifically, our method improves MES by 5.20% and reduces SSD by 4.33 relative to the initial segmentation. On the more challenging ZBFWB dataset, conventional CNN-based methods struggle to handle low-contrast regions, resulting in fragmented segmentation and suppressed recall. Although GBP-Net leverages enhanced multi-scale features to surpass other methods, it remains unable to correct topological defects. Furthermore, we observe that all segmentation refinement approaches yield consistent improvements. Notably, our method elevates the F1 score of 3D U-Net by 13.19%, achieving an F1 score 3.02% higher than GBP-Net. These results validate the effectiveness of our multi-agent system in leveraging topological priors and global features for iterative refinement.
12 (a)
Image
H. Yan et al. (c)
(b)
Ground truth
(e)
(d)
nnUNet
Swin UNETR
GBP-Net
(f) (f)
ADTL-Net
(g)
3D UNet
(h)
(i)
SegRefiner
SegFix
(j)
Ours
Fig. 4: Qualitative comparison of original images, ground truth, and segmentation results on the ZBFWB dataset. The orange boxes highlight regions with weak signals. Our method preserves the complete neuronal structure and yields superior performance. Table 3: Ablation Study on Agents in NeuroRefiner. Global Refinement Change Inspector Advisor Validator ✓ ✓ ✓
✓
✓
✓
ZBFWB SSD F1 (%)
CWMBS SSD F1 (%)
✓
34.52 32.05 25.24 30.69
76.39 77.29 83.47 81.43
30.23 28.47 18.93 24.53
79.96 80.34 87.14 83.12
✓
23.66
86.13
16.65
88.83
Fig. 4 depicts a neuron with long-range projections from the ZBFWB dataset, where irrelevant tissue signals significantly obscure the target structure. All single-pass segmentation methods yield results plagued by topological violations, characterized by extensive fragmentation and artifacts. Although refinementbased paradigms correct local errors in segmentation, they lack a neuron-specific error identification and correction protocol. In contrast, our method leverages agent-based reasoning and neuronal topological priors to guide the refinement with TopoRefineNet, achieving superior reconstructions. 4.3
Ablation Study
Ablation Study on Agents in NeuroRefiner. To validate the efficacy of the proposed multi-agent system, we conducted ablation studies on the ZBFWB and CWMBS datasets, summarized in Table 3. In the absence of agents, TopoRefineNet refines all blocks indiscriminately for Tmax rounds. While this improves initial segmentation, the lack of global context and targeted guidance yields suboptimal results. Employing solely the Global Inspector for both error localization and instruction generation provides negligible gains, as a single agent struggles to manage these distinct tasks and lacks quality control mechanisms. The Refinement Advisor decouples instruction generation from inspection. This functional specialization yields F1 gains of 6.18% and 6.80%, respectively. Further incorporating the Change Validator to filter unreliable refinements yields
NeuroRefiner
13