HyperAIHyperAI

Command Palette

Search for a command to run...

il y a 4 mois

HSI-BERT : Classification d'images hyperspectrales à l'aide de la représentation bidirectionnelle encodée par les Transformers

{Wei Li Mengmeng Zhang HongWei Yang Lina Zhao Ji He}

Résumé

Les méthodes d’apprentissage profond ont été largement utilisées dans la classification des images hyperspectrales et ont atteint des performances de pointe. Toutefois, les méthodes existantes basées sur l’apprentissage profond sont limitées par un champ réceptif restreint, une flexibilité insuffisante et des difficultés de généralisation dans le contexte de la classification des images hyperspectrales. Pour résoudre ces problèmes, nous proposons HSI-BERT, où BERT signifie bidirectional encoder representations from transformers et HSI désigne les images hyperspectrales. Le modèle HSI-BERT dispose d’un champ réceptif global, permettant de capturer les dépendances globales entre les pixels, indépendamment de leur distance spatiale. HSI-BERT est très flexible et permet des régions d’entrée flexibles et dynamiques. En outre, HSI-BERT présente une bonne capacité de généralisation, car le modèle entraîné conjointement peut être appliqué à des régions de formes variées sans nécessiter de réentraînement. HSI-BERT repose principalement sur un mécanisme d’attention auto-attention multi-têtes (MHSA) dans une couche MHSA. Plusieurs types d’attention sont appris par les différentes têtes, et chaque tête de la couche MHSA encode une représentation contextuelle sémantique pour extraire des caractéristiques discriminantes. En fusionnant toutes les caractéristiques encodées par les têtes, les représentations finales intègrent des informations spatiales et spectrales essentielles pour une classification précise au niveau des pixels. Des résultats quantitatifs et qualitatifs montrent que HSI-BERT surpassent tous les modèles basés sur les réseaux de neurones convolutifs (CNN) en termes de précision de classification et de temps de calcul, et atteignent des performances de pointe sur trois jeux de données d’images hyperspectrales largement utilisés.

Benchmarks

BenchmarkMéthodologieMétriques
hyperspectral-image-classification-on-indianHSI-BERT
OA@15perclass: 58.50±1.56
hyperspectral-image-classification-on-kennedyHSI-BERT
OA@15perclass: 82.93±0.94
hyperspectral-image-classification-on-paviaHSI-BERT
OA@15perclass: 75.31±1.59

Construire l'IA avec l'IA

De l'idée au lancement — accélérez votre développement IA avec du co-codage IA gratuit, un environnement prêt à l'emploi et les meilleurs prix GPU.

Co-codage IA
GPU prêts à utiliser
Meilleurs prix
Commencer

Hyper Newsletters

Abonnez-vous à nos dernières mises à jour
Nous vous enverrons les dernières mises à jour de la semaine dans votre boîte de réception à neuf heures chaque lundi matin
Propulsé par MailChimp
HSI-BERT : Classification d'images hyperspectrales à l'aide de la représentation bidirectionnelle encodée par les Transformers | Articles de recherche | HyperAI