IHA-YOLO: Inter-Head Attention for Real-Time Cell Detection
Sheikh, Tooba Tehreem Kumar, Komal Awais, Muhammad Novershtern, Noa Anwer, Rao Muhammad Cholakkal, Hisham
Sheikh, Tooba Tehreem
Kumar, Komal
Awais, Muhammad
Novershtern, Noa
Anwer, Rao Muhammad
Cholakkal, Hisham
Supervisor
Department
Computer Vision
Embargo End Date
Type
Conference proceeding
Date
2025
License
Language
English
Collections
Research Projects
Organizational Units
Journal Issue
Abstract
Multiclass cell detection is a crucial task in numerous biomedical applications, particularly in cell biology. The development of YOLO object detection models has advanced the field of real-time detection, but it is still struggling with challenges in medical imaging due to limited data availability, overlapping tiny objects, diverse cell types, and class imbalances. In this paper, we introduce Inter-Head Attention (IHA)-YOLO, a novel model that proposes an inter-head self-attention block to enhance global representation learning, thereby improving the contextual understanding across feature maps and performing effective detection of small cells and sub-cell structures in medical images. Through extensive experiments on five publicly available datasets, IHA-YOLO outperforms the state-of-the-art methods, achieving an average absolute mAP50 improvement of 2.03% and a 13% faster inference rate. In addition to cell detection, we adapt IHA-YOLO for cell counting to demonstrate its effectiveness. Our source code and models are available at: https://github.com/toobatehreem/IHA-YOLO.
Citation
T. T. Sheikh, K. Kumar, M. Awais, N. Novershtern, R. M. Anwer and H. Cholakkal, "IHA-YOLO: Inter-Head Attention for Real-Time Cell Detection," 2025 IEEE 22nd International Symposium on Biomedical Imaging (ISBI), Houston, TX, USA, 2025, pp. 1-5, doi: 10.1109/ISBI60581.2025.10981067
Source
2025 IEEE 22nd International Symposium on Biomedical Imaging (ISBI)
Conference
IEEE International Symposium on Biomedical Imaging, 2025
Keywords
Cell Detection, Deep Learning
Subjects
Source
IEEE International Symposium on Biomedical Imaging, 2025
Publisher
IEEE