Daily Arxiv

Esta página recopila y organiza artículos sobre inteligencia artificial publicados en todo el mundo.
La información aquí presentada se resume utilizando Google Gemini y el sitio se gestiona sin fines de lucro.
Los derechos de autor de los artículos pertenecen a sus autores y a las instituciones correspondientes; al compartir el contenido, basta con citar la fuente.

ProtoECGNet: Aprendizaje profundo interpretable basado en casos para la clasificación de ECG de múltiples etiquetas con aprendizaje contrastivo

Created by
  • Haebom

Autor

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

Describir

Este artículo presenta ProtoECGNet, un modelo interpretable de aprendizaje profundo multietiqueta para la clasificación de electrocardiogramas (ECG). ProtoECGNet utiliza inferencia basada en prototipos, que basa las decisiones en la similitud con las representaciones aprendidas de segmentos de ECG reales, para proporcionar explicaciones transparentes y fiables basadas en casos. Emplea una arquitectura estructurada multirama que replica los flujos de trabajo de interpretación clínica, integrando una CNN 1D con prototipos globales para la clasificación del ritmo, una CNN 2D con prototipos localizados temporalmente para la inferencia basada en la morfología y una CNN 2D con prototipos globales para las anomalías de difusión. Cada rama se entrena con una pérdida de prototipo diseñada para el aprendizaje multietiqueta y combina una novedosa pérdida contrastiva que promueve la agrupación, la separación, la diversidad y la correcta separación entre prototipos de clases no relacionadas. Evaluamos ProtoECGNet en las 71 etiquetas diagnósticas del conjunto de datos PTB-XL, demostrando que su rendimiento es competitivo frente a los modelos de caja negra de vanguardia, a la vez que proporciona explicaciones estructuradas basadas en casos. La representatividad y claridad de los prototipos se verificaron mediante revisiones clínicas estructuradas. ProtoECGNet demuestra que el aprendizaje de prototipos puede escalarse eficazmente a la clasificación compleja de series temporales multietiqueta, lo que proporciona una vía práctica hacia modelos de aprendizaje profundo transparentes y fiables para el apoyo a la toma de decisiones clínicas.

Takeaways, Limitations

Takeaways:
Demostramos la viabilidad de lograr tanto interpretabilidad como rendimiento en la clasificación de electrocardiogramas utilizando un modelo de aprendizaje profundo basado en prototipos.
Presentamos un método de aprendizaje prototipo efectivo para problemas de clasificación de múltiples etiquetas.
Validar la confiabilidad del prototipo mediante revisión clínica.
Un enfoque práctico para el desarrollo de un sistema de apoyo al diagnóstico del electrocardiograma.
Limitations:
Solo se realizó la validación del rendimiento en el conjunto de datos PTB-XL, por lo que se necesita más investigación sobre el rendimiento de generalización.
La interpretabilidad de un prototipo depende de la evaluación subjetiva de los profesionales clínicos. Por lo tanto, es necesario desarrollar criterios de evaluación objetivos.
La complejidad del modelo puede ser alta, lo que genera costos computacionales elevados.
Se necesita una mayor validación en diversos conjuntos de datos de ECG y entornos clínicos.
👍