AttriPrompter: Auto-Prompting With Attribute Semantics for Zero-Shot Nuclei Detection via Visual-Language Pre-Trained Models
Large-scale visual-language pre-trained models (VLPMs) have demonstrated exceptional performance in downstream object detection through text prompts for natural scenes. However, their application to zero-shot nuclei detection on histopathology images remains relatively unexplored, mainly due to the significant gap between the characteristics of medical images and the web-originated text-image pairs used for pre-training. This paper aims to investigate the potential of the object-level VLPM, Grounded Language-Image Pre-training (GLIP), for zero-shot nuclei detection. Specifically, we propose an innovative auto-prompting pipeline, named AttriPrompter, comprising attribute generation, attribute augmentation, and relevance sorting, to avoid subjective manual prompt design. AttriPrompter utilizes VLPMs' text-to-image alignment to create semantically rich text prompts, which are then fed into GLIP for initial zero-shot nuclei detection. Additionally, we propose a self-trained knowledge distillation framework, where GLIP serves as the teacher with its initial predictions used as pseudo labels, to address the challenges posed by high nuclei density, including missed detections, false positives, and overlapping instances. Our method exhibits remarkable performance in label-free nuclei detection, outperforming all existing unsupervised methods and demonstrating excellent generality. Notably, this work highlights the astonishing potential of VLPMs pre-trained on natural image-text pairs for downstream tasks in the medical field as well. Code will be released at github.com/AttriPrompter.
Authors
- Bingzheng Wei (ORCID: https://orcid.org/0000-0001-6979-0459)
- Jianzhong Shou (ORCID: https://orcid.org/0000-0002-1148-2442)
- Yongjian Wu (ORCID: https://orcid.org/0009-0000-5631-164X)
- Maode Lai (ORCID: https://orcid.org/0000-0001-6830-9774)
- Jiya Saiyin
- Yan Xu (ORCID: https://orcid.org/0000-0002-2636-7594)
- Yang Zhou (ORCID: https://orcid.org/0000-0003-2848-7642)
Institutions
- Chinese Academy of Medical Sciences & Peking Union Medical College (CN)
- Peking Union Medical College Hospital (CN)
- National Cancer Center (US)
- Biomechanics Institute of Valencia (ES)
- National Clinical Research (US)
- Alibaba Group (China) (CN)
- Zhejiang University (CN)
- Beihang University (CN)
Publication Details
- Journal
- IEEE Transactions on Medical Imaging
- Published
- 2024-10-03
- DOI
- https://doi.org/10.1109/tmi.2024.3473745
- Citations
- 3
- Primary Topic
- Topic Modeling
- Type
- article
- Field-Weighted Citation Impact
- 1.04
Funders
- National Natural Science Foundation of China