Implementation of resource-efficient fetal echocardiography detection algorithms in edge computing
Recent breakthroughs in medical AI have proven the effectiveness of deep learning in fetal echocardiography. However, the limited processing power of edge devices hinders real-time clinical application. We aim to pioneer the future of intelligent echocardiography equipment by enabling real-time recognition and tracking in fetal echocardiography, ultimately assisting medical professionals in their practice. Our study presents the YOLOv5s_emn (Extremely Mini Network) Series, a collection of resource-efficient algorithms for fetal echocardiography detection. Built on the YOLOv5s architecture, these models, through backbone substitution, pruning, and inference optimization, while maintaining high accuracy, the models achieve a significant reduction in size and number of parameters, amounting to only 5%-19% of YOLOv5s. Tested on the NVIDIA Jetson Nano, the YOLOv5s_emn Series demonstrated superior inference speed, being 52.8-125.0 milliseconds per frame(ms/f) faster than YOLOv5s, showcasing their potential for efficient real-time detection in embedded systems.
Authors
- Yi Gao (ORCID: https://orcid.org/0000-0001-6343-4138)
- Mei Li (ORCID: https://orcid.org/0000-0001-6547-4424)
- Kun Wang (ORCID: https://orcid.org/0000-0003-3282-4330)
- Yuchen Zhu
- Meng Wang (ORCID: https://orcid.org/0009-0009-4751-7852)
Institutions
- China University of Geosciences (Beijing) (CN)
- Hospital of Hebei Province (CN)
- First Hospital of Shijiazhuang (CN)
Publication Details
- Journal
- PLoS ONE
- Published
- 2024-09-23
- DOI
- https://doi.org/10.1371/journal.pone.0305250
- Citations
- 1
- Primary Topic
- Congenital Heart Disease Studies
- Type
- article
- Field-Weighted Citation Impact
- 0.39