Fast Training of Object Detection Using Stochastic Gradient Descent

• Main Authors: Wijnhoven, R G J, de With, P H N
• Format: Conference Proceeding
• Language: English
• Subjects: classification; Computer vision; detection; Feature extraction; histogram of oriented gradients; HOG; Object detection; object recognition; Optimization; Pattern recognition; stochastic gradient descent; Support vector machines; SVM; Training
• ISSN: 1051-4651; 2831-7475
• DOI: 10.1109/ICPR.2010.112

Training datasets for object detection problems are typically very large and Support Vector Machine (SVM) implementations are computationally complex. As opposed to these complex techniques, we use Stochastic Gradient Descent (SGD) algorithms that use only a single new training sample in each iteration and process samples in a stream-like fashion. We have incorporated SGD optimization in an object detection framework. The object detection problem is typically highly asymmetric, because of the limited variation in object appearance, compared to the background. Incorporating SGD speeds up the optimization process significantly, requiring only a single iteration over the training set to obtain results comparable to state-of-the-art SVM techniques. SGD optimization is linearly scalable in time and the obtained speedup in computation time is two to three orders of magnitude. We show that by considering only part of the total training set, SGD converges quickly to the overall optimum.