Hepato-LLaVA: An Expert MLLM with Sparse Topo-Pack Attention for Hepatocellular Pathology Analysis on Whole Slide Images
Yuxuan Yang, Zhonghao Yan, Yi Zhang, Bo Yun, Muxi Diao, Guowei Zhao, Kongming Liang, Wenbin Li, Zhanyu Ma
TL;DR
Diagnosing hepatocellular carcinoma from gigapixel WSIs is hindered by heterogeneity and the trade-off between patch detail and global context. The authors propose Sparse Topo-Pack Attention to model 2D tissue topology, introduce the HepatoPathoVQA multi-scale dataset, and develop Hepato-LLaVA via a three-stage training pipeline with a Q-Former connector, achieving state-of-the-art results (Avg 0.83) across morphology and diagnosis tasks with robust multi-scale consistency. This topology-aware, multi-scale approach reduces redundancy while preserving critical diagnostic cues, enabling precise, clinically relevant VQA and captioning for HCC pathology. Collectively, the work advances efficient, fine-grained WSI analysis and has potential to improve real-world pathology workflows.
Abstract
Hepatocellular Carcinoma diagnosis relies heavily on the interpretation of gigapixel Whole Slide Images. However, current computational approaches are constrained by fixed-resolution processing mechanisms and inefficient feature aggregation, which inevitably lead to either severe information loss or high feature redundancy. To address these challenges, we propose Hepato-LLaVA, a specialized Multi-modal Large Language Model designed for fine-grained hepatocellular pathology analysis. We introduce a novel Sparse Topo-Pack Attention mechanism that explicitly models 2D tissue topology. This mechanism effectively aggregates local diagnostic evidence into semantic summary tokens while preserving global context. Furthermore, to overcome the lack of multi-scale data, we present HepatoPathoVQA, a clinically grounded dataset comprising 33K hierarchically structured question-answer pairs validated by expert pathologists. Our experiments demonstrate that Hepato-LLaVA achieves state-of-the-art performance on HCC diagnosis and captioning tasks, significantly outperforming existing methods. Our code and implementation details are available at https://pris-cv.github.io/Hepto-LLaVA/.
