{"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-2603-13328","title":"Self-Supervised Multi-Stage Domain Unlearning for White-Matter Lesion Segmentation","arxiv_id":"2603.13328","date":"2026-03-04","proceeding":null,"authors":["Domen Preložnik","Žiga Špiclin"],"abstract":"Inter-scanner variability of magnetic resonance imaging has an adverse impact on the diagnostic and prognostic quality of the scans and necessitates the development of models robust to domain shift inflicted by the unseen scanner data. Review of recent advances in domain adaptation showed that efficacy of strategies involving modifications or constraints on the latent space appears to be contingent upon the level and/or depth of supervision during model training. In this paper, we therefore propose an unsupervised domain adaptation technique based on self-supervised multi-stage unlearning (SSMSU). Building upon the state-of-the-art segmentation framework nnU-Net, we employ deep supervision at deep encoder stages using domain classifier unlearning, applied sequentially across the deep stages to suppress domain-related latent features. Following self-configurable approach of the nnU-Net, the auxiliary feedback loop implements a self-supervised backpropagation schedule for the unlearning process, since continuous unlearning was found to have a detrimental effect on the main segmentation task. Experiments were carried out on four public datasets for benchmarking white-matter lesion segmentation methods. Five benchmark models and/or strategies, covering passive to active unsupervised domain adaptation, were tested. In comparison, the SSMSU demonstrated the advantage of unlearning by enhancing lesion sensitivity and limiting false detections, which resulted in higher overall segmentation quality in terms of segmentation overlap and relative lesion volume error. The proposed model inputs only the FLAIR modality, which simplifies preprocessing pipelines, eliminates the need for inter-modality registration errors and harmonization, which can introduce variability. Source code is available on https://github.com/Pubec/nnunetv2-unlearning.","url_abs":"https://arxiv.org/abs/2603.13328","url_pdf":"https://arxiv.org/pdf/2603.13328","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":{"syntology_url":"https://syntology.ai/paper/2603.13328","atlas_url":"https://app.syntology.ai/?focus=2603.13328","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2603.13328"}},"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":[{"mentioned_in_github":null,"is_official":null,"provenance":"deterministic:regex_extraction","mentioned_in_paper":null,"url":"https://github.com/Pubec/nnunetv2-unlearning","reach":null}],"summary":{"ran":6},"by_repo_kind":{"found_in_text":{"samples":6,"ran":6,"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":"a2c8b38ba9393780","entry":"convert_labels_back_to_BraTS","repo":"Pubec/nnunetv2-unlearning","repo_kind":"found_in_text","path":"nnunetv2/dataset_conversion/Dataset137_BraTS21.py","file_url":"https://github.com/Pubec/nnunetv2-unlearning/blob/HEAD/nnunetv2/dataset_conversion/Dataset137_BraTS21.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"a2c8b38ba9393780"}},{"code_sha256_prefix":"517ad51ee69e8a2d","entry":"copy_files","repo":"Pubec/nnunetv2-unlearning","repo_kind":"found_in_text","path":"nnunetv2/dataset_conversion/Dataset027_ACDC.py","file_url":"https://github.com/Pubec/nnunetv2-unlearning/blob/HEAD/nnunetv2/dataset_conversion/Dataset027_ACDC.py","link_basis":"harvester_set","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"517ad51ee69e8a2d"}},{"code_sha256_prefix":"c66e8feb67225a1b","entry":"copy_files","repo":"Pubec/nnunetv2-unlearning","repo_kind":"found_in_text","path":"nnunetv2/dataset_conversion/Dataset115_EMIDEC.py","file_url":"https://github.com/Pubec/nnunetv2-unlearning/blob/HEAD/nnunetv2/dataset_conversion/Dataset115_EMIDEC.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"c66e8feb67225a1b"}},{"code_sha256_prefix":"2a83a2bc5120febe","entry":"get_vendor_split","repo":"Pubec/nnunetv2-unlearning","repo_kind":"found_in_text","path":"nnunetv2/dataset_conversion/Dataset114_MNMs.py","file_url":"https://github.com/Pubec/nnunetv2-unlearning/blob/HEAD/nnunetv2/dataset_conversion/Dataset114_MNMs.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"2a83a2bc5120febe"}},{"code_sha256_prefix":"86ec2ccc2984b8bf","entry":"merge","repo":"Pubec/nnunetv2-unlearning","repo_kind":"found_in_text","path":"nnunetv2/batch_running/generate_lsf_runs_customDecathlon.py","file_url":"https://github.com/Pubec/nnunetv2-unlearning/blob/HEAD/nnunetv2/batch_running/generate_lsf_runs_customDecathlon.py","link_basis":"harvester_set","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"86ec2ccc2984b8bf"}},{"code_sha256_prefix":"882441705cf75b1d","entry":"read_csv","repo":"Pubec/nnunetv2-unlearning","repo_kind":"found_in_text","path":"nnunetv2/dataset_conversion/Dataset114_MNMs.py","file_url":"https://github.com/Pubec/nnunetv2-unlearning/blob/HEAD/nnunetv2/dataset_conversion/Dataset114_MNMs.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"882441705cf75b1d"}}]},"arxiv_metadata":{"licence":"arXiv metadata, CC0 1.0 (https://info.arxiv.org/help/license)","fields":["title","abstract","authors","date"],"primary_category":"cs.LG","source":"arxiv_2026.jsonl"},"syntology_extracted_results":null}