{"about":{"site":"https://codewithpapers.app","non_affiliation":"Code with Papers and Syntology are not affiliated with, endorsed by, or sponsored by Papers with Code, Meta, or the pwc-archive mirror.","licence":"CC BY-SA 4.0","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","attribution":"https://codewithpapers.app/attribution","modified":"archive material modified by Syntology; see the attribution page"},"url":"/paper/arxiv-2601-15779","title":"Diffusion Model-Based Data Augmentation for Enhanced Neuron Segmentation","arxiv_id":"2601.15779","date":"2026-01-22","proceeding":null,"authors":["Liuyun Jiang","Yanchao Zhang","Jinyue Guo","Yizhuo Lu","Ruining Zhou","Hua Han"],"abstract":"Neuron segmentation in electron microscopy (EM) aims to reconstruct the complete neuronal connectome; however, current deep learning-based methods are limited by their reliance on large-scale training data and extensive, time-consuming manual annotations. Traditional methods augment the training set through geometric and photometric transformations; however, the generated samples remain highly correlated with the original images and lack structural diversity. To address this limitation, we propose a diffusion-based data augmentation framework capable of generating diverse and structurally plausible image-label pairs for neuron segmentation. Specifically, the framework employs a resolution-aware conditional diffusion model with multi-scale conditioning and EM resolution priors to enable voxel-level image synthesis from 3D masks. It further incorporates a biology-guided mask remodeling module that produces augmented masks with enhanced structural realism. Together, these components effectively enrich the training set and improve segmentation performance. On the AC3 and AC4 datasets under low-annotation regimes, our method improves the ARAND metric by 32.1% and 30.7%, respectively, when combined with two different post-processing methods. Our code is available at https://github.com/HeadLiuYun/NeuroDiff.","url_abs":"https://arxiv.org/abs/2601.15779","url_pdf":"https://arxiv.org/pdf/2601.15779","source":{"archive":null,"snapshot":"2025-07-28","note":"not in the Papers with Code archive (frozen at the snapshot)","row_kind":"graph","title_abstract_authors_date":"arXiv metadata, CC0 1.0 (https://info.arxiv.org/help/license)"},"code_links":[],"tasks":[],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2601.15779","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2601.15779"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-24T18:15:14+00:00","read_at_is":"when the build read Syntology's graph, not when any sample ran","claim":"Per-sample execution status on synthesized fixtures; not a correctness claim about the paper. Samples come from repositories linked to the paper, official or community; repo_kind says which.","repos":[{"mentioned_in_github":null,"is_official":null,"provenance":"deterministic:regex_extraction","mentioned_in_paper":null,"url":"https://github.com/HeadLiuYun/NeuroDiff","reach":null}],"summary":{"ran_violates":2,"ran":4,"ran_fixture":1,"ran_honours":1,"ran_draft_wrong":1,"unverified":15},"by_repo_kind":{"found_in_text":{"samples":24,"ran":9,"repositories":1}},"repo_kind_vocabulary":{"official":"The archive marks this repository official for the paper","named_in_paper":"The archive records that the paper mentions this repository; it is not marked official","listed":"In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper","found_in_text":"Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted","community":"Not in the archive's code links for this paper; a community repository Syntology harvested"},"n_pointer_only_for_licence":24,"samples":[{"code_sha256_prefix":"fbf9ec7be545688e","entry":"default","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"diffusion_model/trainer.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/diffusion_model/trainer.py","link_basis":"plan_row","language":"python","status":"ran_violates","verification_level":1,"contract_check":"VIOLATES","metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"fbf9ec7be545688e"}},{"code_sha256_prefix":"608e364a9d2376a3","entry":"exists","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"diffusion_model/trainer.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/diffusion_model/trainer.py","link_basis":"harvester_set","language":"python","status":"ran_violates","verification_level":1,"contract_check":"VIOLATES","metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"608e364a9d2376a3"}},{"code_sha256_prefix":"e41367ad14ff58fd","entry":"get_param_groups_and_shapes","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"diffusion_model/fp16_util.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/diffusion_model/fp16_util.py","link_basis":"harvester_set","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"e41367ad14ff58fd"}},{"code_sha256_prefix":"34484ba7deb03520","entry":"get_window_size","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"segmentation_scripts/model/model_Swin_UNETR.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/segmentation_scripts/model/model_Swin_UNETR.py","link_basis":"plan_row","language":"python","status":"ran_fixture","verification_level":1,"contract_check":"RAISES","metamorphic_tier":"well_formed","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"34484ba7deb03520"}},{"code_sha256_prefix":"e20dd5102da3b050","entry":"make_master_params","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"diffusion_model/fp16_util.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/diffusion_model/fp16_util.py","link_basis":"harvester_set","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"e20dd5102da3b050"}},{"code_sha256_prefix":"bcd8b4acab199405","entry":"make_output_format","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"diffusion_model/logger.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/diffusion_model/logger.py","link_basis":"harvester_set","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"bcd8b4acab199405"}},{"code_sha256_prefix":"4f66ae7f14b3b240","entry":"num_to_groups","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"diffusion_model/trainer.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/diffusion_model/trainer.py","link_basis":"plan_row","language":"python","status":"ran_honours","verification_level":1,"contract_check":"HONOURS","metamorphic_tier":"well_formed","behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"4f66ae7f14b3b240"}},{"code_sha256_prefix":"64fff1e30802b815","entry":"unflatten_master_params","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"diffusion_model/fp16_util.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/diffusion_model/fp16_util.py","link_basis":"harvester_set","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"64fff1e30802b815"}},{"code_sha256_prefix":"129b804760b3115f","entry":"zero_module","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"diffusion_model/neurodiff.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/diffusion_model/neurodiff.py","link_basis":"harvester_set","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"OUTPUT_MISDECLARED","metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"129b804760b3115f"}},{"code_sha256_prefix":"e1909b8866534992","entry":"build_criterion","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"segmentation_scripts/inference.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/segmentation_scripts/inference.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"e1909b8866534992"}},{"code_sha256_prefix":"6c034e82ba5f0bd3","entry":"checkpoint","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"diffusion_model/modules.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/diffusion_model/modules.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"6c034e82ba5f0bd3"}},{"code_sha256_prefix":"fe4eb545bbb728e0","entry":"conv_nd","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"diffusion_model/modules.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/diffusion_model/modules.py","link_basis":"plan_row","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"fe4eb545bbb728e0"}},{"code_sha256_prefix":"3a2fbd46f87f521e","entry":"get_positional_encoding","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"diffusion_model/neurodiff.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/diffusion_model/neurodiff.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"3a2fbd46f87f521e"}},{"code_sha256_prefix":"8f194461987f4c2e","entry":"label_to_condition","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"sample.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/sample.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"8f194461987f4c2e"}},{"code_sha256_prefix":"139a6ad68dd79c69","entry":"load_gt_label","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"segmentation_scripts/evaluate.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/segmentation_scripts/evaluate.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"139a6ad68dd79c69"}},{"code_sha256_prefix":"e4019caa2d0a2f13","entry":"load_h5","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"segmentation_scripts/evaluate.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/segmentation_scripts/evaluate.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"e4019caa2d0a2f13"}},{"code_sha256_prefix":"e515a67f7f32e76d","entry":"mpi_weighted_mean","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"diffusion_model/logger.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/diffusion_model/logger.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"e515a67f7f32e76d"}},{"code_sha256_prefix":"0c6607473a4c4c55","entry":"profile","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"diffusion_model/logger.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/diffusion_model/logger.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"0c6607473a4c4c55"}},{"code_sha256_prefix":"4272aefe07cb6693","entry":"read_mask_stack","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"sample.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/sample.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"4272aefe07cb6693"}},{"code_sha256_prefix":"23e6f260d85a7294","entry":"read_score_lines","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"segmentation_scripts/get_results.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/segmentation_scripts/get_results.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"23e6f260d85a7294"}},{"code_sha256_prefix":"781ac8f422a292c6","entry":"relabel","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"segmentation_scripts/evaluate.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/segmentation_scripts/evaluate.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"781ac8f422a292c6"}},{"code_sha256_prefix":"39e744e9d9af0c31","entry":"uniform_window_3d","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"sample.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/sample.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"39e744e9d9af0c31"}},{"code_sha256_prefix":"8cd5fa7e1d693a92","entry":"window_partition","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"segmentation_scripts/model/model_Swin_UNETR.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/segmentation_scripts/model/model_Swin_UNETR.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"8cd5fa7e1d693a92"}},{"code_sha256_prefix":"1545b891ff0bbdd7","entry":"window_reverse","repo":"HeadLiuYun/NeuroDiff","repo_kind":"found_in_text","path":"segmentation_scripts/model/model_Swin_UNETR.py","file_url":"https://github.com/HeadLiuYun/NeuroDiff/blob/HEAD/segmentation_scripts/model/model_Swin_UNETR.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"1545b891ff0bbdd7"}}]},"arxiv_metadata":{"licence":"arXiv metadata, CC0 1.0 (https://info.arxiv.org/help/license)","fields":["title","abstract","authors","date"],"primary_category":"cs.CV","source":"arxiv_2026.jsonl"},"syntology_extracted_results":null}