Self-Ensembling GAN for Cross-Domain Semantic Segmentation

被引:3
|
作者
Xu, Yonghao [1 ,2 ]
He, Fengxiang [3 ]
Du, Bo [4 ,5 ]
Tao, Dacheng [3 ]
Zhang, Liangpei [1 ]
机构
[1] Wuhan Univ, State Key Lab Informat Engn Surveying Mapping & R, Wuhan 430079, Peoples R China
[2] Inst Adv Res Artificial Intelligence IARAI, A-1030 Vienna, Austria
[3] JDcom Inc, JD Explore Acad, Beijing, Peoples R China
[4] Wuhan Univ, Sch Comp Sci, Natl Engn Res Ctr Multimedia Software, Inst Artificial Intelligence, Wuhan 430079, Peoples R China
[5] Wuhan Univ, Hubei Key Lab Multimedia & Network Commun Engn, Wuhan 430079, Peoples R China
关键词
Deep learning; domain adaptation; semantic segmentation; adversarial learning; DEEP NEURAL-NETWORK;
D O I
10.1109/TMM.2022.3229976
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Deep neural networks (DNNs) have greatly contributed to the performance gains in semantic segmentation. Nevertheless, training DNNs generally requires large amounts of pixel-level labeled data, which is expensive and time-consuming to collect in practice. To mitigate the annotation burden, this paper proposes a self-ensembling generative adversarial network (SE-GAN) exploiting cross-domain data for semantic segmentation. In SE-GAN, a teacher network and a student network constitute a self-ensembling model for generating semantic segmentation maps, which together with a discriminator, forms a GAN. Despite its simplicity, we find SE-GAN can significantly boost the performance of adversarial training and enhance the stability of the model, the latter of which is a common barrier shared by most adversarial training-based methods. We theoretically analyze SE-GAN and provide an O(1/root N) generalization bound (N is the training sample size), which suggests controlling the discriminator's hypothesis complexity to enhance the generalizability. Accordingly, we choose a simple network as the discriminator. Extensive and systematic experiments in two standard settings demonstrate that the proposed method significantly outperforms current state-of-the-art approaches.
引用
收藏
页码:7837 / 7850
页数:14
相关论文
共 50 条
  • [1] Self-Ensembling with GAN-based Data Augmentation for Domain Adaptation in Semantic Segmentation
    Choi, Jaehoon
    Kim, Taekyung
    Kim, Changick
    2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, : 6829 - 6839
  • [2] Self-Ensembling Attention Networks: Addressing Domain Shift for Semantic Segmentation
    Xu, Yonghao
    Du, Bo
    Zhang, Lefei
    Zhang, Qian
    Wang, Guoli
    Zhang, Liangpei
    THIRTY-THIRD AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE / THIRTY-FIRST INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE CONFERENCE / NINTH AAAI SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2019, : 5581 - 5588
  • [3] Domain Adaptation Semantic Segmentation for Urban Scene Combining Self-ensembling and Adversarial Learning
    Zhang G.
    Lu F.
    Long B.
    Miao J.
    Moshi Shibie yu Rengong Zhineng/Pattern Recognition and Artificial Intelligence, 2021, 34 (01): : 58 - 67
  • [4] Unsupervised domain adaptation for medical imaging segmentation with self-ensembling
    Perone, Christian S.
    Ballester, Pedro
    Barros, Rodrigo C.
    Cohen-Adad, Julien
    NEUROIMAGE, 2019, 194 : 1 - 11
  • [5] Unsupervised Domain Adaptation for the Histopathological Cell Segmentation through Self-Ensembling
    Li, Chaoqun
    Zhou, Yitian
    Shi, Tangqi
    Wu, Yenan
    Yang, Meng
    Li, Zhongyu
    MICCAI WORKSHOP ON COMPUTATIONAL PATHOLOGY, VOL 156, 2021, 156 : 151 - 158
  • [6] Self-ensembling with mask-boundary domain adaptation for optic disc and cup segmentation
    He, Yanlin
    Kong, Jun
    Liu, Di
    Li, Juan
    Zheng, Caixia
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2024, 129
  • [7] Cross-Domain Grouping and Alignment for Domain Adaptive Semantic Segmentation
    Kim, Minsu
    Joung, Sunghun
    Kim, Seungryong
    Park, Jungin
    Kim, Ig-Jae
    Sohn, Kwanghoon
    THIRTY-FIFTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, THIRTY-THIRD CONFERENCE ON INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE AND THE ELEVENTH SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2021, 35 : 1799 - 1807
  • [8] Cross-Domain Few-Shot Semantic Segmentation
    Lei, Shuo
    Zhang, Xuchao
    He, Jianfeng
    Chen, Fanglan
    Du, Bowen
    Lu, Chang-Tien
    COMPUTER VISION - ECCV 2022, PT XXX, 2022, 13690 : 73 - 90
  • [9] A global reweighting approach for cross-domain semantic segmentation
    Zhang, Yuhang
    Tian, Shishun
    Liao, Muxin
    Hua, Guoguang
    Zou, Wenbin
    Xu, Chen
    Signal Processing: Image Communication, 2025, 130
  • [10] TACS: Taxonomy Adaptive Cross-Domain Semantic Segmentation
    Gong, Rui
    Danelljan, Martin
    Dai, Dengxin
    Paudel, Danda Pani
    Chhatkuli, Ajad
    Yu, Fisher
    Van Gool, Luc
    COMPUTER VISION, ECCV 2022, PT XXXIV, 2022, 13694 : 19 - 35