HyperAIHyperAI

Command Palette

Search for a command to run...

A Strong Baseline for Generalized Few-Shot Semantic Segmentation

Sina Hajimiri* Malik Boudiaf Ismail Ben Ayed Jose Dolz

Abstract

This paper introduces a generalized few-shot segmentation framework with astraightforward training process and an easy-to-optimize inference phase. Inparticular, we propose a simple yet effective model based on the well-knownInfoMax principle, where the Mutual Information (MI) between the learnedfeature representations and their corresponding predictions is maximized. Inaddition, the terms derived from our MI-based formulation are coupled with aknowledge distillation term to retain the knowledge on base classes. With asimple training process, our inference model can be applied on top of anysegmentation network trained on base classes. The proposed inference yieldssubstantial improvements on the popular few-shot segmentation benchmarks,PASCAL-5i5^i5i and COCO-20i20^i20i. Particularly, for novel classes, the improvementgains range from 7% to 26% (PASCAL-5i5^i5i) and from 3% to 12% (COCO-20i20^i20i) inthe 1-shot and 5-shot scenarios, respectively. Furthermore, we propose a morechallenging setting, where performance gaps are further exacerbated. Our codeis publicly available at https://github.com/sinahmr/DIaM.


Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing

HyperAI Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
A Strong Baseline for Generalized Few-Shot Semantic Segmentation | Papers | HyperAI