3 个月前

基于多相似性的超关系网络用于少样本分割

基于多相似性的超关系网络用于少样本分割

摘要

少样本语义分割旨在仅利用少量标注样本作为监督信号,识别未见类别中的目标区域。该任务的关键在于建立支持图像(support image)与查询图像(query image)之间稳健的语义关联,并有效防止过拟合。本文提出一种高效的多相似性超关系网络(Multi-similarity Hyperrelation Network, MSHNet),以应对少样本语义分割挑战。在MSHNet中,我们引入了一种新型生成式原型相似性(Generative Prototype Similarity, GPS),该方法与余弦相似性相结合,能够构建支持图像与查询图像之间强大的语义关联。基于全局特征生成的局部原型相似性,与基于局部特征计算的全局余弦相似性在逻辑上互为补充,通过联合使用这两种相似性度量,可更全面地刻画查询图像与支持图像之间的关系。此外,我们在MSHNet中设计了对称融合模块(Symmetric Merging Block, SMB),用于高效融合多层、多样本、多相似性超关系特征。MSHNet基于相似性而非特定类别特征进行建模,具有更强的泛化能力,能够有效缓解过拟合问题。在Pascal-5i和COCO-20i两个基准语义分割数据集上,MSHNet在1-shot与5-shot语义分割任务中均取得了新的最先进性能(state-of-the-art)。

代码仓库

alex-shilei/mshnet
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
few-shot-semantic-segmentation-on-pascal-5i-5MSHNet
Mean IoU: 72.3

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于多相似性的超关系网络用于少样本分割 | 论文 | HyperAI超神经