Apr, 2024

类相似度转换:从广义少样本分割中解耦类相似度和不平衡

TL;DR在广义 Few-shot 分割(GFSS)中,通过使用大量基本类别样本对模型进行训练,再对有限的新类别样本进行调整。本文主要关注基本类别和新类别之间的相关性,并在两个方面对 GFSS 进行改进:1)利用基本类别与新类别之间的相似性促进新类别的学习;2)通过缓解支持集和训练集之间的体积差异引起的类别不平衡问题。通过提出相似性转换矩阵指导基于基本类别的知识学习新类别,以及利用 Label-Distribution-Aware Margin(LDAM)损失和 Transductive Inference 来解决类别不平衡和过拟合支持集的问题。此外,通过扩展概率转换矩阵,该方法可以在学习新类别时减轻基本类别的灾难性遗忘。我们通过在 OpenEarthMap 的适应版本上验证了我们的方法。与现有的 GFSS 基线相比,我们的方法在 OpenEarthMap Land Cover Mapping Few-Shot Challenge 中的排名从 3%到 7%不等,并在本文完成时排名第二。