A Deep RL-Based Active Learning Approach for Fetal Ultrasound Image Plane Labeling and Classification
摘要
Deep learning techniques have shown promise in fetal ultrasound image analysis, potentially saving clinicians time and providing better patient care. This study introduces a novel active learning model for fetal ultrasound image datasets, aimed at identifying valuable data for annotation to enhance model performance while reducing the need for extensive labeled data. Unlike traditional heuristic methods, this framework explicitly employs deep reinforcement learning to develop a data selection policy. Features from these images are extracted to represent the state space for the reinforcement learning model using a deep convolutional neural network. A deep Q-learning algorithm trains a Q-network, which outputs actions determining whether to annotate the data. Leveraging Deep Reinforcement Learning (DRL) to learn a data selection policy enables the model to achieve better performance with fewer labeled examples. The results showed 72.9% accuracy in fetal plane labeling within the active learning framework.