Daily Arxiv

Cette page résume et organise les publications en intelligence artificielle du monde entier.
Les contenus sont synthétisés grâce à Google Gemini et le service est proposé à but non lucratif.
Les droits d'auteur des articles appartiennent à leurs auteurs ou institutions respectives ; en cas de partage, il suffit d'en mentionner la source.

ProtoECGNet : Apprentissage profond interprétable basé sur des cas pour la classification d'ECG multi-étiquettes avec apprentissage contrastif

Created by
  • Haebom

Auteur

Sahil Sethi, David Chen, Thomas Statchen, Michael C. Burkhart, Nipun Bhandari, Bashar Ramadan, Brett Beaulieu-Jones

Contour

Cet article présente ProtoECGNet, un modèle d'apprentissage profond multi-étiquettes interprétable pour la classification des électrocardiogrammes (ECG). ProtoECGNet utilise l'inférence basée sur des prototypes, qui fonde les décisions sur la similarité avec les représentations apprises de segments d'ECG réels, afin de fournir des explications transparentes et fiables basées sur des cas. Il utilise une architecture multi-branches structurée qui reflète les flux d'interprétation clinique, intégrant un CNN 1D avec des prototypes globaux pour la classification du rythme, un CNN 2D avec des prototypes temporellement localisés pour l'inférence basée sur la morphologie, et un CNN 2D avec des prototypes globaux pour les anomalies de diffusion. Chaque branche est entraînée avec une perte de prototype conçue pour l'apprentissage multi-étiquettes et combine une nouvelle perte contrastive qui favorise le clustering, la séparation, la diversité et la séparation appropriée entre les prototypes de classes non apparentées. Nous évaluons ProtoECGNet sur les 71 étiquettes diagnostiques de l'ensemble de données PTB-XL, démontrant qu'il est compétitif par rapport aux modèles boîte noire de pointe tout en fournissant des explications structurées basées sur des cas. La représentativité et la clarté des prototypes ont été vérifiées par des évaluations cliniques structurées. ProtoECGNet démontre que l'apprentissage par prototypes peut s'adapter efficacement à la classification complexe de séries chronologiques multi-étiquettes, offrant ainsi une voie pratique vers des modèles d'apprentissage profond transparents et fiables pour l'aide à la décision clinique.

Takeaways, Limitations_

Takeaways:
Nous démontrons la faisabilité d’atteindre à la fois l’interprétabilité et la performance dans la classification des électrocardiogrammes en utilisant un modèle d’apprentissage profond basé sur un prototype.
Présentation d'une méthode d'apprentissage de prototype efficace pour les problèmes de classification multi-étiquettes.
Valider la fiabilité du prototype par une revue clinique.
Une approche pratique pour développer un système d’aide au diagnostic de l’électrocardiogramme.
Limitations:
Seule la validation des performances a été effectuée sur l'ensemble de données PTB-XL, des recherches supplémentaires sur les performances de généralisation sont donc nécessaires.
L'interprétabilité d'un prototype repose sur l'évaluation subjective des cliniciens. Il est donc nécessaire d'élaborer des critères d'évaluation objectifs.
La complexité du modèle peut être élevée, ce qui entraîne des coûts de calcul élevés.
Une validation supplémentaire est nécessaire dans divers ensembles de données ECG et contextes cliniques.
👍