HyperAIHyperAI

Command Palette

Search for a command to run...

长尾多标签图像分类的概率引导损失

Dekun Lin

摘要

近年来,长尾学习(long-tailed learning)受到了越来越多的关注。长尾多标签图像分类作为其一个重要子任务,仍面临巨大挑战且研究相对不足。本文从概率视角提出了一种全新的解决方案。具体而言,我们发现现有的针对长尾多标签分类的代价敏感学习方法在训练过程中会不同程度地影响正负标签的预测概率,而不同的概率演化过程又将直接影响最终的模型性能。为此,我们提出了一种概率引导的损失函数(probability-guided loss),该损失函数包含两个组成部分,用于调控这一概率演化过程:其一是概率再平衡机制,可灵活调节训练过程中概率的分布;其二是自适应概率感知焦点损失(adaptive probability-aware focal loss),能够进一步缩小正负标签之间的概率差距。我们在两个长尾多标签图像分类数据集(VOC-LT 和 COCO-LT)上进行了大量实验,结果充分验证了所提方法的合理性与优越性。


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供