Lune

ICML2025Top-tier venue

MaskTwins: Dual-form Complementary Masking for Domain-Adaptive Image Segmentation

Jiawen Wang, Yinda Chen, Xiaoyu Liu, Che Liu, Dong Liu, Jianqing Gao, Zhiwei Xiong

Abstract

Recent works have correlated Masked Image Modeling (MIM) with consistency regularization in Unsupervised Domain Adaptation (UDA). However, they merely treat masking as a special form of deformation on the input images and neglect the theoretical analysis, which leads to a superficial understanding of masked reconstruction and insufficient exploitation of its potential in enhancing feature extraction and representation learning. In this paper, we reframe masked reconstruction as a sparse signal reconstruction problem and theoretically prove that the dual form of complementary masks possesses superior capabilities in extracting domain-agnostic image features. Based on this compelling insight, we propose MaskTwins, a simple yet effective UDA framework that integrates masked reconstruction directly into the main training pipeline. MaskTwins uncovers intrinsic structural patterns that persist across disparate domains by enforcing consistency between predictions of images masked in complementary ways, enabling domain generalization in an end-to-end manner. Extensive experiments verify the superiority of MaskTwins over baseline methods in natural and biological image segmentation. These results demonstrate the significant advantages of MaskTwins in extracting domain-invariant features without the need for separate pre-training, offering a new paradigm for domain-adaptive segmentation. The source code is available at https://github.com/jwwang0421/masktwins .

Ask Lune

Ask anything. The answer quotes the paper.

Lune reads all of it and the papers around it, then shows you the passages behind every answer.

Questions to start from

Lune answers from

the full paper · 2 top-tier papers that cite it · related work

Get the answer

Free to start. No credit card required.

Cited by top-tier papers2

  1. DVP-MVS: Synergize Depth-Edge and Visibility Prior for Multi-View StereoZhenlong Yuan, Jinguo Luo, Fei Shen, Zhaoxin Li et al.AAAI 2025 · 19 citations
  2. Johnson-Lindenstrauss Lemma Guided Network for Efficient 3D Medical SegmentationJinpeng Lu, Linghan Cai, Yinda Chen, Guo Tang et al.ICLR 2026

Builds on28

  1. An Image is Worth 16x16 Words: Transformers for Image Recognition at ScaleAlexey Dosovitskiy, Lucas Beyer, Alexander Kolesnikov, Dirk Weissenborn et al.ICLR 2021 · 21,477 citations
  2. SegFormer: Simple and Efficient Design for Semantic Segmentation with TransformersEnze Xie, Wenhai Wang, Zhiding Yu, Anima Anandkumar et al.NeurIPS 2021 · 9,661 citations
  3. BEiT: BERT Pre-Training of Image TransformersHangbo Bao, Li Dong, Songhao Piao, Furu WeiICLR 2022 · 3,632 citations
  4. SimMIM: a Simple Framework for Masked Image ModelingZhenda Xie, Zheng Zhang, Yue Cao, Yutong Lin et al.CVPR 2022 · 1,129 citations
  5. Confidence Regularized Self-TrainingYang Zou, Zhiding Yu, Xiaofeng Liu, B. V. K. Vijaya Kumar et al.ICCV 2019 · 901 citations

Related papers

  1. CDEA: Context- and Detail-Enhanced Unsupervised Learning for Domain Adaptive Semantic SegmentationShuyuan Wen, Bingrui Hu, Wenchao LiACM MM 2024 · 4 citations
  2. Single-Domain Generalization in Medical Image Segmentation via Test-Time Adaptation from Shape DictionaryQuande Liu, Cheng Chen, Qi Dou, Pheng-Ann HengAAAI 2022 · 50 citations
  3. Reconstruction Target Matters in Masked Image Modeling for Cross-Domain Few-Shot LearningRan Ma, Yixiong Zou, Yuhua Li, Ruixuan LiAAAI 2025 · 2 citations
  4. SemiGDA: Generative Dual-distribution Alignment for Semi-Supervised Medical Image Segmentationkaiwen Huang, Yi Zhou, Yizhe Zhang, Jingxiong Li et al.CVPR 2026
  5. Architecture-Agnostic Masked Image Modeling - From ViT back to CNNSiyuan Li, Di Wu, Fang Wu, Zelin Zang et al.ICML 2023 · 60 citations
Dusk over the sea between two cliffs drawn in fine vertical lines