Desarrollando nuevas capas completamente conectadas para redes neuronales convolucionales con optimización de hiperparámetros para una mejor clasificación de imágenes multietiqueta
Autores: Katona, Tamás; Tóth, Gábor; Petró, Mátyás; Harangi, Balázs
Idioma: Inglés
Editor: MDPI
Año: 2024
Acceso abierto
Artículo científico
2024
Desarrollando nuevas capas completamente conectadas para redes neuronales convolucionales con optimización de hiperparámetros para una mejor clasificación de imágenes multietiqueta
Categoría
Matemáticas
Subcategoría
Matemáticas generales
Palabras clave
Radiografía de tórax
Modelo de aprendizaje profundo
Redes neuronales convolucionales
Algoritmo Hyperband
Métrica AUC-ROC
Interpretación automatizada
Licencia
CC BY-SA – Atribución – Compartir Igual
Consultas: 36
Citaciones: Sin citaciones
La evaluación de radiografías de tórax es desafiante debido a su alta demanda y la complejidad de los diagnósticos. En este estudio, proponemos un modelo optimizado de aprendizaje profundo para la clasificación multietiqueta de imágenes de radiografías de tórax. Aprovechamos redes neuronales convolucionales preentrenadas (CNN) como VGG16, ResNet 50 y DenseNet 121, modificando sus capas de salida y ajustando los modelos. Empleamos una estrategia de optimización novedosa utilizando el algoritmo Hyperband para buscar eficientemente el espacio de hiperparámetros mientras ajustamos las capas completamente conectadas de las CNN. La efectividad de nuestro enfoque se evalúa en función del Área Bajo la Curva Característica de Operación del Receptor (AUC-ROC). Nuestra metodología propuesta podría ayudar en la interpretación automatizada de radiografías de tórax, ofreciendo una herramienta valiosa que los clínicos podrían utilizar en el futuro.
Descripción
La evaluación de radiografías de tórax es desafiante debido a su alta demanda y la complejidad de los diagnósticos. En este estudio, proponemos un modelo optimizado de aprendizaje profundo para la clasificación multietiqueta de imágenes de radiografías de tórax. Aprovechamos redes neuronales convolucionales preentrenadas (CNN) como VGG16, ResNet 50 y DenseNet 121, modificando sus capas de salida y ajustando los modelos. Empleamos una estrategia de optimización novedosa utilizando el algoritmo Hyperband para buscar eficientemente el espacio de hiperparámetros mientras ajustamos las capas completamente conectadas de las CNN. La efectividad de nuestro enfoque se evalúa en función del Área Bajo la Curva Característica de Operación del Receptor (AUC-ROC). Nuestra metodología propuesta podría ayudar en la interpretación automatizada de radiografías de tórax, ofreciendo una herramienta valiosa que los clínicos podrían utilizar en el futuro.