Paper Reading AI Learner

Balanced Learning for Domain Adaptive Semantic Segmentation

2025-12-07 15:21:22
Wangkai Li, Rui Sun, Bohao Liao, Zhaoyang Li, Tianzhu Zhang

Abstract

Unsupervised domain adaptation (UDA) for semantic segmentation aims to transfer knowledge from a labeled source domain to an unlabeled target domain. Despite the effectiveness of self-training techniques in UDA, they struggle to learn each class in a balanced manner due to inherent class imbalance and distribution shift in both data and label space between domains. To address this issue, we propose Balanced Learning for Domain Adaptation (BLDA), a novel approach to directly assess and alleviate class bias without requiring prior knowledge about the distribution shift. First, we identify over-predicted and under-predicted classes by analyzing the distribution of predicted logits. Subsequently, we introduce a post-hoc approach to align the logits distributions across different classes using shared anchor distributions. To further consider the network's need to generate unbiased pseudo-labels during self-training, we estimate logits distributions online and incorporate logits correction terms into the loss function. Moreover, we leverage the resulting cumulative density as domain-shared structural knowledge to connect the source and target domains. Extensive experiments on two standard UDA semantic segmentation benchmarks demonstrate that BLDA consistently improves performance, especially for under-predicted classes, when integrated into various existing methods. Code is available at this https URL.

Abstract (translated)

无监督领域适应(UDA)在语义分割中的目标是将已标记的源域知识转移到未标记的目标域中。尽管自我训练技术在UDA方面非常有效,但由于不同域之间数据和标签空间中存在的类别不平衡及分布偏移问题,它们难以以平衡的方式学习每个类别。为了解决这个问题,我们提出了领域适应的均衡学习(BLDA),这是一种新颖的方法,旨在直接评估并缓解类别的偏差,而无需预先知道分布变化的情况。 首先,通过分析预测分数(logits)的分布来识别过度预测和不足预测的类别。随后,引入了一种事后方法,利用共享锚点分布对不同类别间的logits分布进行校准。此外,在自我训练过程中为了使网络能够生成无偏伪标签,我们在线估计了logits的分布,并将logits修正项纳入损失函数中。最后,利用累积密度作为领域共有的结构化知识来连接源域和目标域。 在两个标准UDA语义分割基准测试中的大量实验表明,在与各种现有方法结合使用时,BLDA能够持续提高性能,特别是在改进不足预测的类别方面表现突出。代码可以在提供的URL上获取。

URL

https://arxiv.org/abs/2512.06886

PDF

https://arxiv.org/pdf/2512.06886.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Time_Series Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot