Rethinking Box Supervision: Bias-Free Weakly Supervised Medical Segmentation
Jun Wei, Hui Huang
Abstract
Pixel-level annotations for medical image segmentation are costly and labor-intensive, often requiring expert knowledge. Bounding box labels provide a more scalable alternative but introduce strong box-shaped bias that hampers segmentation quality. We propose WeakMed, a general-purpose weakly supervised segmentation framework that removes the dependence on pixel-level masks while overcoming the structural limitations of box supervision. WeakMed introduces two lightweight, plug-and-play training components: (1) a Mask-to-Box (M2B) transformation that aligns predicted masks with box annotations to reduce label mismatch and box-induced bias, and (2) a Scale Consistency (SC) loss that enforces multi-scale self-supervision to address the ambiguity and instability of weak labels. Both modules are used only during training and impose no inference overhead. Across 9 segmentation tasks, 10 datasets, and 6 imaging modalities, WeakMed consistently surpasses existing weakly supervised methods and achieves performance competitive with fully supervised baselines. These results demonstrate its practicality as a low-cost yet high-quality solution for medical image segmentation. Codes will be released.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 121bed02-bfb8-4aec-a98d-509bd7fa2165Builds on8
- UniverSeg: Universal Medical Image SegmentationVictor Ion Butoi, Jose Javier Gonzalez Ortiz, Tianyu Ma, Mert R. Sabuncu et al.ICCV 2023 · 163 citations
- Pointly-Supervised Instance SegmentationBowen Cheng, Omkar Parkhi, Alexander KirillovCVPR 2022 · 140 citations
- C-CAM: Causal CAM for Weakly Supervised Semantic Segmentation on Medical ImageZhang Chen, Zhiqiang Tian, Jihua Zhu, Ce Li et al.CVPR 2022 · 116 citations
- DiscoBox: Weakly Supervised Instance Segmentation and Semantic Correspondence from Box SupervisionShiyi Lan, Zhiding Yu, Christopher B. Choy, Subhashree Radhakrishnan et al.ICCV 2021 · 97 citations
- CycleMix: A Holistic Strategy for Medical Image Segmentation from Scribble SupervisionKe Zhang, Xiahai ZhuangCVPR 2022 · 88 citations
Related papers
- WeakPCSOD: Overcoming the Bias of Box Annotations for Weakly Supervised Point Cloud Salient Object DetectionJun Wei, S. Kevin Zhou, Shuguang Cui, Zhen LiAAAI 2024 · 6 citations
- LooBox: Loose-box-supervised 3D Tumor Segmentation with Self-correcting Bidirectional LearningTianzhong Lan, Zhang Yi, Xiuyuan Xu, Min ZhuACM MM 2025 · 2 citations
- SemiGDA: Generative Dual-distribution Alignment for Semi-Supervised Medical Image Segmentationkaiwen Huang, Yi Zhou, Yizhe Zhang, Jingxiong Li et al.CVPR 2026
- Single-Stage Semantic Segmentation From Image LabelsNikita Araslanov, Stefan RothCVPR 2020
- MonoBox: Tightness-Free Box-Supervised Polyp Segmentation Using Monotonicity ConstraintQiang Hu, Zhenyu Yi, Ying Zhou, Fan Huang et al.AAAI 2025 · 8 citations
