Medical hierarchical image classification via dual-geometry image–text learning

Read full text
Authors: Lei Fan, Arcot Sowmya, Erik Meijering, Zhen Yu, Zongyuan Ge, Yang Song
Cited by: 1
References: 38

Abstract

Hierarchical image classification is a fundamental challenge in medical image analysis, as tree-structured taxonomies inherently reflect biological and clinical relationships, spanning the general categorisation of disease entities and fine-grained cellular distinctions. Existing approaches primarily rely on multi-task learning and fine-grained detection, often requiring intricate model design and complex training strategies. In this paper, we aim to exploit the negative curvature property of hyperbolic space, which allows efficient representation of hierarchical structures. We propose a dual-geometry image-text framework, termed H 2 CL. Specifically, we introduce a lightweight classifier head on top of image backbones to extract both Euclidean and hyperbolic features, which are then combined to simultaneously preserve taxonomic consistency from an etiological perspective and enhance instance discrimination from a morphological perspective. Furthermore, a text branch is incorporated to integrate label semantics, where an entailment loss is employed to jointly model image–text alignment and inter-sample relationships. Extensive experiments on cervical cell, skin lesion, and gallbladder disease datasets demonstrate that our framework consistently outperforms advanced methods. Compared to the standard Swin Transformer, H 2 CL achieves an average accuracy improvement of 7% across all three datasets at the fine-grained level, with similarly consistent gains observed when integrated with other backbone models. The source code is publicly available at https://github.com/MCPathology/H2CL .

Fields of Study & Subjects

AI in cancer detection COVID-19 diagnosis using AI Medical Imaging and Analysis

Publication details

Journal / source
Medical Image Analysis
Publisher
Elsevier BV
Volume
112
Pages
104120–104120
Publication date
30 April 2026
Language
EN