{"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/u-rwkv-lightweight-medical-image-segmentation","title":"U-RWKV: Lightweight medical image segmentation with direction-adaptive RWKV","arxiv_id":"2507.11415","date":"2025-07-15","proceeding":null,"authors":["Hongbo Ye","Fenghe Tang","Peiang Zhao","Zhen Huang","Dexin Zhao","Minghao Bian","S. Kevin Zhou"],"abstract":"Achieving equity in healthcare accessibility requires lightweight yet high-performance solutions for medical image segmentation, particularly in resource-limited settings. Existing methods like U-Net and its variants often suffer from limited global Effective Receptive Fields (ERFs), hindering their ability to capture long-range dependencies. To address this, we propose U-RWKV, a novel framework leveraging the Recurrent Weighted Key-Value(RWKV) architecture, which achieves efficient long-range modeling at O(N) computational cost. The framework introduces two key innovations: the Direction-Adaptive RWKV Module(DARM) and the Stage-Adaptive Squeeze-and-Excitation Module(SASE). DARM employs Dual-RWKV and QuadScan mechanisms to aggregate contextual cues across images, mitigating directional bias while preserving global context and maintaining high computational efficiency. SASE dynamically adapts its architecture to different feature extraction stages, balancing high-resolution detail preservation and semantic relationship capture. Experiments demonstrate that U-RWKV achieves state-of-the-art segmentation performance with high computational efficiency, offering a practical solution for democratizing advanced medical imaging technologies in resource-constrained environments. The code is available at https://github.com/hbyecoding/U-RWKV.","url_abs":"https://arxiv.org/abs/2507.11415v1","url_pdf":"https://arxiv.org/pdf/2507.11415v1.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":"u-rwkv-lightweight-medical-image-segmentation","repo_url":"https://github.com/hbyecoding/u-rwkv","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":0,"framework":"pytorch","reach":{"status":"ok","spdx":"MIT"}}],"tasks":[{"task_slug":"computational-efficiency","task_name":"Computational Efficiency"},{"task_slug":"image-segmentation","task_name":"Image Segmentation"},{"task_slug":"long-range-modeling","task_name":"Long-range modeling"},{"task_slug":"medical-image-segmentation","task_name":"Medical Image Segmentation"},{"task_slug":"semantic-segmentation","task_name":"Semantic Segmentation"}],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"syntology_url":"https://syntology.ai/paper/2507.11415","atlas_url":"https://app.syntology.ai/?focus=2507.11415","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2507.11415"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-25T09:33:49+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":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/hbyecoding/u-rwkv","reach":{"status":"ok","spdx":"MIT"}}],"summary":{"ran_honours":1,"ran_fixture":1,"unverified":5},"by_repo_kind":{"official":{"samples":7,"ran":2,"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":0,"samples":[{"code_sha256_prefix":"616b5e82dd2a38f9","entry":"count_params","repo":"hbyecoding/u-rwkv","repo_kind":"official","path":"models/components/compnet1223_enc_rwkv.py","file_url":"https://github.com/hbyecoding/u-rwkv/blob/HEAD/models/components/compnet1223_enc_rwkv.py","link_basis":"plan_row","language":"python","status":"ran_honours","verification_level":2,"contract_check":"HONOURS","metamorphic_tier":"well_formed","behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"616b5e82dd2a38f9"}},{"code_sha256_prefix":"87577b3ff9d32712","entry":"drop_path","repo":"hbyecoding/u-rwkv","repo_kind":"official","path":"models/cmunext/cmunext_rwkv_test_bk.py","file_url":"https://github.com/hbyecoding/u-rwkv/blob/HEAD/models/cmunext/cmunext_rwkv_test_bk.py","link_basis":"harvester_set","language":"python","status":"ran_fixture","verification_level":1,"contract_check":"RAISES","metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"87577b3ff9d32712"}},{"code_sha256_prefix":"ea5a7ec9e9b7cd6c","entry":"autopad","repo":"hbyecoding/u-rwkv","repo_kind":"official","path":"models/cmunext/cmunext_rwkv_test_bk1226.py","file_url":"https://github.com/hbyecoding/u-rwkv/blob/HEAD/models/cmunext/cmunext_rwkv_test_bk1226.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"ea5a7ec9e9b7cd6c"}},{"code_sha256_prefix":"989e1f12a6c53ec3","entry":"cmunext","repo":"hbyecoding/u-rwkv","repo_kind":"official","path":"models/cmunext/CMUNeXt_rwkv_0122.py","file_url":"https://github.com/hbyecoding/u-rwkv/blob/HEAD/models/cmunext/CMUNeXt_rwkv_0122.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"989e1f12a6c53ec3"}},{"code_sha256_prefix":"e19c71ca429478cb","entry":"in_vertical_scan","repo":"hbyecoding/u-rwkv","repo_kind":"official","path":"models/components/compnet1225_enc_rwkv.py","file_url":"https://github.com/hbyecoding/u-rwkv/blob/HEAD/models/components/compnet1225_enc_rwkv.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"e19c71ca429478cb"}},{"code_sha256_prefix":"40e93dc8c16f85db","entry":"lordd","repo":"hbyecoding/u-rwkv","repo_kind":"official","path":"models/components/compnet1225_enc_rwkv_LORDD.py","file_url":"https://github.com/hbyecoding/u-rwkv/blob/HEAD/models/components/compnet1225_enc_rwkv_LORDD.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"40e93dc8c16f85db"}},{"code_sha256_prefix":"8224b1bffcd806c3","entry":"q_shift","repo":"hbyecoding/u-rwkv","repo_kind":"official","path":"models/cmunext/CMUNeXt_rwkv_0122.py","file_url":"https://github.com/hbyecoding/u-rwkv/blob/HEAD/models/cmunext/CMUNeXt_rwkv_0122.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"8224b1bffcd806c3"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}