PathMem: Toward Cognition-Aligned Memory Transformation for Pathology MLLMs
Organizations: University of Science and Technology of China · Shenzhen University · Nanyang Technological University · IGS, Imperial College London ·
Abstract
Computational pathology demands both visual pattern recognition and dynamic integration of structured domain knowledge, including taxonomy, grading criteria, and clinical evidence. In practice, diagnostic reasoning requires linking morphological evidence with formal diagnostic and grading criteria. Although multimodal large language models (MLLMs) demonstrate strong vision language reasoning capabilities, they lack explicit mechanisms for structured knowledge integration and interpretable memory control. As a result, existing models struggle to consistently incorporate pathology-specific diagnostic standards during reasoning. Inspired by the hierarchical memory process of human pathologists, we propose PathMem, a memory-centric multimodal framework for pathology MLLMs. PathMem organizes structured pathology knowledge as a long-term memory (LTM) and introduces a Memory Transformer that models the dynamic transition from LTM to working memory (WM) through multimodal memory activation and context-aware knowledge grounding, enabling context-aware memory refinement for downstream reasoning. PathMem achieves SOTA performance across benchmarks, improving WSI-Bench report generation (12.8% WSI-Precision, 10.1% WSI-Relevance) and open-ended diagnosis by 9.7% and 8.9% over prior WSI-based models.
Figures & tables
| Morphological Analysis | Diagnosis | Treatment Planning | Average | ||||||||
| Model | Input | Open | Close | Open | Close | Open | Close | ||||
| WSI-P | WSI-R | Acc | WSI-P | WSI-R | Acc | WSI-P | WSI-R | Acc | |||
| Quilt-LLaVA Seyfioglu et al. (2024) | Thumbnail | 0.448 | 0.447 | 0.947 | 0.586 | 0.604 | 0.849 | 0.788 | 0.816 | 1.000 | 0.721 |
| GPT-4o Islam and Moushi (2024) | Thumbnail | 0.220 | 0.204 | 0.471 | 0.472 | 0.457 | 0.530 | 0.496 | 0.841 | 0.875 | 0.507 |
| WSI-VQA Chen et al. (2024c) | WSI | 0.395 | 0.462 | 0.758 | 0.436 | 0.525 | 0.577 | 0.791 | 0.595 | 0.771 | 0.590 |
| WSI-LLaVA Liang et al. (2025) | WSI | 0.488 | 0.610 | 0.951 | 0.610 | 0.612 | 0.863 | 0.810 | 0.845 | 1.000 | 0.754 |
| Models | BLEU-1 | BLEU-2 | BLEU-3 | BLEU-4 | ROUGE-L | METEOR | WSI-Precision | WSI-Relevance |
| GPT-4o Islam and Moushi (2024) | 0.202 | 0.069 | 0.030 | 0.016 | 0.132 | 0.167 | 0.067 | 0.138 |
| Quilt-LLaVA Seyfioglu et al. (2024) | 0.474 | 0.351 | 0.282 | 0.236 | 0.475 | 0.460 | 0.324 | 0.333 |
| MI-Gen Chen et al. (2024b) | 0.403 | 0.306 | 0.248 | 0.209 | 0.446 | 0.407 | 0.310 | 0.377 |
| Hist-Gen Guo et al. (2024) | 0.406 | 0.307 | 0.248 | 0.208 | 0.448 | 0.416 | 0.300 | 0.397 |
| WSI-LLaVA Liang et al. (2025) | 0.480 | 0.358 | 0.287 | 0.240 | 0.490 | 0.465 | 0.380 | 0.429 |
| PathMem (Ours) | 0.548 0.151 | 0.420 0.173 | 0.347 0.188 | 0.302 0.196 | 0.536 0.155 | 0.531 0.158 | 0.508 0.147 | 0.530 0.144 |
| Model | Input | WSI-VQA | SlideBench-VQA (BCNB) | CPTAC | |||
| Tumor | Grade | Subtype | Avg | ||||
| Quilt-LLaVA Seyfioglu et al. (2024) | Thumb | 0.132 | 0.113 | 0.194 | 0.181 | 0.163 | 0.603 |
| WSI-VQA Chen et al. (2024c) | WSI | 0.469 | 0.039 | 0.300 | 0.000 | 0.113 | 0.000 |
| SlideChat Chen et al. (2025c) | WSI | - | 0.902 | 0.231 | 0.175 | 0.436 | - |
| WSI-LLaVA Liang et al. (2025) | WSI | 0.546 | 0.905 | 0.463 | 0.292 | 0.553 | 0.721 |
| PathMem (Ours) | WSI | 0.572 | 0.918 | 0.482 | 0.313 | 0.571 | 0.754 |
| Version | Baseline | LTM | PathMem | ||||||||
| - | D | S | BLEU-1 | BLEU-2 | BLEU-3 | BLEU-4 | ROUGE-L | METEOR | WSI-P | WSI-R | |
| a | ✓ | 0.481 0.181 | 0.357 0.196 | 0.287 0.205 | 0.241 0.212 | 0.477 0.177 | 0.464 0.178 | 0.404 0.162 | 0.445 0.163 | ||
| b | ✓ | ✓ | 0.528 0.153 | 0.405 0.176 | 0.332 0.195 | 0.284 0.198 | 0.518 0.168 | 0.508 0.164 | 0.492 0.155 | 0.516 0.147 | |
| c | ✓ | ✓ | 0.521 0.161 | 0.396 0.183 | 0.324 0.205 | 0.278 0.205 | 0.510 0.170 | 0.511 0.163 | 0.489 0.158 | 0.510 0.148 | |
| d | ✓ | ✓ | ✓ | 0.548 0.151 | 0.420 0.173 | 0.347 0.188 | 0.302 0.196 | 0.536 0.155 | 0.531 0.158 | 0.508 0.147 | 0.530 0.144 |
| Version | Token Cap | PathMem | ||||||||
| D | S | BLEU-1 | BLEU-2 | BLEU-3 | BLEU-4 | ROUGE-L | METEOR | WSI-P | WSI-R | |
| a | 1 | 1 | 0.517 0.161 | 0.393 0.184 | 0.320 0.198 | 0.274 0.210 | 0.508 0.172 | 0.500 0.169 | 0.477 0.163 | 0.496 0.151 |
| b | 2 | 2 | 0.525 0.160 | 0.402 0.181 | 0.330 0.204 | 0.284 0.205 | 0.517 0.168 | 0.509 0.164 | 0.484 0.159 | 0.497 0.148 |
| c | 3 | 3 | 0.531 0.153 | 0.410 0.175 | 0.339 0.193 | 0.293 0.196 | 0.525 0.162 | 0.514 0.161 | 0.493 0.155 | 0.513 0.147 |
| d | 4 | 4 | 0.537 0.154 | 0.414 0.173 | 0.340 0.194 | 0.292 0.197 | 0.528 0.158 | 0.517 0.161 | 0.502 0.149 | 0.526 0.146 |
| g | 5 | 5 | 0.542 0.151 | 0.418 0.174 | 0.346 0.188 | 0.298 0.197 | 0.532 0.156 | 0.523 0.158 | 0.506 0.147 | 0.528 0.143 |
| Metric | Value |
| Inference Time | 1.45 items (WSI) / s |
| GPU Utilization | 1 GPU |
| GPU Memory Usage | 16.3 GB |
| Image Feature Size | tokens |
| KG Construction Cost | Offline embedding + indexing |
| KG Update Cost | Re-embedding + re-indexing |
Appendix figures & tables5 assets
Supplementary material from the paper’s appendix.
Appendix
Memory-Augmented VLMs
Memory-augmented vision-language models use persistent or retrieved context to improve reasoning across long inputs and interactions.
See all 157 papers