Searching sharing relationship for instance segmentation decoder

doi:10.1007/s10489-022-04434-y

Searching sharing relationship for instance segmentation decoder

2023-05-02

1

PRE

AI

DOI：10.1007/s10489-022-04434-y

Original

Original request for help

Save

Abstract

En 中文

Instance segmentation is a typical visual task that requires per-pixel mask prediction with a category label for each instance. For the decoder in instance segmentation network, parallel branches or towers are commonly adopted to deal with instance- and dense-level predictions. However, this parallelism ignores inter-branch and inner-branch relationships. Besides, how the different branches are connected is unclear, which is difficult to explore manually in practice. To address the above issues, we introduce Neural Architecture Search (NAS) to automatically search for hardware and memory-friendly feature sharing branch. Concretely, applying to instance segmentation, we design a search space considering both operations and sharing connections of parallel branches. Through a tailored reinforcement learning(RL) paradigm, we can efficiently search multiple architectures with different shared patterns and tap more feature selection possibilities. Our method is generically useful and can be transferred to analogous multi-task networks. The searched architecture shares features in the middle of the head branches and utilizes instance-level head features to generate pixel-level predictions. Extensive experiments demonstrate the effectiveness and surpass classical parallel decoder networks, exceeding BlendMask by 1.2% on bounding box mAP and 0.9% on segmentation mAP.

Keywords:

Instance segmentation

Neural architecture search

Reinforcement learning

Feature sharing

Searching sharing relationship for instance segmentation decoder

Abstract

Journal

Organization