BCG-Former: Toward Pareto-Efficient Hyperspectral Image Classification via Band-Contextual Gating

· Source: Computer Vision and Pattern Recognition · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision · Depth: Expert, quick

Summary

BCG-Former is a lightweight CNN-Transformer hybrid designed for Pareto-efficient hyperspectral image (HSI) classification, addressing the need for both high accuracy and strict computational efficiency on platforms like UAVs and small spaceborne sensors. The model integrates three innovations: Band-Contextual Gating (BCG) for adaptive spectral recalibration, a spectral summary token to bridge spectral and spatial features, and single-pass Band-RoPE with linear attention for efficient joint representation learning. Evaluated on eight airborne and UAV-borne benchmark datasets, BCG-Former achieved overall accuracies from 91.51% on Houston 2018 to 99.49% on Houston 2013. It maintains sub-millisecond inference latency (0.91-0.95ms) and uses only 0.10-0.23M parameters, consistently residing on or near the Pareto frontier of accuracy versus latency, outperforming or matching recent CNN-, Transformer-, and Mamba-based methods at a fraction of their computational cost.

Key takeaway

For Machine Learning Engineers developing hyperspectral image classification systems for resource-constrained platforms, BCG-Former offers a compelling solution. You should evaluate its hybrid CNN-Transformer architecture, particularly its Band-Contextual Gating, to achieve high accuracy (up to 99.49%) while maintaining sub-millisecond inference latency (0.91-0.95ms) and low parameter counts (0.10-0.23M). This approach can significantly improve the feasibility of real-time and large-scale remote sensing applications.

Key insights

BCG-Former achieves Pareto-efficient HSI classification by balancing high accuracy with sub-millisecond latency and minimal parameters via novel architectural components.

Principles

Method

The article describes BCG-Former's architecture, which combines Band-Contextual Gating, a spectral summary token, and single-pass Band-RoPE with linear attention for joint spectral-spatial learning.

In practice

Topics

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, Computer Vision Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computer Vision and Pattern Recognition.