HYATT-Net is Grand: A Hybrid Attention Network for Performant Anatomical Landmark Detection

TL;DR

HYATT-Net is a hybrid CNN-Transformer model designed for accurate and efficient anatomical landmark detection in medical images, achieving state-of-the-art results across multiple datasets.

Contribution

The paper introduces HYATT-Net, a novel hybrid architecture combining CNNs and Transformers with specialized modules for improved ALD performance.

Findings

Achieves state-of-the-art accuracy on five datasets.

Demonstrates robustness and efficiency in complex medical images.

Outperforms existing methods in precision and computational cost.

Abstract

Anatomical landmark detection (ALD) from a medical image is crucial for a wide array of clinical applications. While existing methods achieve quite some success in ALD, they often struggle to balance global context with computational efficiency, particularly with high-resolution images, thereby leading to the rise of a natural question: where is the performance limit of ALD? In this paper, we aim to forge performant ALD by proposing a {\bf HY}brid {\bf ATT}ention {\bf Net}work (HYATT-Net) with the following designs: (i) A novel hybrid architecture that integrates CNNs and Transformers. Its core is the BiFormer module, utilizing Bi-Level Routing Attention for efficient attention to relevant image regions. This, combined with Attention Residual Module(ARM), enables precise local feature refinement guided by the global context. (ii) A Feature Fusion Correction Module that aggregates multi-scale features and thus mitigates a resolution loss. Deep supervision with a mean-square error loss on multi-resolution heatmaps optimizes the model. Experiments on five diverse datasets demonstrate state-of-the-art performance, surpassing existing methods in accuracy, robustness, and efficiency. The HYATT-Net provides a promising solution for accurate and efficient ALD in complex medical images. Our codes and data are already released at: \url{https://github.com/ECNUACRush/HYATT-Net}.