atlas

Research map / Vision-language and multimodal / Human and embodied modeling

Failures Are Fated, But Can Be Faded: Characterizing and Mitigating Unwanted Behaviors in Large-Scale Vision and Language Models

ICML 2024 · Vision-language and multimodal › Human and embodied modeling · approach: Human-like visual cognition in VLMs · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes a DQN-based post-hoc RL framework to explore and summarize failure landscapes of pretrained vision, NLP, and text-to-image models, then uses a few human feedback points to fine-tune models away from discovered failure modes.

Problem

Exhaustively testing deep vision and language models for failures across combinations of factors is infeasible, and existing failure-finding methods do not characterize the whole failure landscape or help mitigate it.

Approach

Frames failure discovery as a deep RL problem over an actionable concept space (macroscopic and microscopic exploration with DQN), summarizes discovered failures with Q-value-based probability landscapes and Wasserstein barycenters, and restructures the model by fine-tuning on failure-free or user-preferred data guided by a few human feedback points.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use DQN instead of the strongest listed search baselineImage classification failure discovery · ImageNet-1K, AlexNetfailure count: 44 -> 499Helped
Use DQN instead of the strongest listed search baselineImage classification failure discovery · ImageNet-1K, ResNetfailure count: 22 -> 153Helped
Use DQN instead of the strongest listed search baselineImage classification failure discovery · ImageNet-1K, EfficientNetfailure count: 22 -> 109Helped
Use DQN instead of the strongest listed search baselineSummarization failure discovery · Summarize-from-Feedback, BARTfailure count: 42 -> 109Helped
Use DQN instead of the strongest listed search baselineSummarization failure discovery · Summarize-from-Feedback, T5failure count: 35 -> 83Helped
Use DQN instead of the strongest listed search baselineImage-generation failure discovery · Stable Diffusion v1-4failure count: 31 -> 85Helped
Use DQN instead of the lowest-entropy listed search baselineFailure discovery · AlexNet, ResNet, EfficientNet, BART, T5, and Stable Diffusion v1-4entropy: 6.91 -> 5.69; 6.79 -> 5.88; 6.79 -> 5.99; 3.80 -> 3.20; 3.70 -> 3.10; 4.25 -> 3.65Helped
Fine-tune the failure-selected BART modelBART summarization · Summarize-from-Feedbackreward: 13584.5 to -10077Helped
Fine-tune the failure-selected T5 modelT5 summarization · Summarize-from-Feedbackreward: 115317.21 to -276394Helped
LoRA fine-tune on balanced male and female generated imagesStable Diffusion v1-4 · Bias promptsmale-to-female bias ratio: 1.65 to 1.16Helped
LoRA fine-tune on balanced male and female generated imagesStable Diffusion v1-4 · Bias promptsambiguous image generation: 43% dropHelped
Use CLIP embeddings instead of human feedback for rewardStable Diffusion v1-4 failure discovery · Human survey reward comparisonWasserstein distance: 0.0011No clear effect

Surprising findings

Limitations

Experimental setup

Datasets: ImageNet-1K openai/summarize_from_feedback DALL-E 3 generated images

Benchmarks: ImageNet-1K openai/summarize_from_feedback

Metrics: failure count entropy of failure distribution Wasserstein distance BLEU male-to-female bias ratio running time

Baselines: Random search Greedy search Threshold search FGSM adversarial training Bayesian Optimization

Builds on: deep Q-network Bradley-Terry model LoRA FGSM

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy