Hyperspectral Smoke Segmentation via Mixture of Prototypes
Lujian Yao, Haitao Zhao, Xianghai Kong, Yuhan Xu
TL;DR
The paper tackles smoke segmentation under challenging cloud and semi-transparent conditions by leveraging hyperspectral imaging. It introduces the MoP network, which preserves per-band spectral signatures via band split, models diverse spectral patterns with band-level prototypes, and adaptively weights bands through a dual-level router, achieving superior performance on the HSSDataset and generalizing to MSSDataset. The work provides a first hyperspectral smoke segmentation dataset with a Many-to-One annotation protocol and demonstrates competitive results against strong baselines in both hyperspectral and multispectral settings. This approach offers a path toward more reliable early-warning systems for wildfire management and industrial safety by exploiting rich spectral information.
Abstract
Smoke segmentation is critical for wildfire management and industrial safety applications. Traditional visible-light-based methods face limitations due to insufficient spectral information, particularly struggling with cloud interference and semi-transparent smoke regions. To address these challenges, we introduce hyperspectral imaging for smoke segmentation and present the first hyperspectral smoke segmentation dataset (HSSDataset) with carefully annotated samples collected from over 18,000 frames across 20 real-world scenarios using a Many-to-One annotations protocol. However, different spectral bands exhibit varying discriminative capabilities across spatial regions, necessitating adaptive band weighting strategies. We decompose this into three technical challenges: spectral interaction contamination, limited spectral pattern modeling, and complex weighting router problems. We propose a mixture of prototypes (MoP) network with: (1) Band split for spectral isolation, (2) Prototype-based spectral representation for diverse patterns, and (3) Dual-level router for adaptive spatial-aware band weighting. We further construct a multispectral dataset (MSSDataset) with RGB-infrared images. Extensive experiments validate superior performance across both hyperspectral and multispectral modalities, establishing a new paradigm for spectral-based smoke segmentation.
