Classification Task-Driven Hyperspectral Band Selection via Interpretability From XGBoost

Abstract

Band selection (BS) identifies key bands from hyperspectral imagery (HSI) for specific downstream tasks, playing a pivotal role in practical applications. eXtreme Gradient Boosting (XGBoost), an interpretable tree-based ensemble learning classifier, explicitly implements the complex nonlinear hyperspectral classification. The interpretable information extracted from the tree structure offers a novel basis for supervised BS. To this end, this article proposes a supervised BS method, named classification task-driven hyperspectral BS via interpretability from XGBoost (XGBS). It leverages prior knowledge to train a classification task-driven XGBoost and interprets the tree structure to extract multivariate interpretable information, encompassing band split gain and two types of band dependencies. Subsequently, a heuristic search framework is employed to evaluate band performance, facilitating the selection of bands with strong classification capability, high interdependency, and low redundancy. Experiments conducted on six real HSI datasets demonstrate the effectiveness and stability of the proposed XGBS.

Keywords

• Band dependence
• classification task-driven band selection (BS)
• eXtreme Gradient Boosting (XGBoost)
• hyperspectral imagery (HSI)
• multivariate interpretable information (MII)