HyperAIHyperAI

Command Palette

Search for a command to run...

用于少样本分割的密集高斯过程

Joakim Johnander Johan Edstedt Michael Felsberg Fahad Shahbaz Khan Martin Danelljan

摘要

少样本分割是一项具有挑战性的密集预测任务,其目标是在仅提供少量标注的支持图像(support set)的情况下,对新的查询图像进行分割。该任务的核心挑战在于设计一种方法,能够在充分聚合支持集中细节信息的同时,对图像外观和上下文的大幅变化保持鲁棒性。为此,我们提出了一种基于密集高斯过程(Dense Gaussian Process, GP)回归的少样本分割方法。给定支持集,我们的密集高斯过程学习从局部深层图像特征到掩码值之间的映射关系,能够有效捕捉复杂的外观分布特征。此外,该方法还提供了一种理论严谨的不确定性建模机制,该不确定性可作为辅助线索,进一步提升最终分割结果的精度,而最终分割结果由一个卷积神经网络(CNN)解码器生成。不同于传统的单维掩码输出,我们进一步利用该方法的端到端学习能力,学习一个高维输出空间以表示高斯过程。在PASCAL-5i^ii和COCO-20i^ii基准测试中,我们的方法取得了新的最先进性能,在COCO-20i^ii的5-shot设置下实现了+8.4 mIoU的绝对提升。此外,随着支持集规模的增加,本方法的分割质量能够平稳提升,同时展现出优异的跨数据集迁移能力。代码与训练好的模型已开源,地址为:\url{https://github.com/joakimjohnander/dgpnet}


用 AI 构建 AI

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

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

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
用于少样本分割的密集高斯过程 | 论文 | HyperAI超神经