Este estudio no implicó la reclutación de participantes humanos ni la recopilación de datos identificables de pacientes. Todos los experimentos se realizaron utilizando el conjunto de datos público de lesiones cutáneas ISIC 2020 obtenido del repositorio de Kaggle; por lo tanto, no se requirió la aprobación del comité de ética institucional ni el consentimiento informado. Todos los materiales utilizados en este estudio se encuentran en la Tabla de Materiales.
Extracción y fusión de características
Las imágenes de lesiones cutáneas se procesaron utilizando múltiples modelos CNN preentrenados. Los vectores de características profundas extraídos de las arquitecturas CNN preentrenadas seleccionadas se concatenan para construir una representación unificada de características para cada imagen de lesión cutánea. La estrategia de fusión adoptada conserva la información visual complementaria aprendida por diferentes arquitecturas CNN, permitiendo que el clasificador SVM subsiguiente aproveche tanto las características texturales de bajo nivel como las representaciones semánticas de nivel superior. Aunque técnicas de reducción de dimensionalidad, como el análisis de componentes principales o la selección de características basada en información mutua, podrían reducir la dimensionalidad de las características, se mantuvo intencionadamente la representación completa fusionada porque el espacio de características fusionado sigue siendo computacionalmente tratable para el clasificador RBF-SVM empleado, a la vez que preserva la información diagnóstica complementaria extraída de los diferentes modelos CNN base.
Clasificación
Los vectores de características fusionados se utilizaron para entrenar un clasificador SVM con un kernel RBF. Se emplearon técnicas de optimización de hiperparámetros para determinar la configuración óptima de clasificación. Luego, el clasificador categorizó las lesiones cutáneas en sus clases respectivas.
Explicabilidad
Para mejorar la interpretabilidad, se aplicó LIME para generar explicaciones visuales que resaltan las regiones de la imagen que contribuyen de manera más significativa a los resultados de clasificación. Estas explicaciones ayudarían a los clínicos a comprender y validar las decisiones del modelo.
Plan de evaluación
El marco propuesto se evaluó utilizando un conjunto de datos de referencia de lesiones cutáneas. El rendimiento se evaluó mediante Exactitud, Precisión, Exhaustividad y Puntaje F1. Se realizaron experimentos comparativos frente a enfoques existentes de aprendizaje automático y aprendizaje profundo para demostrar la eficacia del marco propuesto.
Resultados esperados
Se esperaba que el estudio produjera un sistema preciso e interpretable de clasificación de lesiones cutáneas. Los resultados experimentales preliminares indican que EDLF-SLC alcanza una exactitud del 88,0 %, una precisión del 89,0 %, una recuperación del 87,0 % y una puntuación F1 del 88,0 %, superando a varios métodos competidores. Se anticipó que la integración de explicaciones LIME mejoraría la confiabilidad y facilitaría la adopción de sistemas de diagnóstico asistidos por inteligencia artificial en la práctica clínica.
Importancia
Esta investigación contribuye al creciente campo de la inteligencia artificial explicativa en atención médica al abordar tanto los desafíos de rendimiento como los de transparencia en el diagnóstico de lesiones cutáneas. El marco propuesto tiene el potencial de ayudar a los dermatólogos a tomar decisiones diagnósticas más confiables e interpretables, mejorando así la atención al paciente. Además, en esta sección, los autores proporcionaron información sobre los materiales y la metodología aplicada en este estudio de investigación. En concreto, los autores comenzaron con la descripción del conjunto de datos utilizado para los experimentos y continuaron con la discusión detallada del marco de aprendizaje profundo explicativo para la clasificación de lesiones cutáneas.
Conjunto de datos
El trabajo presentado se basa en el conjunto de datos ISIC 2020 (Colaboración Internacional de Imágenes Dermatológicas), disponible públicamente, al que se puede acceder en el sitio web de Kaggle (https://www.kaggle.com/datasets/fanconic/skin-cancer-malignant-vs-benign). El repositorio ISIC ha sido reconocido como uno de los mejores recursos en el campo de la imagenología dermatológica, ya que proporciona una gran variedad de imágenes dermatoscópicas de diferentes tipos de lesiones cutáneas, como se muestra en Figura 1. Es importante destacar que el conjunto de datos incluye imágenes tanto de condiciones benignas como malignas, lo que lo hace útil para realizar tareas de clasificación binaria en cáncer de piel 46. En el conjunto de datos actual hay 3.297 imágenes, de las cuales 1.800 son benignas y 1.497 son malignas. Para experimentos más eficientes, fue necesario dividir los datos en un conjunto de entrenamiento y un conjunto de prueba. En particular, hay 2.637 imágenes de entrenamiento, incluyendo 1.440 benignas y 1.197 malignas, mientras que el conjunto de prueba consta de 660 imágenes, de las cuales 360 son benignas y 300 son malignas. Un resumen del conjunto de datos se muestra en Tabla 2.
El modelo propuesto EDLF-SLC
En este artículo se propone una solución eficiente y comprensible para clasificar lesiones cutáneas en categorías benignas y malignas, utilizando una técnica novedosa de aprendizaje profundo interpretable basada en un enfoque híbrido que combina las ventajas de múltiples modelos de redes neuronales convolucionales preentrenados. Las redes neuronales convolucionales han demostrado ser altamente efectivas para extraer características semánticas de alto nivel a partir de imágenes médicas. Aprovechando esta capacidad, el marco de aprendizaje profundo interpretable propuesto para la clasificación de lesiones cutáneas (EDLF-SLC) utiliza múltiples modelos CNN preentrenados para lograr un rendimiento superior. Para garantizar la transparencia necesaria en el proceso de toma de decisiones médicas, se ha adoptado LIME en el enfoque propuesto para generar explicaciones locales comprensibles para los resultados obtenidos. El marco completo puede dividirse en tres etapas principales, como se ilustra en Figura 2, a saber, (1) preprocesamiento de datos, (2) extracción de características y clasificación, y (3) interpretabilidad.
Arquitectura del modelo e integración
Como paso preliminar, se seleccionaron y evaluaron siete modelos populares de aprendizaje profundo (MobileNetV2, ResNet50, EfficientNetB0, Xception, InceptionResNetV2, NASNetLarge y MobileNet) en el conjunto de datos de validación de forma independiente, y se eligieron los cuatro modelos más efectivos (ResNet50, EfficientNetB0, MobileNet y Xception) para la etapa de extracción de características. En el marco propuesto, cada imagen de entrada x se procesa de manera independiente a través de los modelos preentrenados seleccionados. Se ha eliminado la última capa completamente conectada de cada uno de estos modelos, y se han obtenido vectores de características a partir de las capas anteriores. fResNet50(x), fEfficientNetB0(x), fMobileNet(x) y fXception(x) hacen referencia a los vectores de características de salida de cada uno de los modelos mencionados anteriormente. Al concatenar esos vectores de características, se obtiene un nuevo vector de características agregado F(xi):
F(xi) = [f(ResNet50)(xi);f(EfficientNetB0)(xi);f(MobileNet)(xi);f(Xception)(xi)] (1)
Posteriormente, F(xi) se introdujo en un clasificador SVM con un kernel de función de base radial (RBF) para obtener el resultado de clasificación. Todo el algoritmo del marco propuesto se presenta en el Archivo Suplementario 1.
El marco propuesto adopta la fusión directa a nivel de características porque cada arquitectura de CNN preentrenada captura características discriminativas diferentes de las lesiones cutáneas mediante su arquitectura de red y jerarquía de características aprendida. En consecuencia, la concatenación de estas representaciones de características heterogéneas conserva información complementaria que contribuye a una caracterización más completa de la lesión antes de la clasificación. Aunque técnicas de reducción de dimensionalidad, como el análisis de componentes principales (PCA) o la selección de características basada en información mutua, pueden reducir la dimensionalidad de la representación fusionada, se mantuvo intencionalmente el vector completo de características fusionadas porque su dimensionalidad sigue siendo computacionalmente tratable para el clasificador RBF-SVM adoptado, a la vez que preserva la información diagnóstica complementaria extraída por los diferentes modelos de CNN. Por lo tanto, este diseño prioriza la retención de representaciones profundas complementarias en lugar de eliminar características potencialmente informativas antes de la clasificación.
Preparación de los datos
La preparación de los datos ha incluido el preprocesamiento y la división del conjunto de datos en conjuntos de entrenamiento y prueba. El preprocesamiento tiene como objetivo mejorar la calidad de los datos eliminando inconsistencias, suprimiendo elementos duplicados, formateando las imágenes de entrada y realizando una normalización de características para un entrenamiento estable de un buen modelo. Todas las imágenes se han normalizado al rango [−1, 1] mediante la normalización por puntuación Z:
(2)
donde µ y σ representan el valor medio y la desviación estándar de la imagen correspondiente, respectivamente. El conjunto de datos se ha dividido en dos subconjuntos, es decir, un conjunto de entrenamiento (70,0 %) y un conjunto de prueba (30,0 %).
Aumento de datos
Para evitar el sobreajuste y aumentar la capacidad de generalización del modelo, se han utilizado algunas ampliaciones para el conjunto de entrenamiento. Las ampliaciones de imágenes implican modificar las imágenes para expandir artificialmente el conjunto de datos sin alterar las características esenciales de las afecciones médicas. La canalización de ampliación se ha construido utilizando la API secuencial de Keras. Se han empleado transformaciones como volteo horizontal aleatorio, rotación dentro de un ángulo pequeño, cambio de tamaño, escalado, ajuste de brillo y contraste, zoom y algunos movimientos menores. En Figura 3 se ilustran ejemplos representativos de imágenes aumentadas.
Modelos preentrenados
Varios modelos de aprendizaje profundo preentrenados se han adoptado en el marco propuesto para extraer características de imágenes de entrada. Dichos modelos incluyen MobileNet, ResNet50, EfficientNetB0, Xception, NASNetLarge, Inception-ResNet-v2 y MobileNetV2. MobileNet y MobileNetV2 son modelos de aprendizaje profundo ligeros diseñados para realizar cálculos eficientes mediante convoluciones separables por profundidad. ResNet50 emplea el mecanismo de conexiones residuales para entrenar el modelo, evitando así el problema del gradiente que desaparece durante el entrenamiento. EfficientNetB0 logra un equilibrio entre eficiencia y precisión mediante el enfoque de escalado compuesto. Xception extiende la red Inception utilizando convoluciones separables por profundidad. NASNetLarge implica el uso de búsqueda de arquitectura neuronal para construir estructuras óptimas de redes neuronales profundas, e Inception-ResNet-v2 crea una hibridación del modelo Inception y el mecanismo de conexiones residuales. Los vectores de características extraídos de ResNet50 (1.024 características), EfficientNetB0 (1.280 características), MobileNet (1.024 características) y Xception (2.048 características) se concatenan para producir una representación unificada de 5.376 dimensiones. Esta estrategia de fusión se seleccionó para preservar las representaciones complementarias aprendidas por las diferentes arquitecturas de CNN, en lugar de reducir la representación antes de la clasificación. El vector de características resultante se proporciona posteriormente al clasificador RBF-SVM, que determina el límite de decisión no lineal óptimo dentro del espacio de características fusionado.
Modelo de IA interpretable (LIME)
Con el fin de proporcionar cierta interpretabilidad local de las predicciones del modelo, los autores adoptan el método de generar explicaciones localizadas y comprensibles para el ser humano para cada predicción particular del modelo, denominado LIME47. Para cualquier imagen de entrada, LIME primero la divide en unidades más pequeñas llamadas superpíxeles. Luego se generan perturbaciones a partir de la imagen original, y se estiman las predicciones del modelo neuronal profundo para cada una de estas perturbaciones. A continuación, se ajusta un modelo lineal ponderado sobre las perturbaciones, utilizando pesos que dependen de su proximidad a la instancia original de entrada. Tras ajustar el modelo lineal, se estiman las puntuaciones de importancia de las características, lo que indica el papel de cada superpíxel en la predicción de la etiqueta final. Estas puntuaciones de importancia se resaltan visualmente en la imagen final de explicación. El algoritmo LIME se muestra en Archivo Suplementario 2.