Fine-Grained Rib Fracture Diagnosis with Hyperbolic Embeddings: A Detailed Annotation Framework and Multi-Label Classification Model
Journal:
arXiv
Published Date:
Apr 15, 2025
Abstract
Accurate rib fracture identification and classification are essential for
treatment planning. However, existing datasets often lack fine-grained
annotations, particularly regarding rib fracture characterization, type, and
precise anatomical location on individual ribs. To address this, we introduce a
novel rib fracture annotation protocol tailored for fracture classification.
Further, we enhance fracture classification by leveraging cross-modal
embeddings that bridge radiological images and clinical descriptions. Our
approach employs hyperbolic embeddings to capture the hierarchical nature of
fracture, mapping visual features and textual descriptions into a shared
non-Euclidean manifold. This framework enables more nuanced similarity
computations between imaging characteristics and clinical descriptions,
accounting for the inherent hierarchical relationships in fracture taxonomy.
Experimental results demonstrate that our approach outperforms existing methods
across multiple classification tasks, with average recall improvements of 6% on
the AirRib dataset and 17.5% on the public RibFrac dataset.
