{"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/recurrent-mask-refinement-for-few-shot","title":"Recurrent Mask Refinement for Few-Shot Medical Image Segmentation","arxiv_id":"2108.00622","date":"2021-08-02","proceeding":"ICCV 2021 10","authors":["Hao Tang","Xingwei Liu","Shanlin Sun","Xiangyi Yan","Xiaohui Xie"],"abstract":"Although having achieved great success in medical image segmentation, deep convolutional neural networks usually require a large dataset with manual annotations for training and are difficult to generalize to unseen classes. Few-shot learning has the potential to address these challenges by learning new classes from only a few labeled examples. In this work, we propose a new framework for few-shot medical image segmentation based on prototypical networks. Our innovation lies in the design of two key modules: 1) a context relation encoder (CRE) that uses correlation to capture local relation features between foreground and background regions; and 2) a recurrent mask refinement module that repeatedly uses the CRE and a prototypical network to recapture the change of context relationship and refine the segmentation mask iteratively. Experiments on two abdomen CT datasets and an abdomen MRI dataset show the proposed method obtains substantial improvement over the state-of-the-art methods by an average of 16.32%, 8.45% and 6.24% in terms of DSC, respectively. Code is publicly available.","url_abs":"https://arxiv.org/abs/2108.00622v2","url_pdf":"https://arxiv.org/pdf/2108.00622v2.pdf","source":{"archive":"pwc-archive (Hugging Face), CC BY-SA 4.0","snapshot":"2025-07-28","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","row_kind":"abstracts"},"code_links":[{"paper_slug":"recurrent-mask-refinement-for-few-shot","repo_url":"https://github.com/uci-cbcl/RP-Net","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":0,"framework":"pytorch","reach":null}],"tasks":[{"task_slug":"few-shot-learning","task_name":"Few-Shot Learning"},{"task_slug":"image-segmentation","task_name":"Image Segmentation"},{"task_slug":"medical-image-segmentation","task_name":"Medical Image Segmentation"},{"task_slug":null,"task_name":"Relation"},{"task_slug":"segmentation","task_name":"Segmentation"},{"task_slug":"semantic-segmentation","task_name":"Semantic Segmentation"}],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2108.00622","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2108.00622"}},"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":[{"provenance":"deterministic:regex_extraction","url":"https://github.com/uci-cbcl/RP-Net","reach":null}],"summary":{"ran":4,"ran_fixture":1,"ran_honours":1,"ran_violates":1,"unverified":2},"by_repo_kind":{"official":{"samples":9,"ran":7,"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":9,"samples":[{"code_sha256_prefix":"00b63b1f519aabeb","entry":"ContextCorrelationEncoder","repo":"uci-cbcl/RP-Net","repo_kind":"official","path":"net/rp_net.py","file_url":"https://github.com/uci-cbcl/RP-Net/blob/HEAD/net/rp_net.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"00b63b1f519aabeb"}},{"code_sha256_prefix":"c85584d983655915","entry":"Correlation","repo":"uci-cbcl/RP-Net","repo_kind":"official","path":"net/rp_net.py","file_url":"https://github.com/uci-cbcl/RP-Net/blob/HEAD/net/rp_net.py","link_basis":"first_harvest_node","language":"python","status":"ran_fixture","verification_level":1,"contract_check":"RAISES","metamorphic_tier":"invariant","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"c85584d983655915"}},{"code_sha256_prefix":"22fe653a5af36242","entry":"Encoder","repo":"uci-cbcl/RP-Net","repo_kind":"official","path":"net/rp_net.py","file_url":"https://github.com/uci-cbcl/RP-Net/blob/HEAD/net/rp_net.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"22fe653a5af36242"}},{"code_sha256_prefix":"05b7ff548b510f24","entry":"ResNet18","repo":"uci-cbcl/RP-Net","repo_kind":"official","path":"net/rp_net.py","file_url":"https://github.com/uci-cbcl/RP-Net/blob/HEAD/net/rp_net.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"05b7ff548b510f24"}},{"code_sha256_prefix":"d383df4b83bcaf60","entry":"Unet_2D","repo":"uci-cbcl/RP-Net","repo_kind":"official","path":"net/rp_net.py","file_url":"https://github.com/uci-cbcl/RP-Net/blob/HEAD/net/rp_net.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"d383df4b83bcaf60"}},{"code_sha256_prefix":"7c8c4c5cb4271dfa","entry":"coords_grid","repo":"uci-cbcl/RP-Net","repo_kind":"official","path":"net/rp_net.py","file_url":"https://github.com/uci-cbcl/RP-Net/blob/HEAD/net/rp_net.py","link_basis":"first_harvest_node","language":"python","status":"ran_honours","verification_level":1,"contract_check":"HONOURS","metamorphic_tier":"well_formed","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"7c8c4c5cb4271dfa"}},{"code_sha256_prefix":"33a4e659a9a6eea3","entry":"dice_loss","repo":"uci-cbcl/RP-Net","repo_kind":"official","path":"net/rp_net.py","file_url":"https://github.com/uci-cbcl/RP-Net/blob/HEAD/net/rp_net.py","link_basis":"first_harvest_node","language":"python","status":"ran_violates","verification_level":1,"contract_check":"VIOLATES","metamorphic_tier":"invariant","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"33a4e659a9a6eea3"}},{"code_sha256_prefix":"5f0d64464cd67f61","entry":"RP_Net","repo":"uci-cbcl/RP-Net","repo_kind":"official","path":"net/rp_net.py","file_url":"https://github.com/uci-cbcl/RP-Net/blob/HEAD/net/rp_net.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":"5f0d64464cd67f61"}},{"code_sha256_prefix":"9d42c0cdaad327a9","entry":"U_Net","repo":"uci-cbcl/RP-Net","repo_kind":"official","path":"net/rp_net.py","file_url":"https://github.com/uci-cbcl/RP-Net/blob/HEAD/net/rp_net.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":"9d42c0cdaad327a9"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}