Beyond Correlation Filters: Learning Continuous Convolution Operators for Visual Tracking
Martin Danelljan, Andreas Robinson, Fahad Shahbaz Khan, Michael Felsberg
Introduction
Visual tracking is the task of estimating the trajectory of a target in a video. It is one of the fundamental problems in computer vision. Tracking of objects or feature points has numerous applications in robotics, structure-from-motion, and visual surveillance. In recent years, Discriminative Correlation Filter (DCF) based approaches have shown outstanding results on object tracking benchmarks . DCF methods train a correlation filter for the task of predicting the target classification scores. Unlike other methods, the DCF efficiently utilize all spatial shifts of the training samples by exploiting the discrete Fourier transform.
Deep convolutional neural networks (CNNs) have shown impressive performance for many tasks, and are therefore of interest for DCF-based tracking. A CNN consists of several layers of convolution, normalization and pooling operations. Recently, activations from the last convolutional layers have been successfully employed for image classification. Features from these deep convolutional layers are discriminative while preserving spatial and structural information. Surprisingly, in the context of tracking, recent DCF-based methods have demonstrated the importance of shallow convolutional layers. These layers provide higher spatial resolution, which is crucial for accurate target localization. However, fusing multiple layers in a DCF framework is still an open problem.
The conventional DCF formulation is limited to a single-resolution feature map. Therefore, all feature channels must have the same spatial resolution, as in e.g. the HOG descriptor. This limitation prohibits joint fusion of multiple convolutional layers with different spatial resolutions. A straightforward strategy to counter this restriction is to explicitly resample all feature channels to the same common resolution. However, such a resampling strategy is both cumbersome, adds redundant data and introduces artifacts. Instead, a principled approach for integrating multi-resolution feature maps in the learning formulation is preferred.
In this work, we propose a novel formulation for learning a convolution operator in the continuous spatial domain. The proposed learning formulation employs an implicit interpolation model of the training samples. Our approach learns a set of convolution filters to produce a continuous-domain confidence map of the target. This enables an elegant fusion of multi-resolution feature maps in a joint learning formulation. Figure 1 shows a visualization of our continuous convolution operator, when integrating multi-resolution deep feature maps. We validate the effectiveness of our approach on three object tracking benchmarks: OTB-2015 , Temple-Color and VOT2015 . On the challenging OTB-2015 with 100 videos, our object tracking framework improves the state-of-the-art from to in mean overlap precision.
In addition to multi-resolution fusion, our continuous domain learning formulation enables accurate sub-pixel localization. This is achieved by labeling the training samples with sub-pixel precise continuous confidence maps. Our formulation is therefore also suitable for accurate feature point tracking. Further, our learning-based approach is discriminative and does not require explicit interpolation of the image to achieve sub-pixel accuracy. We demonstrate the accuracy and robustness of our approach by performing extensive feature point tracking experiments on the popular MPI Sintel dataset .
Related Work
Discriminative Correlation Filters (DCF) have shown promising results for object tracking. These methods exploit the properties of circular correlation for training a regressor in a sliding-window fashion. Initially, the DCF approaches were restricted to a single feature channel. The DCF framework was later extended to multi-channel feature maps . The multi-channel DCF allows high-dimensional features, such as HOG and Color Names, to be incorporated for improved tracking. In addition to the incorporation of multi-channel features, the DCF framework has been significantly improved lately by, e.g., including scale estimation , non-linear kernels , a long-term memory , and by alleviating the periodic effects of circular convolution .
With the advent of deep CNNs, fully connected layers of the network have been commonly employed for image representation . Recently, the last (deep) convolutional layers were shown to be more beneficial for image classification . On the other hand, the first (shallow) convolutional layer was shown to be more suitable for visual tracking, compared to the deeper layers . The deep convolutional layers are discriminative and possess high-level visual information. In contrast, the shallow layers contain low-level features at high spatial resolution, beneficial for localization. Ma et al. employed multiple convolutional layers in a hierarchical ensemble of independent DCF trackers. Instead, we propose a novel continuous formulation to fuse multiple convolutional layers with different spatial resolutions in a joint learning framework.
Unlike object tracking, feature point tracking is the task of accurately estimating the motion of distinctive key-points. It is a core component in many vision systems . Most feature point tracking methods are derived from the classic Kanade-Lucas-Tomasi (KLT) tracker . The KLT tracker is a generative method, that is based on minimizing the squared sum of differences between two image patches. In the last decades, significant effort has been spent on improving the KLT tracker . In contrast, we propose a discriminative learning based approach for feature point tracking.
Our approach: Our main contribution is a theoretical framework for learning discriminative convolution operators in the continuous spatial domain. Our formulation has two major advantages compared to the conventional DCF framework. Firstly, it allows a natural integration of multi-resolution feature maps, e.g. combinations of convolutional layers or multi-resolution HOG and color features. This property is especially desirable for object tracking, detection and action recognition applications. Secondly, our continuous formulation enables accurate sub-pixel localization, crucial in many feature point tracking problems.
Learning Continuous Convolution Operators
In this section, we present a theoretical framework for learning continuous convolution operators. Our formulation is generic and can be applied for supervised learning tasks, such as visual tracking and detection.
In this paper, we utilize basic concepts and results in continuous Fourier analysis. For clarity, we first formulate our learning method for data defined in a one-dimensional domain, i.e. for functions of a single spatial variable. We then describe the generalization to higher dimensions, including images, in section 3.5.
Here, the bar denotes complex conjugation. In (1) we have also defined the circular convolution operation .
2 Our Continuous Learning Formulation
The interpolated sample J_{d}\big{\{}x^{d}\big{\}}(t) is constructed as a superposition of shifted versions of an interpolation function . In (2), the feature values act as weights for each shifted function. Similar to the periodic assumption in the conventional discrete DCF formulation, a periodic extension of the feature map is also performed here in (2).
In our continuous formulation, the operator is parametrized by a set of convolution filters . Here, is the continuous filter for feature channel . We define the convolution operator as,
Here, each feature channel is first interpolated using (2) and then convolved with its corresponding filter. Note that the convolutions are performed in the continuous domain, as defined in (1). In the last step, the convolution responses from all filters are summed to produce the final confidence function.
In the standard DCF, each training sample is labeled by a discrete function that represents the desired convolution output. In contrast, our samples are labeled by confidence functions , defined in the continuous spatial domain. Here, is the desired output of the convolution operator applied to the training sample . This enables sub-pixel accurate information to be incorporated in the learning. The filter is trained, given a set of training sample pairs , by minimizing the functional,
Here, the weights control the impact of each training sample. We additionally include a spatial regularization term, similar to , determined by the penalty function . This regularization enables the filter to be learned on arbitrarily large image regions by controlling the spatial extent of the filter . Spatial regions typically corresponding to background features are assigned a large penalty in , while the target region has small penalty values. Thus, encodes the prior reliability of features depending on their spatial location. Unlike , the penalty function is defined on the whole continuous interval and periodically extended to . Hence, is required in (4). This is implied by our later assumption of having finitely many non-zero Fourier coefficients . Next, we derive the procedure to train the continuous filter , using the proposed formulation (4).
3 Training the Continuous Filter
By applying Parseval’s formula to (4) and using (5), we obtain
Hence, the functional can equivalently be minimized with respect to the Fourier coefficients for each filter . We exploit the Fourier domain formulation (6) to minimize the original loss (4).
For practical purposes, the filter needs to be represented by a finite set of parameters. One approach is to employ a parametric model to represent an infinite number of coefficients. In this work, we instead obtain a finite representation by minimizing (6) over the finite–dimensional subspace . That is, we minimize (6) with respect to the coefficients , while assuming for . In practice, determines the number of filter coefficients to be computed for feature channel during learning. Increasing leads to a better estimate of the filter at the cost of increased computations and memory consumption. In our experiments, we set such that the number of stored filter coefficients for channel equals the spatial resolution of the training sample .
To obtain a simple expression of the normal equations, we define the sample matrix , the diagonal weight matrix and the label vector . The minimizer of (7) is found by solving the normal equations,
Here, denotes the conjugate-transpose of a matrix. Note that (8) forms a sparse linear equation system if has a small number of non-zero Fourier coefficients . In our object tracking framework, presented in section 4.2, we employ the Conjugate Gradient method to iteratively solve (8). For our feature point tracking approach, presented in section 4.3, we use a single-channel feature map and a constant penalty function for improved efficiency. This results in a diagonal system (8), which can be efficiently solved by a direct computation.
4 Desired Confidence and Interpolation Function
Here, we describe the choice of the desired convolution output and the interpolation function . We construct both and by periodically repeating functions defined on the real line. In general, the -periodic repetition of a function is defined as . In the derived Fourier domain formulation (6), the functions and are represented by their respective Fourier coefficients. The Fourier coefficients of a periodic repetition can be retrieved from the continuous Fourier transform of as .3.4 We use this property to compute the Fourier coefficients of and .
To construct the desired convolution output , we let denote the estimated location of the target object or feature point in sample . We define as the periodic repetition of the Gaussian function \exp\big{(}-\frac{(t-u_{j})^{2}}{2\sigma^{2}}\big{)} centered at . This provides the following expression for the Fourier coefficients,
The variance is set to a small value to obtain a sharp peak. Further, this ensures a negligible spatial aliasing. In our work, the functions are constructed based on the cubic spline kernel . The interpolation function is set to the periodic repetition of a scaled and shifted version of the kernel b\big{(}\frac{N_{d}}{T}\big{(}t-\frac{T}{2N_{d}}\big{)}\big{)}, to preserve the spatial arrangement of the feature pyramid. The Fourier coefficients of are then obtained as \hat{b}_{d}[k]=\frac{1}{N_{d}}\exp\big{(}-i\frac{\pi}{N_{d}}k\big{)}\hat{b}\big{(}\frac{k}{N_{d}}\big{)}.Further details are given in the supplementary material.
5 Generalization to Higher Dimensions
The proposed formulation can be extended to domains of arbitrary number of dimensions. For our tracking applications we specifically consider the two-dimensional case, but higher-dimensional spaces can be treated similarly. For images, we use the space of square-integrable periodic functions of two variables . The complex exponentials are then given by . For the desired convolution output , we employ a 2-dimensional Gaussian function. Further, the interpolation functions are obtained as a separable combination of the cubic spline kernel, i.e. . The derivations presented in section 3.3 also hold for the higher dimensional cases.
Our Tracking Frameworks
We apply our continuous learning formulation for two problems: visual object tracking and feature point tracking. We first present the localization procedure, which is based on maximizing the continuous confidence function. This is shared for both the object and feature point tracking frameworks.
Here, the aim is to localize the tracked target or feature point using the learned filter . This is performed by first extracting a feature map from the region of interest in an image. The Fourier coefficients of the confidence score function are then calculated using (5). We employ a two-step approach for maximizing the score on the interval . To find a rough initial estimate, we first perform a grid search, where the score function is evaluated at the discrete locations s\big{(}\frac{Tn}{2K+1}\big{)} for . This is efficiently implemented as a scaled inverse DFT of the non-zero Fourier coefficients . The maximizer obtained in the grid search is then used as the initialization for an iterative optimization of the Fourier series expansion . We employ the standard Newton’s method for this purpose. The gradient and Hessian are computed by analytic differentiation of .
2 Object Tracking Framework
We first present the object tracking framework based on our continuous learning formulation introduced in section 3.2. We employ multi-resolution feature maps extracted from a pre-trained deep network.We use imagenet-vgg-m-2048, available at: http://www.vlfeat.org/matconvnet/. Similar to DCF based trackers , we extract a single training sample in each frame. The sample is extracted from an image region centered at the target location and the region size is set to times the area of the target box. Its corresponding importance weight is set to using a learning rate parameter . The weights are then normalized such that . We store a maximum of samples by replacing the sample with the smallest weight. The Fourier coefficients of the penalty function are computed as described in . To detect the target, we perform a multi-scale search strategy with scales and a relative scale factor . The extracted confidences are maximized using the grid search followed by five Newton iterations, as described in section 4.1.
The training of our continuous convolution filter is performed by iteratively solving the normal equations (8). The work of employed the Gauss-Seidel method for this purpose. However, this approach suffers from a quadratic complexity in the number of feature channels . Instead, we employ the Conjugate Gradient (CG) method due to its computational efficiency. Our numerical optimization scales linearly and is therefore especially suitable for high-dimensional deep features. In the first frame, we use iterations to find an initial estimate of the filter coefficients . Subsequently, iterations per frame are sufficient by initializing CG with the current filter.3.4
3 Feature Point Tracking Framework
Here, we describe the feature point tracking framework based on our learning formulation. For computational efficiency, we assume a single-channel feature map (), e.g. a grayscale image, and a constant penalty function . Under these assumptions, the normal equations (8) form a diagonal system of equations. The filter coefficients are directly obtained as,
Here, we have dropped the feature dimension index for the sake of clarity. In this case (single feature channel and constant penalty function), the training equation (10) resembles the original MOSSE filter . However, our continuous formulation has several advantages compared to the original MOSSE. Firstly, our formulation employs an implicit interpolation model, given by . Secondly, each sample is labeled by a continuous-domain confidence , that enables sub-pixel information to be incorporated in the learning. Thirdly, our convolution operator outputs continuous confidence functions, allowing accurate sub-pixel localization of the feature point. In our experiments, we show that the advantages of our continuous formulation are crucial for accurate feature point tracking.
Experiments
We validate our learning framework for two applications: tracking of objects and feature points. For object tracking, we perform comprehensive experiments on three datasets: OTB-2015 , Temple-Color , and VOT2015 . For feature point tracking, we perform extensive experiments on the MPI Sintel dataset .
We first evaluate the impact of fusing multiple convolutional layers from the deep network in our object tracking framework. Table 1 shows the tracking results, in mean overlap precision (OP) and area-under-the-curve (AUC), on the OTB-2015 dataset. OP is defined as the percentage of frames in a video where the intersection-over-union overlap exceeds a threshold of . AUC is computed from the success plot, where the mean OP over all videos is plotted over the range of thresholds $$. For details about the OTB protocol, we refer to .
In our experiments, we investigate the impact of the input RGB image layer (layer 0), the first convolutional layer (layer 1) and the last convolutional layer (layer 5). No significant gain in performance was observed when adding intermediate layers. The shallow layer (layer 1) alone provides superior performance compared to using only the deep convolutional layer (layer 5). Fusing the shallow and deep layers provides a large improvement. The best results are obtained when combining all three convolutional layers in our learning framework. We employ this three-layer combination for all further object tracking experiments.
We also compare our continuous formulation with the discrete DCF formulation by performing explicit resampling of the feature layers to a common resolution. For a fair comparison, all shared parameters are left unchanged. The layers (0, 1 and 5) are resampled with bicubic interpolation such that the data size of the training samples is preserved. On OTB-2015, the discrete DCF with resampling obtains an AUC score of , compared to for our continuous formulation. This dramatic reduction in performance is largely attributed to the reduced resolution in layer 1. To mitigate this effect, we also compare with only resampling layers 0 and 5 to the resolution of layer 1. This improves the result of the discrete DCF to in AUC, but at the cost of a 5-fold increase in data size. Our continuous formulation still outperforms the discrete DCF as it avoids artifacts introduced by explicit resampling.
2 OTB-2015 Dataset
We validate our Continuous Convolution Operator Tracker (C-COT) in a comprehensive comparison with 20 state-of-the-art methods: ASLA , TLD , Struck , LSHT , EDFT , DFT , CFLB , ACT , TGPR , KCF , DSST , SAMF , MEEM , DAT , LCT , HCF , Staple and SRDCF . We also compare with SRDCFdecon, which integrates the adaptive decontamination of the training set in SRDCF, and DeepSRDCF employing activations from the first convolutional layer.
State-of-the-art Comparison: Table 2 (first row) shows a comparison with state-of-the-art methods on the OTB-2015 dataset.Detailed results are provided in the supplementary material. The results are reported as mean OP over all the 100 videos. The HCF tracker, based on hierarchical convolutional features, obtains a mean OP of . The DeepSRDCF employs the first convolutional layer, similar to our baseline “Layer 1” in table 1, and obtains a mean OP of . Our approach achieves the best results with a mean OP of , significantly outperforming DeepSRDCF by .
Figure 2a shows the success plot on the OTB-2015 dataset. We report the AUC score for each tracker in the legend. The DCF-based trackers HCF and Staple obtain AUC scores of and respectively. Among the compared methods, the SRDCF and its variants SRDCFdecon and DeepSRDCF provide the best results, all obtaining AUC scores above . Overall, our tracker achieves the best results, outperforming the second best method by .
Robustness to Initialization: We evaluate the robustness to initializations using the protocol provided by . Each tracker is evaluated using two different initialization strategies: spatial robustness (SRE) and temporal robustness (TRE). The SRE criteria initializes the tracker with perturbed boxes, while the TRE criteria starts the tracker at 20 frames. Figure 3 provides the SRE and TRE success plots. Our approach obtains consistent improvements in both cases.
3 Temple-Color Dataset
Here, we evaluate our approach on the Temple-Color dataset containing 128 videos. The second row of table 2 shows a comparison with state-of-the-art methods. The DeepSRDCF tracker provides a mean OP score of . MEEM and SRDCFdecon obtain mean OP scores of and respectively. Different from these methods, our C-COT does not explicitly manage the training set to counter occlusions and drift. Our approach still improves the start-of-the-art by a significant margin, achieving a mean OP score of . A further gain in performance is expected by incorporating the unified learning framework to handle corrupted training samples. In the success plot in Figure 2b, our method obtains an absolute gain of in AUC compared to the previous best method.
4 VOT2015 Dataset
The VOT2015 dataset consists of 60 challenging videos compiled from a set of more than 300 videos. Here, the performance is measured both in terms of accuracy (overlap with the ground-truth) and robustness (failure rate). In VOT2015, a tracker is restarted in the case of a failure. We refer to for details. Table 3 shows the comparison of our approach with the top 10 participants in the challenge according to the VOT2016 rules . Among the compared methods, RAJSSC achieves favorable results in terms of accuracy, at the cost of a higher failure rate. EBT achieves the best robustness among the compared methods. Our approach improves the robustness with a reduction in failure rate, without any significant degradation in accuracy.
5 Feature Point Tracking
We validate our approach for robust and accurate feature point tracking. Here, the task is to track distinctive local image regions. We perform experiments on the MPI Sintel dataset , based on the 3D-animated movie “Sintel”. The dataset consists of 23 sequences, featuring naturalistic and dynamic scenes with realistic lighting and camera motion blur. The ground-truth dense optical flow and occlusion maps are available for each frame. Evaluation is performed by selecting approximately 2000 feature points in the first frame of each sequence. We use the Good Features to Track (GFTT) feature selector, but discard points at motion boundaries due to their ambiguous motion. The ground-truth tracks are then generated by integrating flow vectors over the sequence. The flow vectors are obtained by a bilinear interpolation of the dense ground-truth flow. We terminate the ground-truth tracks using the provided occlusion maps.
We compare our approach to MOSSE and KLT . The OpenCV implementation of KLT, used in our experiments, employs a pyramidal search to accommodate for large translations. For a fair comparison, we adopt a similar pyramid approach for our method and MOSSE, by learning an independent filter for each pyramid level. Further, we use the window size of pixels and pyramid levels for all methods. For both our method and MOSSE we use a learning rate of and set the regularization parameter to . For the KLT we use the default settings in OpenCV. Unlike ours and the MOSSE tracker, the KLT tracks feature points frame-to-frame without memorizing earlier appearances. In addition to our standard tracker, we also evaluate a frame-to-frame version (Ours-FF) of our method by setting the learning rate to .
For quantitative comparisons, we use the endpoint error (EPE), defined as the Euclidian distance between the tracked point and its corresponding ground-truth location. Tracked points with an EPE smaller than 3 pixels are regarded as inliers. Figure 4 (left) shows the distribution of EPE computed over all sequences and tracked points. We also report the average inlier EPE for each method in the legend. Our approach achieves superior accuracy, with an inlier error of pixels. We also provide the precision plot (Figure 4, center), where the fraction of points with an EPE smaller than a threshold is plotted. The legend shows the inlier ratio for each method. Our tracker achieves superior robustness in comparison to the KLT, with an inlier ratio of . Compared to MOSSE, our method obtains significantly improved precision at sub-pixel thresholds ( pixel). This clearly demonstrates that our continuous formulation enables accurate sub-pixel feature point tracking, while being robust. Unlike the frame-to-frame KLT, our method provides a principled procedure for updating the tracking model, while memorizing old samples. The experiments show that already our frame-to-frame variant (Ours-FF) provides a spectacular improvement compared to the KLT. Hence, our gained performance is due to both the model update and the proposed continuous formulation. On a desktop machine, our Matlab code achieves real-time tracking of 300 points at a single scale, utilizing only a single CPU.
Conclusions
We propose a generic framework for learning discriminative convolution operators in the continuous spatial domain. We validate our framework for two problems: object tracking and feature point tracking. Our formulation enables the integration of multi-resolution feature maps. In addition, our approach is capable of accurate sub-pixel localization. Experiments on three object tracking benchmarks demonstrate that our approach achieves superior performance compared to the state-of-the-art. Further, our method obtains substantially improved accuracy and robustness for real-time feature point tracking.
Note that, in this work, we do not use any video data to learn an application specific deep feature representation. This is expected to further improve the performance of our object tracking framework. Another research direction is to incorporate motion-based deep features into our framework, similar to .
Acknowledgments: This work has been supported by SSF (CUAS), VR (EMC2), CENTAURO, the Wallenberg Autonomous Systems Program, NSC and Nvidia.