Safety as a Classifier, Not a Promise
Granite Vision's safety design asks a different question from its document benchmarks: can the model classify whether a multimodal input is harmful or whether a claim is hallucinated, rather than only generate a helpful-looking response? This chapter explains that separate classification path, called Sparse Attention Vectors, and distinguishes its reported gains from the paper's VLM-as-a-Judge results. The practical judgment is narrow but important: feature separability on selected safety tasks is evidence about a useful mechanism, not a complete guarantee for an enterprise deployment.
Why classify safety instead of asking for a refusal
A generative multimodal model usually demonstrates safety by producing a response. A prompt is sent to the model, and an evaluator judges whether the answer is harmful, faithful, fair, or appropriately refused. That setup resembles ordinary use, but it makes the measurement depend on the wording and quality of a generated answer. It can be difficult to tell whether a model recognized the unsafe input, followed a refusal pattern, or simply produced a response that a judge happened to rate favorably.
Granite Vision's paper adds a second setup in which safety questions are reformulated as classification problems. The model is asked to choose among explicit labels, such as “safe” and “unsafe,” or “hallucinating” and “not hallucinating.” Classification produces a direct decision that can be compared with a ground-truth label. The authors motivate this approach by noting that generative multimodal models do not always excel at discriminative tasks, even though strong discrimination is important for safety evaluation and other applications.
The distinction is not merely academic. Consider an image and an instruction that might contain harmful content. In a VLM-as-a-Judge setup, the model generates text and another model scores that text. In the classification setup, the system must decide which class the input belongs to. These tests can reveal different weaknesses. A model may produce a cautious-sounding answer without reliably separating safe and unsafe cases. It may classify a case correctly while still generating a poor explanation. Treating the setups as interchangeable would hide those differences.
Granite Vision's separate method uses Sparse Attention Vectors, or Safety Vectors. An attention head is one of the small processing units inside a transformer layer. At the end of a sequence, each head has an attention vector that summarizes part of the model's internal state for the final token. The method does not assume that every head is equally useful for safety. It searches for a sparse subset whose vectors separate the safety classes well.
The search begins with a small set of labeled examples. The labels describe a task-specific property, such as harmful versus non-harmful or hallucinating versus not hallucinating. For each example, the method extracts the final-token attention vector from each layer and head. This produces many candidate features: one vector for one head at one layer, evaluated across the labeled examples.
Next, the method builds a centroid for each class and each candidate head. A centroid is the average vector for the examples assigned to a class.