Self-Ensembling GAN for Cross-Domain Semantic Segmentation

被引:7
作者
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 条
  • [31] ResiDualGAN: Resize-Residual DualGAN for Cross-Domain Remote Sensing Images Semantic Segmentation
    Zhao, Yang
    Guo, Peng
    Sun, Zihao
    Chen, Xiuwan
    Gao, Han
    REMOTE SENSING, 2023, 15 (05)
  • [32] Joint alignment of the distribution in input and feature space for cross-domain aerial image semantic segmentation
    Chen, Zhe
    Yang, Bisheng
    Ma, Ailong
    Peng, Mingjun
    Li, Haiting
    Chen, Tao
    Chen, Chi
    Dong, Zhen
    INTERNATIONAL JOURNAL OF APPLIED EARTH OBSERVATION AND GEOINFORMATION, 2022, 115
  • [33] Feature Re-Representation and Reliable Pseudo Label Retraining for Cross-Domain Semantic Segmentation
    Li, Jing
    Zhou, Kang
    Qian, Shenhan
    Li, Wen
    Duan, Lixin
    Gao, Shenghua
    IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2024, 46 (03) : 1682 - 1694
  • [34] Cross-Domain Incremental Feature Learning for ALS Point Cloud Semantic Segmentation With Few Samples
    Dai, Mofan
    Xing, Shuai
    Xu, Qing
    Li, Pengcheng
    Pan, Jiechen
    Wang, Hanyun
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2025, 63
  • [35] Active Bidirectional Self-Training Network for Cross-Domain Segmentation in Remote-Sensing Images
    Yang, Zhujun
    Yan, Zhiyuan
    Diao, Wenhui
    Ma, Yihang
    Li, Xinming
    Sun, Xian
    REMOTE SENSING, 2024, 16 (13)
  • [36] Cross-domain Semantic Feature Learning via Adversarial Adaptation Networks
    Li, Rui
    Cao, Wenming
    Qian, Sheng
    Wong, Hau-San
    Wu, Si
    2018 24TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2018, : 37 - 42
  • [37] Cross-Scanner Low-Dose Brain-PET Image Noise Reduction With Self-Ensembling
    Wang, Jiale
    Guo, Rui
    Miao, Ying
    Xue, Song
    Zhang, Yu
    Shi, Kuangyu
    Zheng, Guoyan
    Li, Biao
    IEEE TRANSACTIONS ON RADIATION AND PLASMA MEDICAL SCIENCES, 2024, 8 (04) : 391 - 401
  • [38] Enhanced Target Domain Representation Based Unsupervised Cross-Domain Medical Image Segmentation
    Liu, Kai
    Lu, Runuo
    Zheng, Xiaorou
    Dong, Shoubin
    Moshi Shibie yu Rengong Zhineng/Pattern Recognition and Artificial Intelligence, 2024, 37 (09): : 755 - 769
  • [39] Cross-Domain Echocardiography Segmentation with Multi-Space Joint Adaptation
    Chen, Tongwaner
    Xia, Menghua
    Huang, Yi
    Jiao, Jing
    Wang, Yuanyuan
    SENSORS, 2023, 23 (03)
  • [40] SAFENet: Semantic-Aware Feature Enhancement Network for unsupervised cross-domain road scene segmentation
    Ren, Dexin
    Li, Minxian
    Wang, Shidong
    Ren, Mingwu
    Zhang, Haofeng
    IMAGE AND VISION COMPUTING, 2024, 152