Infant Care Video Dataset for Classification of Interventions Using Transformers
Organizations: Department of Systems and Computer Engineering Carleton University Ottawa, Canada
Abstract
Healthcare documentation in the neonatal intensive care unit (NICU) presents significant challenges, with nurses spending approximately 25% of their time on record-keeping, while up to 60% of interventions remain undocumented. Motivated by the need to detect interventions from video automatically, we present the Infant Care Video Dataset (ICVD), a collection of 4,144 videos spanning 12 simulated intervention classes designed for developing automated documentation systems. Our manikin-based approach systematically varies conditions, such as camera angle and clinician skin tone, while ensuring privacy compliance. Using video transformer architectures (TimeSformer and MotionFormer), we establish strong baseline performance (93.97% and 93.17% top-1 accuracy) among the 12 infant care classes. Our ablation study comparing temporal models with a framewise approach (23.17% accuracy) demonstrates a 70.80% performance gap, validating the need for temporal modeling. The ICVD provides a foundation for developing automated documentation systems to reduce clinical burden in neonatal care environments and improve existing practices.
Figures & tables
| Split | Number of videos |
|---|---|
| Total | 4144 |
| Train | 2892 |
| Validation | 624 |
| Test | 631 |
| Classes | Index | Light | Dark | Total |
|---|---|---|---|---|
| Giving Pacifier | 7 | 172 | 162 | 334 |
| Changing Diaper | 10 | 160 | 145 | 305 |
| Taking Pacifier | 5 | 163 | 158 | 321 |
| Feeding | 6 | 163 | 152 | 315 |
| Removing from crib | 1 | 165 | 172 | 337 |
| Replacing infant in crib | 8 | 166 | 173 | 339 |
| Model | Top-1 (%) | Top-5 (%) | Precision (%) | Recall (%) | F1 (%) |
|---|---|---|---|---|---|
| TSF | 93.97 | 100.00 | 94.11 | 93.29 | 92.92 |
| MF | 93.17 | 99.84 | 93.73 | 92.32 | 91.29 |
| FW | 23.17 | 65.87 | 23.08 | 23.00 | 19.45 |
Healthcare
Healthcare AI develops and evaluates models and systems for clinical care, medical research, public health, and healthcare operations.
See all 621 papers