You May Speak Freely: Improving the Fine-Grained Visual Recognition Capabilities of Multimodal Large Language Models with Answer Extraction
Lawrence, Saha, Wei et al.
Despite the renewed interest in zero-shot visual classification due to the rise of Multimodal Large Language Models (MLLMs), the problem of evaluating free-form responses of auto-regressive models remains a persistent challenge. Most existing works focus on language-only tasks or don't consider Multiple Choice Questions (MCQs) beyond 5-way options, both of which are critical capabilities to solve tasks in Fine-Grained Visual Classification (FGVC) where choice counts are in the hundreds to thousands and the choices are highly related. Furthermore, in this highly multi-way MCQ setting it is not clear how to extend LLM choice extraction to retrieval-based problems, where computing probabilities over the choice set is computationally costly. In this work we investigate nlg2choice, a simple two-stage method which first asks the MLLM an open-ended question for the task with minimal constraints, then uses text-only constrained decoding to predict the most likely choice. In retrieval settings, we compute the probability of the constrained response taking that choice with an early stopping method to significantly improve throughput. Our results show improvement over a suite of seven fine-grained visual datasets when evaluating in terms of classification and retrieval, and show that this performance holds over the various ways that users of LLMs can implement tasks in natural language.
academic
Usted Puede Hablar Libremente: Mejora de las Capacidades de Reconocimiento Visual Fino de Modelos de Lenguaje Grandes Multimodales con Extracción de Respuestas
A pesar del auge de los modelos de lenguaje grandes multimodales (MLLMs), la evaluación de respuestas de forma libre de modelos autorregresivos sigue siendo un desafío continuo. La mayoría de los trabajos existentes se centran en tareas puramente lingüísticas o no consideran preguntas de opción múltiple con más de 5 opciones, ambas siendo capacidades clave para resolver tareas de clasificación visual de grano fino (FGVC), donde el número de opciones alcanza cientos a miles y las opciones son altamente correlacionadas. Además, en esta configuración de MCQ altamente multiseleccionable, no está claro cómo extender la extracción de selección de LLM a problemas basados en recuperación, ya que calcular probabilidades sobre el conjunto de selecciones es computacionalmente costoso. Este artículo investiga nlg2choice, un enfoque simple de dos etapas que primero plantea una pregunta abierta al MLLM con restricciones mínimas, y luego utiliza decodificación restringida de texto puro para predecir la selección más probable. En la configuración de recuperación, se adopta un método de parada temprana para calcular la probabilidad de que la respuesta restringida seleccione esa opción, mejorando significativamente el rendimiento.
Desafíos de la Clasificación Visual de Grano Fino: Los métodos tradicionales de opción múltiple funcionan mal cuando se enfrentan a cientos o miles de opciones altamente similares, como en la identificación de especies de aves donde LLaVA-1.5 logra un desempeño casi perfecto en clasificación de grano grueso (como "ave" vs "no ave"), pero solo alcanza una precisión del 1-2% en etiquetas de especies de grano fino.
Limitaciones de los Métodos de Evaluación: Los métodos existentes o bien fuerzan un formato de salida restringido (que puede obstaculizar el razonamiento) o permiten interpretaciones de forma libre (pero difíciles de extraer), careciendo de mecanismos efectivos de extracción de respuestas.
Problemas de Eficiencia Computacional: En escenarios de recuperación, el costo computacional de calcular probabilidades para cientos o miles de selecciones es prohibitivo.
Propuesta del Método nlg2choice: Un método simple y efectivo de extracción de respuestas de dos etapas que mejora significativamente el desempeño de clasificación y recuperación en 7 conjuntos de datos de visión de grano fino.
Verificación de Robustez: Mediante la generación de variantes de indicaciones semánticamente equivalentes, se demuestra la robustez del método ante variaciones en la entrada del usuario, con mejoras estadísticamente significativas.
Optimización de Parada Temprana Propuesta: Se introduce un método de parada temprana en la configuración de recuperación, mejorando el rendimiento en 15 veces (alcanzando una mejora del 1362% en algunos conjuntos de datos).
Análisis Sistemático: Se demuestra que la decodificación restringida es un extractor de respuestas confiable sin necesidad de entrenamiento adicional, siendo el cuello de botella principal la falta de contenido extraíble en las respuestas de forma libre en sí mismas, no la capacidad de extracción de respuestas.
Dado una imagen y una tarea de clasificación visual de grano fino, el objetivo es identificar con precisión el contenido de la imagen de entre un gran número de categorías altamente similares (cientos a miles), como especies de aves, variedades de flores, modelos de automóviles, etc.
Indicación: "¿Cuál es la especie de ave más probable indicada en esta respuesta?
Respuesta: [nlg]
Respuesta de las siguientes opciones: [lista_de_opciones]"
Se utiliza decodificación restringida para garantizar que la salida provenga de una lista de categorías predefinida.
En escenarios de recuperación, se mejora la eficiencia mediante el truncamiento del cálculo de probabilidades:
Para el nombre de categoría "Baltimore Oriole", descompuesto en "B", "altimore", " Ori", "ole", cuando "altimore" es único entre todas las categorías, se detiene el cálculo de probabilidades de tokens posteriores:
En todos los modelos y conjuntos de datos, nlg2choice supera significativamente la decodificación restringida directa:
Modelo
Mejora de Precisión Promedio
Qwen-2.5VL
+17.46%
Llama-3.2V
+8.49%
Intern3VL
+6.87%
Mejor Desempeño: Qwen-2.5VL alcanza una precisión promedio del 56.91% con indicaciones abiertas, siendo el conjunto de datos Flowers el que alcanza el 78.03%.
Métodos tempranos: Análisis de expresiones regulares, pero con bajo desempeño en tareas de grano fino
Ordenamiento de probabilidades: Basado en probabilidades del primer token de ID de opción (A/B/C/D), ampliamente adoptado pero computacionalmente costoso
Decodificación restringida: Garantiza que la salida esté dentro del conjunto de selecciones, pero evaluaciones recientes muestran disminución de desempeño
La Extracción de Respuestas Mejora Significativamente las Capacidades de Reconocimiento Visual: Mejora en todas las arquitecturas y conjuntos de datos probados
El Método es Robusto ante Variaciones del Usuario: Las mejoras de desempeño son estadísticamente significativas e independientes del formato de indicación específico
La Decodificación Restringida es un Extractor Confiable: Funciona efectivamente sin necesidad de entrenamiento adicional
Limitaciones de Escala de Modelo: Principalmente se prueban modelos de tamaño medio (8B-11B), utilizando solo modelos de código abierto
Requisitos de Recursos Computacionales: Aunque se evita el entrenamiento especializado, aún requiere recursos computacionales significativos para procesar descripciones de texto
Escalabilidad Multietiqueta: La aplicabilidad en problemas multietiqueta requiere verificación adicional
Método Simple y Efectivo: El diseño de dos etapas es intuitivo, sin requerir datos de entrenamiento adicionales o modificaciones de arquitectura
Experimentación Integral: Pruebas en múltiples modelos, conjuntos de datos y dimensiones de evaluación, incluyendo verificación de robustez
Alto Valor Práctico: La optimización de parada temprana resuelve problemas de eficiencia computacional en despliegue real
Análisis Profundo: Mediante anotación manual se verifica la efectividad de la extracción de respuestas, identificando los verdaderos cuellos de botella
Análisis Teórico Insuficiente: Falta de explicación teórica sobre por qué el método de dos etapas es más efectivo
Cobertura de Modelos Limitada: No se prueban modelos propietarios de primer nivel como GPT-4V
Alcance de Tareas: Se enfoca principalmente en clasificación de etiqueta única, con cobertura insuficiente de tareas multietiqueta y otras tareas visuales
Este trabajo proporciona una solución práctica para clasificación visual de grano fino, siendo particularmente valioso en aplicaciones reales que requieren clasificación precisa entre un gran número de categorías similares. La simplicidad del método y la característica de no requerir entrenamiento adicional lo hacen fácil de adoptar e implementar.
El artículo cita 47 referencias relacionadas, abarcando trabajos importantes en campos clave como modelos de lenguaje grandes multimodales, decodificación restringida y extracción de respuestas, proporcionando una base teórica sólida para la investigación.