Nov, 2023

基于主动学习的领域自适应语义分割的类别平衡动态获取

TL;DR领域自适应主动学习通过使用不确定性和多样性两个准则与像素级获取策略相结合,有效提高神经网络在标注效率训练中的性能。然而,目前这种方法出现了类别不平衡问题,导致在更大规模的主动学习预算下性能下降。为缓解这一问题,本文提出了一种新型的主动学习方法,即类别平衡动态获取(CBDA),特别适用于高预算情况。通过增加较小类别的训练样本,改进了少数类别的性能,使模型在 5%、10% 和 20% 预算下的 mIoU 分别提高了 0.6、1.7 和 2.4。此外,对少数类别的关注还使最小类别性能分别提高了 0.5、2.9 和 4.6 IoU。最优模型甚至超过了全监督基准模型,表明相较于完整的真实标注,更平衡的标注对模型是有益的。