Skip to yearly menu bar Skip to main content


Poster

Automated Label Unification for Multi-Dataset Semantic Segmentation with GNNs

Ma Rong · Jie Chen · Xiangyang Xue · Jian Pu

[ ]
Wed 11 Dec 11 a.m. PST — 2 p.m. PST

Abstract:

Deep supervised models possess significant capability to assimilate extensive training data, thereby presenting an opportunity to enhance model performance through training on multiple datasets. However, conflicts arising from different label spaces among datasets may adversely affect model performance. In this paper, we propose a novel approach to automatically construct a unified label space across multiple datasets using graph neural networks. This enables semantic segmentation models to be trained simultaneously on multiple datasets, resulting in performance improvements. Unlike existing methods, our approach facilitates seamless training without the need for additional manual reannotation or taxonomy reconciliation. This significantly enhances the efficiency and effectiveness of multi-dataset segmentation model training. The results demonstrate that our method significantly outperforms other multi-dataset training methods when trained on seven datasets simultaneously, and achieves state-of-the-art performance on the WildDash 2 benchmark.

Live content is unavailable. Log in and register to view live content