本論文介紹了一種名為 WaveMamba 的新型高光譜影像分類方法。WaveMamba 結合了小波轉換和空間光譜 Mamba 架構,旨在提高高光譜影像分類的準確性。
高光譜成像 (HSI) 是一種強大的技術,可以在連續的波長範圍內捕捉豐富的光譜資訊,廣泛應用於各種領域。然而,高光譜影像的高維度特性也為有效分類帶來了挑戰。近年來,深度學習 (DL) 的進步顯著提高了高光譜影像分類的性能,特別是卷積神經網路 (CNNs) 在提取空間和光譜特徵方面表現出色。然而,CNNs 依賴於局部感受野,需要大量的標記數據,且難以利用全局上下文資訊。Transformer 架構的出現進一步提升了高光譜影像分類的性能,它能夠有效地建模長距離依賴關係和全局上下文,但面臨著計算效率和標記數據需求方面的挑戰。
WaveMamba 結合了小波轉換和空間光譜 Mamba 架構的優勢,以提高高光譜影像分類的準確性。該方法首先將高光譜影像分割成重疊的三維區塊,並將每個區塊分解為光譜和空間分量,生成光譜標記和空間標記。然後,利用小波轉換提取多尺度空間光譜特徵,並透過狀態空間架構對這些特徵進行處理,以建模空間光譜關係和時間依賴關係。最後,使用線性分類器對提取的特徵進行分類。
在 University of Houston 和 Pavia University 兩個數據集上進行的實驗表明,WaveMamba 在總體精度 (OA)、平均精度 (AA) 和 Kappa (κ) 指標方面均優於其他先進的深度學習方法,包括基於 CNN 的方法、基於 Transformer 的方法和混合方法。
本論文的主要貢獻如下:
未來的工作可以集中在探索自監督預訓練和進一步的網路優化,以提高 WaveMamba 在數據稀缺環境中的性能。
Egy másik nyelvre
a forrásanyagból
arxiv.org
Mélyebb kérdések