InfantFace: Detecting infant faces in neonatal clinical environments

· Source: Computer Vision and Pattern Recognition · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision · Depth: Advanced, quick

Summary

InfantFace is a one-stage YOLOv11m-based model designed for detecting infant faces in challenging neonatal clinical environments. This model addresses significant accuracy issues faced by general face detectors due to cluttered backgrounds, poor lighting, and obstructions from medical equipment, which hinder critical non-contact assessments like pain analysis and cardiorespiratory monitoring. InfantFace was initially trained using a combination of publicly available datasets including VGGFace2 and WIDER FACE. Before fine-tuning, it achieved an AP50 of 0.87, surpassing three general face detectors. Its performance further improved to an AP50 of 0.96 after domain adaptation using a neonatal research dataset comprising 228 videos from 114 recording sessions of 113 independent infants. The authors highlight the urgent need for more publicly available neonatal datasets, emphasizing privacy and ethical considerations.

Key takeaway

For Computer Vision Engineers developing non-contact neonatal monitoring systems, InfantFace demonstrates that domain-specific fine-tuning of models like YOLOv11m is essential for achieving high accuracy in challenging clinical environments. You should prioritize creating or acquiring specialized neonatal datasets, ensuring robust privacy safeguards and ethical standards, to overcome the limitations of general face detectors and enable reliable applications such as pain scoring and breathing alerts. This approach will significantly improve the robustness of your clinical vision systems.

Key insights

InfantFace, a YOLOv11m-based model, significantly improves infant face detection in complex neonatal clinical environments through domain-specific fine-tuning.

Principles

Method

A one-stage YOLOv11m model was trained on combined public datasets (VGGFace2, CelebA, FDDB, WIDER FACE), then fine-tuned using a specialized neonatal research video dataset for clinical domain adaptation.

In practice

Topics

Best for: AI Scientist, Computer Vision Engineer, Research Scientist

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computer Vision and Pattern Recognition.