HyperAIHyperAI

Command Palette

Search for a command to run...

分布对齐:面向长尾视觉识别的统一框架

Songyang Zhang Zeming Li Shipeng Yan Xuming He Jian Sun

摘要

尽管深度神经网络近年来取得了显著成功,但在视觉识别任务中有效建模长尾类别分布仍然是一个重大挑战。为解决这一问题,我们首先通过消融实验深入分析了两阶段学习框架的性能瓶颈。基于这一发现,我们提出了一种统一的分布对齐策略,用于长尾视觉识别。具体而言,我们设计了一种自适应校准函数,能够针对每个数据点动态调整分类得分。随后,在两阶段学习框架中引入了一种广义重加权方法,以平衡类别先验分布,从而为视觉识别任务中多样化的场景提供一种灵活且统一的解决方案。我们在四个任务上进行了大量实验验证,包括图像分类、语义分割、目标检测和实例分割。实验结果表明,所提方法在所有四项识别任务上均取得了当前最优性能,且仅依赖于一个简单而统一的框架。相关代码与模型将公开发布于:https://github.com/Megvii-BaseDetection/DisAlign


用 AI 构建 AI

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

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

HyperAI Newsletters

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