Keywords: Object detection, Imbalanced dataset, Quadratic optimization, Fetal anatomy detection, Ultrasound.
TL;DR: We suggest a new method, based on quadratic optimization, to tackle the issue of imbalanced datasets in object detection problems.
Abstract: In this paper, we propose a novel approach to overcome the problem of imbalanced datasets for object detection tasks, when the distribution is not uniform over all classes. The general idea is to compute a probability vector, encoding the probability for each image to be fed to the network during the training phase. This probability vector is computed by solving some quadratic optimization problem and ensures that all classes are seen with similar frequency. We apply this method to a fetal anatomies detection problem, and conduct a thorough statistical analysis of the resulting performance to show that it performs significantly better than two baseline models: one with images sampled uniformly and one implementing classical oversampling.
Registration: I acknowledge that publication of this at MIDL and in the proceedings requires at least one of the authors to register and present the work during the conference.
Authorship: I confirm that I am the author of this work and that it has not been submitted to another publication before.
Paper Type: both
Source Latex: zip
Primary Subject Area: Detection and Diagnosis
Secondary Subject Area: Application: Other