Position-Aware WaveMLP in Polar Coordinate System for Hyperspectral Image Classification

Yuhang Wang,Meng Zhang,Ping Tang
DOI: https://doi.org/10.1109/lgrs.2024.3362361
IF: 5.343
2024-01-01
IEEE Geoscience and Remote Sensing Letters
Abstract:WaveMLP has demonstrated remarkable performance in various vision tasks; such as dense feature detection and semantic segmentation. However, WaveMLP, as a local model, imposes limitations on fully connected layers by only allowing connections between tokens within the same local window. This constraint makes the model neglect the relationship among tokens in different windows, leading to a local token fusion and a degraded modelling performance. Specially, it poses challenges when dealing with hyperspectral image (HSI) classification tasks that require capturing long-range dependencies. To address this issue, this letter proposes a new Position-Aware WaveMLP, dubbed PA-WaveMLP, which incorporates a global polar positional encoding module (PPEM) into WaveMLP. PPEM is a light-weight method to encode the spatial relationship between land objects in distance and direction by using the radius and angle. By PPEM, the proposed PA-WaveMLP enables tokens to include their own spatial position information to the fusion process, allowing for the capture of long-range dependencies while maintaining the excellent modeling capabilities of WaveMLP. The experimental results on three publicly available HSI datasets validate the effectiveness and generalizability of this newly proposed PA-WaveMLP. In particular, PA-WaveMLP model achieved an overall accuracy of 99.16%, 99.71%, and 99.47% on Indian Pines,Pavia University, and Salinas respectively.
imaging science & photographic technology,remote sensing,engineering, electrical & electronic,geochemistry & geophysics
What problem does this paper attempt to address?