{"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/translation-consistent-semi-supervised","title":"Translation Consistent Semi-supervised Segmentation for 3D Medical Images","arxiv_id":"2203.14523","date":"2022-03-28","proceeding":null,"authors":["Yuyuan Liu","Yu Tian","Chong Wang","Yuanhong Chen","Fengbei Liu","Vasileios Belagiannis","Gustavo Carneiro"],"abstract":"3D medical image segmentation methods have been successful, but their dependence on large amounts of voxel-level annotated data is a disadvantage that needs to be addressed given the high cost to obtain such annotation. Semi-supervised learning (SSL) solve this issue by training models with a large unlabelled and a small labelled dataset. The most successful SSL approaches are based on consistency learning that minimises the distance between model responses obtained from perturbed views of the unlabelled data. These perturbations usually keep the spatial input context between views fairly consistent, which may cause the model to learn segmentation patterns from the spatial input contexts instead of the segmented objects. In this paper, we introduce the Translation Consistent Co-training (TraCoCo) which is a consistency learning SSL method that perturbs the input data views by varying their spatial input context, allowing the model to learn segmentation patterns from visual objects. Furthermore, we propose the replacement of the commonly used mean squared error (MSE) semi-supervised loss by a new Cross-model confident Binary Cross entropy (CBC) loss, which improves training convergence and keeps the robustness to co-training pseudo-labelling mistakes. We also extend CutMix augmentation to 3D SSL to further improve generalisation. Our TraCoCo shows state-of-the-art results for the Left Atrium (LA) and Brain Tumor Segmentation (BRaTS19) datasets with different backbones. Our code is available at https://github.com/yyliu01/TraCoCo.","url_abs":"https://arxiv.org/abs/2203.14523v2","url_pdf":"https://arxiv.org/pdf/2203.14523v2.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":"translation-consistent-semi-supervised","repo_url":"https://github.com/yyliu01/tracoco","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":1,"framework":"pytorch","reach":{"status":"ok","spdx":"MIT"}},{"paper_slug":"translation-consistent-semi-supervised","repo_url":"https://github.com/MindSpore-scientific/code-13/tree/main/Translation-Invariant/model","is_official":0,"mentioned_in_paper":0,"mentioned_in_github":0,"framework":"mindspore","reach":null}],"tasks":[{"task_slug":"brain-tumor-segmentation","task_name":"Brain Tumor Segmentation"},{"task_slug":"image-segmentation","task_name":"Image Segmentation"},{"task_slug":"medical-image-segmentation","task_name":"Medical Image Segmentation"},{"task_slug":"segmentation","task_name":"Segmentation"},{"task_slug":"semantic-segmentation","task_name":"Semantic Segmentation"},{"task_slug":"semi-supervised-semantic-segmentation","task_name":"Semi-Supervised Semantic Segmentation"},{"task_slug":"semi-supervised-medical-image-segmentation","task_name":"Semi-supervised Medical Image Segmentation"},{"task_slug":"translation","task_name":"Translation"},{"task_slug":"tumor-segmentation","task_name":"Tumor Segmentation"}],"methods":[{"method_slug":"cutmix","method_name":"CutMix"}],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2203.14523","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2203.14523"}},"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":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/MindSpore-scientific/code-13/tree/main/Translation-Invariant/model","reach":null},{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/yyliu01/tracoco","reach":{"status":"ok","spdx":"MIT"}}],"summary":{"unverified":9},"by_repo_kind":{"official":{"samples":9,"ran":0,"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":"3a1cb8e0a7bf0e2f","entry":"formatter_message","repo":"yyliu01/tracoco","repo_kind":"official","path":"Code/UnetACDC/Utils/logger.py","file_url":"https://github.com/yyliu01/tracoco/blob/HEAD/Code/UnetACDC/Utils/logger.py","link_basis":"harvester_set","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":"3a1cb8e0a7bf0e2f"}},{"code_sha256_prefix":"8cb07b40b112a0b6","entry":"get_acdc_2d_cct","repo":"yyliu01/tracoco","repo_kind":"official","path":"Code/UnetACDC/validate.py","file_url":"https://github.com/yyliu01/tracoco/blob/HEAD/Code/UnetACDC/validate.py","link_basis":"harvester_set","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":"8cb07b40b112a0b6"}},{"code_sha256_prefix":"bafb7215e3542d98","entry":"grouper","repo":"yyliu01/tracoco","repo_kind":"official","path":"Code/UnetACDC/Dataloader/dataloader.py","file_url":"https://github.com/yyliu01/tracoco/blob/HEAD/Code/UnetACDC/Dataloader/dataloader.py","link_basis":"harvester_set","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":"bafb7215e3542d98"}},{"code_sha256_prefix":"bcd062b45bf8f796","entry":"iterate_eternally","repo":"yyliu01/tracoco","repo_kind":"official","path":"Code/UnetACDC/Dataloader/dataloader.py","file_url":"https://github.com/yyliu01/tracoco/blob/HEAD/Code/UnetACDC/Dataloader/dataloader.py","link_basis":"harvester_set","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":"bcd062b45bf8f796"}},{"code_sha256_prefix":"e301de7d9ddb6c3f","entry":"iterate_once","repo":"yyliu01/tracoco","repo_kind":"official","path":"Code/UnetACDC/Dataloader/dataloader.py","file_url":"https://github.com/yyliu01/tracoco/blob/HEAD/Code/UnetACDC/Dataloader/dataloader.py","link_basis":"harvester_set","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":"e301de7d9ddb6c3f"}},{"code_sha256_prefix":"25617baa9830b14c","entry":"kaiming_normal_init_weight","repo":"yyliu01/tracoco","repo_kind":"official","path":"Code/UnetACDC/Model/Unet2D.py","file_url":"https://github.com/yyliu01/tracoco/blob/HEAD/Code/UnetACDC/Model/Unet2D.py","link_basis":"harvester_set","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":"25617baa9830b14c"}},{"code_sha256_prefix":"d49d0cc5fe4d30cd","entry":"softmax_helper","repo":"yyliu01/tracoco","repo_kind":"official","path":"Code/UnetACDC/Model/nnUnet.py","file_url":"https://github.com/yyliu01/tracoco/blob/HEAD/Code/UnetACDC/Model/nnUnet.py","link_basis":"harvester_set","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":"d49d0cc5fe4d30cd"}},{"code_sha256_prefix":"65b6d4bec11b0fc1","entry":"sparse_init_weight","repo":"yyliu01/tracoco","repo_kind":"official","path":"Code/UnetACDC/Model/Unet2D.py","file_url":"https://github.com/yyliu01/tracoco/blob/HEAD/Code/UnetACDC/Model/Unet2D.py","link_basis":"harvester_set","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":"65b6d4bec11b0fc1"}},{"code_sha256_prefix":"fd42e6c10ebb5866","entry":"test_single_case","repo":"yyliu01/tracoco","repo_kind":"official","path":"Code/UnetACDC/validate.py","file_url":"https://github.com/yyliu01/tracoco/blob/HEAD/Code/UnetACDC/validate.py","link_basis":"harvester_set","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":"fd42e6c10ebb5866"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}