Uncertainty-driven active developmental learning

doi:10.1016/j.patcog.2024.110384

Return

Uncertainty-driven active developmental learning

2024-07-01

PRE

DOI：10.1016/j.patcog.2024.110384

Original

Original request for help

Save

Abstract

En 中文

Existing machine learning models can well handle common classes but struggle to detect unfamiliar or unknown classes due to environmental variations. To address this challenge, we propose a new task called active developmental learning (ADL), which empowers models to actively determine what to learn in the open world, thereby progressively enhancing the capability of detecting unfamiliar and unknown classes. Considering the uncertain essence of the task, we design an uncertainty -driven method for ADL (UADL) that measures and utilizes uncertainty to evaluate unfamiliar known classes and unknown classes separately, which consists of two stages: (1) unfamiliar detection of known classes and (2) unknown detection of novel classes. In the first stage, UADL identifies unfamiliar samples of known classes via known -class uncertainty calculated by GMMs on detectors' heads. In the second stage, UADL identifies samples containing unknown classes via unknown -class uncertainty computed by class -specific GMMs in feature space. In both stages, uncertainty is used to select a minimal number of unlabeled samples for manual labeling, facilitating the model's active self -development. Experiments on multiple object detection benchmark datasets demonstrate the feasibility and significant performance of UADL and show its effectiveness against the ADL task compared to other state-of-the-art approaches.

Keywords:

Object detection

Active developmental learning

Uncertainty estimation