Logit-Level Uncertainty Quantification in Vision-Language Models for Histopathology Image Analysis
Organizations: Department of Electrical and Electronics Engineering, Izmir Bakircay University, 35665 Izmir, Turkey · Department of Electrical Engineering and Computer Science, University of Stavanger, 4021 Stavanger, Norway · Batten College of Engineering and Technology, Old Dominion University, Norfolk, VA 23529, USA
Abstract
Vision-Language Models (VLMs) with their multimodal capabilities have demonstrated remarkable success in almost all domains, including education, transportation, healthcare, energy, finance, law, and retail. Nevertheless, the utilization of VLMs in healthcare applications raises crucial concerns due to the sensitivity of large-scale medical data and the trustworthiness of these models (reliability, transparency, and security). This study proposes a logit-level uncertainty quantification (UQ) framework for histopathology image analysis using VLMs to deal with these concerns. UQ is evaluated for three VLMs using metrics derived from temperature-controlled output logits. The proposed framework demonstrates a critical separation in uncertainty behavior. While VLMs show high stochastic sensitivity (cosine similarity (CS) and , Jensen-Shannon divergence (JS) and , and Kullback-Leibler divergence (KL) and , respectively for mean values of VILA-M3-8B and LLaVA-Med v1.5), near-maximal temperature impacts (), and displaying abrupt uncertainty transitions, particularly for complex diagnostic prompts. In contrast, the pathology-specific PRISM model maintains near-deterministic behavior (mean CS , JS , KL ) and significantly minimal temperature effects across all prompt complexities. These findings emphasize the importance of logit-level uncertainty quantification to evaluate trustworthiness in histopathology applications utilizing VLMs.
Figures & tables
| Property | VILA-M3 | LLaVA-Med | PRISM |
|---|---|---|---|
| Parameters | 8B | 7B | 0.6B |
| Domain | General | Biomedical | Pathology |
| Vision Encoder | CLIP | CLIP | Virchow |
| Base LLM | Llama-3 | Mistral | Custom |
| KV-Cache | ✓ | ✓ | |
| Stochastic | ✓ | ✓ | Limited |
| Temperatures | |||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Metric | Question | Model | 0.0 | 0.1 | 0.2 | 0.3 | 0.4 | 0.5 | 0.6 | 0.7 | 0.8 | 0.9 | 1.0 |
| Cosine Similarity | Q1 | VILA-M3 | 0.999 | 0.968 | 0.843 | 0.714 | 0.600 | 0.492 | 0.423 | 0.367 | 0.293 | 0.262 | 0.223 |
| LLaVA-Med | 1.000 | 0.980 | 0.937 | 0.839 | 0.677 | 0.562 | 0.387 | 0.261 | 0.192 | 0.173 | 0.175 | ||
| PRISM | 0.988 | 0.992 | 0.991 | 0.984 | 0.967 | 0.931 | 0.867 | 0.759 | 0.596 | 0.359 | 0.029 | ||
| Q2 | VILA-M3 | 0.999 | 0.937 | 0.809 | 0.596 | 0.408 | 0.261 | 0.176 | 0.100 | 0.051 | 0.026 | 0.000 | |
| LLaVA-Med | 1.000 | 0.910 | 0.638 | 0.460 | 0.357 | 0.299 | 0.246 | 0.201 | 0.146 | 0.102 | 0.034 | ||
| Model | Q | Cosine Similarity | JS Divergence | KL Divergence | MAE | ||||
|---|---|---|---|---|---|---|---|---|---|
| VILA-M3 | Q1 | 0.712 | 0.644 | 0.188 | 0.482 | 0.173 | 0.330 | 0.358 | 0.649 |
| Q2 | 0.519 | 0.624 | 0.495 | 1.000 | 0.486 | 0.906 | 0.541 | 0.972 | |
| Q3 | 0.397 | 0.707 | 0.572 | 0.752 | 0.556 | 0.821 | 0.601 | 0.999 | |
| LLaVA-Med | Q1 | 0.837 | 0.433 | 0.018 | 0.116 | 0.013 | 0.090 | 0.278 | 0.609 |
| Q2 | 0.580 | 1.000 | 0.321 | 1.000 | 0.336 | 0.899 | 0.552 | 0.990 | |
| Metric | CS | JS | KL | MAE |
|---|---|---|---|---|
| Cosine Similarity (CS) | 1.000 | |||
| JS Divergence (JS) | 1.000 | 0.997 | 0.958 | |
| KL Divergence (KL) | 0.997 | 1.000 | 0.951 | |
| MAE | 0.958 | 0.951 | 1.000 |
Uncertainty Quantification
Uncertainty quantification estimates how reliable predictions and scientific inferences are, and how uncertainty propagates into decisions.
See all 649 papers