Configura la visualización basada en funciones

Vertex Explainable AI proporciona funciones de visualización integradas para los datos de imagen. Puedes configurar visualizaciones para los modelos de imagen de entrenamiento personalizado.

Cuando solicitas una explicación de un modelo de clasificación de imágenes, obtienes la clase prevista junto con una superposición de imágenes que muestra qué píxeles (gradientes integrados) o regiones (gradientes integrados o XRAI) contribuyeron a la predicción.

En las siguientes imágenes, se muestran visualizaciones en la imagen de un husky siberiano. Para la visualización izquierda, se usa el método de gradientes integrados y se destacan las áreas de atribución positiva. En la visualización derecha, se usa un método XRAI con un gradiente de color que indica áreas de menor influencia (azul) y mayor influencia (amarillo) para hacer una predicción positiva.

Visualización de una atribución de atributos de un husky siberiano mediante gradientes integrados
Visualización de una atribución de atributos de un husky siberiano mediante XRAI

El tipo de datos con el que trabajas puede influir en la decisión de usar un enfoque XRAI o de gradientes integrados para visualizar las explicaciones.

  • XRAI suele funcionar mejor con imágenes naturales y, a su vez, proporciona un mejor resumen de alto nivel de estadísticas, como, por ejemplo, mostrar que la atribución positiva se relaciona con la forma del rostro del perro.
  • Los gradientes integrados (IG) suelen proporcionar detalles a nivel de píxel y son útiles para descubrir atribuciones más detalladas.

Obtén más información sobre los métodos de atribución en la página Overview de Vertex Explainable AI.

Comenzar

Configura la visualización cuando crees un recurso Model que admita Vertex Explainable AI o cuando anules la ExplanationSpec de Model.

Para configurar la visualización de tu modelo, propaga el campo visualization del mensaje InputMetadata correspondiente al atributo que deseas visualizar. En este mensaje de configuración, puedes incluir opciones, como el tipo de superposición usada, las atribuciones destacadas, el color y más. Todos los parámetros de configuración son opcionales.

Opciones de visualización

La configuración predeterminada y recomendada depende del método de atribución (XRAI o gradientes integrados). En la siguiente lista, se describen las opciones de configuración y cómo se pueden usar. Si deseas ver una lista completa de opciones, consulta la referencia de la API para el mensaje Visualization.

  • type: Es el tipo de visualización usado: OUTLINES o PIXELS. Especifica este campo solo si usas gradientes integrados. No puedes especificarlo si usas XRAI.

    En el caso de los gradientes integrados, el campo predeterminado es OUTLINES, que muestra regiones de atribución. Para mostrar la atribución por píxel, configura el campo como PIXELS.

  • polarity: Es la direccionalidad de las atribuciones destacadas. positive se establece de forma predeterminada, por lo que se destacan las áreas con las atribuciones positivas más altas. Esto implica que se destacan los píxeles que fueron más influyentes para la predicción positiva del modelo. Si configuras la polaridad en negative, se destacan las áreas que llevaron al modelo a no predecir la clase positiva. Usar una polaridad negativa puede ser útil para depurar el modelo mediante la identificación de regiones de falsos negativos. También puedes establecer la polaridad en both, lo que muestra las atribuciones positivas y negativas.

  • clip_percent_upperbound: Excluye las atribuciones por encima del percentil especificado de las áreas destacadas. Usar los parámetros de recorte a la vez puede ser útil para filtrar el ruido y facilitar la visualización de áreas de atribución sólida.

  • clip_percent_lowerbound: Excluye las atribuciones por debajo del percentil especificado de las áreas destacadas.

  • color_map: Es el esquema de colores que se usa para las áreas destacadas. El valor predeterminado es pink_green para los gradientes integrados, lo que muestra las atribuciones positivas en verde y las negativas en rosa. Para las visualizaciones de XRAI, el mapa de colores es un gradiente. El valor predeterminado de XRAI es viridis, lo que destaca las regiones más influyentes en amarillo y las menos influyentes en azul.

    Para obtener una lista completa de los valores posibles, consulta la referencia de la API para el mensaje Visualization.

  • overlay_type: Es la forma en la que se muestra la imagen original en la visualización Ajustar la superposición permite aumentar la claridad visual si la imagen original es difícil de visualizar.

    Para obtener una lista completa de los valores posibles, consulta la referencia de la API para el mensaje Visualization.

Configuración de ejemplo

Para comenzar, a continuación, se muestran ejemplos de configuraciones de Visualization que puedes usar como punto de partida e imágenes que muestran un rango de configuraciones aplicadas.

Gradientes integrados

En el caso de los gradientes integrados, es posible que debas ajustar los valores de recorte si las áreas de atribución presentan demasiado ruido.

visualization: {
  "type": "OUTLINES",
  "polarity": "positive",
  "clip_percent_lowerbound": 70,
  "clip_percent_upperbound": 99.9,
  "color_map": "pink_green",
  "overlay_type": "grayscale"
}

A continuación, se muestran dos visualizaciones en las que se usan los tipos outlines y pixels. Las columnas con la etiqueta “Highly predictive only” (“Solo nivel de predicción alto”), “Moderately predictive” (“Nivel de predicción moderado”) y “Almost all” (“Casi todo”) son ejemplos de recorte en diferentes niveles que pueden ayudarte a centrar tu atención en la visualización.

Una visualización de atribución de atributos con descripciones para la atribución de IG

Una visualización de atribución de atributos con píxeles para la atribución de IG

XRAI

Para las visualizaciones de XRAI, recomendamos comenzar sin valores de recorte de XRAI porque la superposición usa un gradiente a fin de mostrar áreas de atribución alta y baja.

visualization: {
  "type": "PIXELS",
  "polarity": "positive",
  "clip_percent_lowerbound": 0,
  "clip_percent_upperbound": 100,
  "color_map": "viridis",
  "overlay_type": "grayscale"
}

La imagen a continuación es una visualización de XRAI en la que se usa el mapa de colores predeterminado viridis y un rango de tipos de superposición. Mediante las áreas en amarillo se indican las regiones más influyentes que contribuyeron de manera positiva a la predicción.

Una visualización de la atribución de atributos para la atribución de XRAI

¿Qué sigue?