A Rao-Blackwellized Particle Filter for Superelliptical Extended Target Tracking
Abstract
In this work, we propose a new method to track extended targets of different shapes such as ellipses, rectangles and rhombi. We provide an analytical framework to express these shapes as superelliptical contours and propose a Bayesian filtering scheme that can handle measurements from the contour of the object. The method utilizes the Rao-Blackwellized particle filtering algorithm with novel sensor-object geometry constraints. The success of the algorithm is demonstrated using both simulations and real-data experiments, and the algorithm has been demonstrated to be of high performance in various challenging scenarios.
Index Terms:
Extended Target Tracking, Elliptical Target Tracking, Particle Filtering, Superellipse, Lamé CurveI Introduction
With the ever-growing number of applications involving closer target-to-sensor proximity and high-resolution sensors, targets can occupy multiple sensor resolution cells, generating multiple measurements. This motivates the extended target tracking (ETT) problem where both the target extent (shape) and the kinematic states are to be estimated [1] from these multiple measurements. ETT algorithms can be used in crowd surveillance, maritime vessel tracking, and tracking both vehicles and pedestrians [1, 2]. The target extent can be modeled as elliptical [3, 4, 5], rectangular [6] or star-convex [7, 8].
The random matrix models (RMM) [9, 3, 10, 11] are commonly used in ETT algorithms. The underlying assumption is that measurements are distributed across the body of the target. However, this assumption limits their direct use in scenarios with contour measurements [12], such as those from Lidar sensors. For these cases, algorithms such as the Gaussian process model (GP) [7, 13] and random hypersurface model (RHS) [14, 8] can be used to model complex contours as well as complex surfaces. One approach in [15] is to use multiple ellipses to represent the target extent when the measurements are generated from the target contour.
Another approach for ETT with Lidar measurements is to use geometric object descriptions where the sensor-object geometry can be integrated naturally. In [16], a Gaussian mixture (GM) model probability hypothesis density filter is used to model elliptical and rectangular contours as GMs, where the sensor-object geometry is used to solve the measurement association problem. The work by [17] uses a particle filter approach for ETT of a known shape where the expected measurements are obtained using ray casting. The virtual measurement model (VMM) introduced in [18] tracks an elliptical contour using virtual measurements generated via ray tracing. However, the VMM’s virtual measurement generation is computationally expensive, so a later work [19] trained a Gaussian process regression model to predict the VMM’s adaptation law, eliminating the need for virtual measurement generation during runtime.
In this paper, we propose using contours known as superellipses or Lamé curves [20] to represent the target extent, which are a family of curves containing rhombi, ellipses and more rectangular shapes. We choose the Rao-Blackwellized particle filter (RBPF) [21] as our inference method to explore the potential of using superellipses in extended target tracking. Furthermore, we incorporate the sensor-object geometry into our inference framework analytically, overcoming the challenges of tracking with partially observed contour measurements frequently encountered in Lidar sensor applications.
II Problem Formulation
II-A Superellipses, Extent State and Measurement Model
![Refer to caption](x1.png)
![Refer to caption](x2.png)
![Refer to caption](x3.png)
![Refer to caption](x4.png)
An ellipse in can be expressed as the set of points that satisfy the equation
(1) |
where is called the centroid of the ellipse and is a symmetric positive-definite matrix. Due to its positive-definiteness, can be expressed as , where is a diagonal matrix of positive half-lengths along major and minor axes, respectively, and is the rotation matrix by radians in the counterclockwise direction. With these parameters, (1) can be rewritten as
(2) |
which implies
(3) |
where denotes the or the Euclidean norm. One generalization of (3) can be obtained by not restricting it to the norm, which yields
(4) |
where the norm of a vector is defined as
(5) |
for . Note that (4) has a solution for all , although they do not define a norm for . Figure 1 shows the solution sets to (4) for different values of . These mathematical objects are called Lamé ovals or superellipses when [20], and have found use in various domains of science, engineering and architecture [22, 23, 24, 25].111With an abuse of nomenclature, we refer to Lamé curves with all as superellipses.
Consider the variables and , which are defined as
(6) |
and
(7) |
respectively. With these definitions, we can write (4) as
(8) |
where the absolute value and the exponentiation on are interpreted element-wise. Note that ’s are strictly positive due to (7).
We can use equation (8) to define a non-linear and implicit equation to describe measurements taken from the contour of a superelliptical object. When a measurement lies exactly on the object contour, we have
(9) |
satisfied, where
(10) |
Changing the value of changes the contour shape, as illustrated in Figure 1.
II-B Analytical Sensor-Object Geometry Relations
One problem with extent estimation with contour measurements occurs when the sensor only takes measurements from one side of the target, possibly causing the estimated extent along the unobserved side to deteriorate. In this section, we derive some analytical conditions for determining which axes of the target are visible. These conditions can be incorporated into the filtering framework to update only the observed extension state estimate.
![Refer to caption](x5.png)
![Refer to caption](x6.png)
![Refer to caption](x7.png)
![Refer to caption](x8.png)
![Refer to caption](x9.png)
In relation to the target, the sensor can be located in eight different regions as described by tangent lines to the target contour when , located on its four sides. The boundaries of these eight regions are described by the equations
(11a) | ||||
(11b) |
where is the unit vector along the -th dimension. Each equation in (11) result in two pairs of lines parallel to each other and perpendicular to the other two, as shown in Figure 2 with dotted lines. When the sensor is in regions R2, R4, R6 and R8, both axes of the target are visible, and hence extensions along both axes can be inferred successfully. However, only the extension along the major axis can be inferred when the sensor is in regions R1 and R5, and that along the minor axis (i.e. ) can be inferred when it is in regions R3 and R7. If the sensor location is denoted with , then its location relative to the target can be computed according to . Then, the condition the sensor satisfies in a region where the extension along the dimension is visible becomes
(12a) | ||||
(12b) |
These boundaries can be extended by a margin to account for the limited angular sensor resolution as
(13a) | ||||
(13b) |
which push the boundaries in (11) outwards from the target by along the relevant dimensions. For future use, we define the binary variables
(14) |
to describe the visibility of the object extension along the -th axis, where is the indicator function obtaining the value 1 if the condition in the parentheses is met and 0 otherwise, and when and vice versa.
Finally, we note that these conditions are valid for sufficiently rectangular targets, namely those that can be modeled with a high value in the superellipse equation (4). In the case of lower , especially when , the regions as well as observable properties of the extent contour change.
II-C Implicit Measurement with Scale Constraints
Using the implicit equation (9) to find a suitable superellipse fitting a set of measurements requires minimizing the total deviation
(15) |
Four alternative superellipses that could minimize (15) are shown in Figure 3, where measurements are taken from two sides of the contour in Figure 3a. Aside from the original superellipse, the other three are also viable options in the sense of (15), as they also have the measurements approximately along their contours in relation to their extent parameters. Notice that if we had measurements from all four sides of the vehicle, the bad fits in Figures 3b and 3c would not occur, while the one in Figure 3d can still occur if the algorithm is initialized poorly.
To prevent the algorithm from converging towards estimates such as in Figures 3b and 3c, we suggest enforcing additional scale constraints based on the transformed measurements . Assuming no occlusions and sufficiently high sensor resolution, the contour measurements are generated along the full length of each visible side. This means that the transformed measurements will lie exactly in the interval in their -th dimension. The limits of the interval are attained regardless of the value of . That is, for each dimension we have the constraints
(16a) | ||||
(16b) |
Lastly, it is important to note that each scaling constraint should only be applied if the relevant extension is visible by the sensor according to (13).
III ETT with Superelliptical Extents
In this section, we introduce our state-space model and our inference method based on the Rao-Blackwellized particle filter (RBPF). In the first subsection, we derive the filtering equations for a superelliptical extent model with a known value and extend the formulation to the unknown case in the following subsection.
Let the state-space model with discrete time index have the unknown state vector
(17) |
where and are the time derivatives of the centroid. The state dynamics and the measurement equation of the state-space model are defined as
(18a) | ||||
(18b) | ||||
with measurements , where | ||||
(18c) |
is the state transition matrix, is the process noise and is the additive pseudomeasurement noise accounting for the mismatch between the predicted contour and the obtained measurements, describing how much discrepancy from the superellipse equation as given in (9) is tolerated.
Consider the filtering problem where we are interested in computing the posterior density . Following the RBPF framework, we partition our state variable into the linear state and nonlinear state , and decompose their joint posterior as
(19a) | |||
The second factor, i.e. the posterior density of nonlinear states, is approximated by an empirical density | |||
(19b) | |||
with sample trajectories and weights . This approximation allows the conditional density of the linear state given the nonlinear state to be expressed analytically. Under Gaussianity assumptions, the conditional posterior density of the linear state becomes | |||
(19c) |
where the parameters and are the conditional posterior mean and covariance, respectively, which can be computed using a Kalman filter [26] run conditionally on the nonlinear state.
III-A Superelliptical Extent Model with Known
Let denote our nonlinear state, while is assumed to be known. Conditioned on , the measurement equation in (18b) is linear in the remaining states . Therefore, we partition [27] the state dynamics equation (18a) as
(20a) | ||||
(20b) |
with
(21a) | |||||
(21b) | |||||
(21c) | |||||
(21d) |
where is the vector of angle and acceleration noises of the constant velocity model, is the extension process noise, denotes the identity matrix and denotes the Kronecker product.
In the RBPF framework, the linear state measurement update (KF-MU) and time update (KF-TU) steps are performed analytically conditioned on particles representing the nonlinear states. For the nonlinear states, the weight update (PF-WU) and sampling (PF-S) steps are performed. We introduce the details of these steps in the following. A pseudocode for this filter is given in Algorithm 1.
PF-WU
The weight update of the particles can be expressed as a multiplication of the previous weights with corresponding incremental weights, denoted as . Using (18b), the incremental pseudomeasurement update weights are
(22) |
where is the pseudomeasurement covariance parameter.
If the extension along the -th axis is deemed to be visible according to (13), its corresponding scale constraints (16) can be applied as a soft constraint using Gaussian factors. For each dimension the corresponding scale constraint in (16) is expressed as
(23a) | ||||
(23b) | ||||
where is the variance parameter for the scaling constraint, and is calculated using from the relation between and according to (7). The corresponding binary variables for the sensor-object geometry criteria along each dimension are calculated according to (14). The combined scale constraint for both axes including the sensor-object geometry criteria can be expressed as | ||||
(23c) |
which results in the weight update
(24) |
where is the proposal density from which the particles are sampled. Finally, to ensure the positivity of the extent variables, the weights of the particles with negative are set to zero.
KF-MU
Conditional on the nonlinear states, the measurement equation in (18b) can be compactly written as
(25a) | |||
where denotes the matrix of ones, and | |||
(25b) |
where . Then, omitting the dependence of on and for notational brevity, the Kalman filter measurement update equations can be written as
(26a) | ||||
(26b) | ||||
where | ||||
(26c) | ||||
(26d) | ||||
(26e) |
Notice that (26e) would be the standard Kalman gain, except for the term with two diagonal entries given in (14). This matrix, which we refer to as the Kalman gain mask, multiplies the rows of the original Kalman gain matrix by zero if the corresponding extent is not visible. This allows the mean and variance of that extent variable to remain unchanged after the update equations in (26a) and (26b). While such an update results in a drift from a fully Bayesian approach, it enables analytical incorporation of the sensor-object geometry constraints into our filtering framework easily.
PF-S
To perform the particle filter time update step, we draw new particles from the proposal density
(27) |
If the proposal density is selected as the transition density , the samples will be drawn from the following Gaussian proposal
(28) |
KF-TU
Finally, the Kalman filter time update is performed in accordance with the dynamics in (20b), as
(29a) | ||||
(29b) |
III-B Superelliptical Extent Model with Unknown
The algorithm can be easily modified to estimate the unknown shape of an object by defining the superellipse exponent as an unknown variable. This allows the algorithm to adapt to the target shape with a suitable superellipse. In this case, the nonlinear state vector can be augmented with , i.e. , while the linear state is kept the same. Then, the nonlinear state dynamics parameters in (21) are modified to be
(30a) | ||||
(30b) | ||||
(30c) |
with the variable assumed to be slowly varying together with the strict positivity constraint.
IV Experimental Results
The performance of the algorithm is evaluated in four different simulation scenarios and one scenario with real data. In the scenarios where is known, the nonlinear states are sampled from the Gaussian with
(31a) | ||||
(31b) | ||||
where are the center coordinates of the bounding box containing the first set of measurements and is the velocity estimate obtained via the means of the first two sets of measurements. We follow this two-step initialization approach for the velocity component to improve the particle efficiency. In the known shape scenarios, the superellipse exponent is selected as . Each particle’s conditional mean and variance are set to | ||||
(31c) |
for where is the number of particles. Notice that this selection of initializes the extent half-lengths as and .
Throughout all simulations, the process noise standard deviations are selected as , and , while the proposal density is chosen as in (28). The extension observation conditions in (13) are used with . The implicit measurement covariance in (22) and scale constraint covariance in (23) are selected as .
IV-A Simulation Results
We test the performance of our algorithm in the following four simulated scenarios.
- Linear,
-
where the target follows a linear trajectory along its main axis without drifting, starting from (-40, -10) and ending at (35, -10) with a linear velocity of ,
- Curved,
-
where the target follows a similar trajectory starting from (-40, -10), except for a slight turn towards the positive direction (Figure 4a),
- Drifting,
-
where the trajectory is similar to that of the Curved scenario, except the target also starts drifting while turning (Figure 4b), and
- U-turn,
-
where the target follows a semicircle with a radius of centered around the sensor between brief periods of linear motion, with a linear velocity of (Figure 4c).
![Refer to caption](x10.png)
![Refer to caption](x11.png)
![Refer to caption](x12.png)
Roughly ordered in an increasingly challenging order, each of these scenarios serves to test the performance of the algorithm in different respects. The first two are relatively simple scenarios to demonstrate elementary performance. The latter two illustrate performance in cases where the second extension of the target is not visible for a significant amount of time, testing the effectiveness of gain masking in (26d).
Throughout the simulations, the sampling time is chosen as . The underlying true shape is a superellipse with and half-lengths and . The measurement noise is simulated by adding zero-mean Gaussian noises to the range and bearing of the noise-free Lidar measurements, with the standard deviations set to and , respectively. The Lidar resolution is chosen to be with a field of view and it is assumed to be positioned at the origin.
Scenario | Linear | Curved | Drifting | U-turn | |
---|---|---|---|---|---|
RMSE | 0.3302 | 0.3694 | 0.3249 | 0.3728 | |
0.2744 | 0.3217 | 0.2978 | 0.3145 | ||
0.2139 | 0.2247 | 0.2032 | 0.1383 | ||
0.1340 | 0.2193 | 0.1757 | 0.3055 | ||
IOU | 0.8297 | 0.8202 | 0.6846 | 0.7609 | |
Time | 0.2023 | 0.2280 | 0.2314 | 0.3363 |
We use root mean square error (RMSE) and Intersection-Over-Union (IOU) [28] as our performance metrics over Monte Carlo runs. RMSEs of the state estimates are
(32) |
where denotes the true state and is its estimate, calculated according to
(33) |
Velocity and centroid errors are calculated using their respective Euclidean norms. To account for the performance in extent estimation, we calculate IOU according to
(34) |
where denotes the posterior superellipse region described by the posterior extent state in Monte Carlo run , and is the true extent of the target.
IV-A1 Known Shape Scenarios
Sample outputs of the algorithm with a known shape, i.e. a known value, are given in Figure 4 while the performance metrics and computation times are given in Table I. Note that the RMSEs are comparable in the U-turn and Drifting scenarios, even when one side of the target is invisible throughout most of the simulation. The half-length state corresponding to , unobservable during these scenarios, also exhibits sufficiently small RMSE values in these scenarios, indicating that the corresponding state estimate has not deteriorated. These show the effectiveness of the Kalman gain mask , preventing erroneous updates to the extension estimates and preserving the centroid estimate.
IV-A2 Unknown Shape Scenario
To illustrate the performance of the algorithm for targets with unknown shapes, i.e. unknown , we run the algorithm with the augmented state in Section III-B in three elementary scenarios. In these scenarios, targets follow a linear trajectory with constant speed over time steps. The state is initialized by sampling from the prior with its process noise standard deviation set to , while particles are used to compensate for the increase in the nonlinear state dimension. All other parameters are left identical. The results are presented in Figure 5. It is seen that the algorithm can find a superellipse that accommodates the obtained measurements. The estimated centroids also appear to reflect a suitable trajectory for the target.
IV-B Real Data Application
Finally, we illustrate the performance of our algorithm on real data. For this purpose, we use the data presented in [15], where a Lidar sensor with sampling time and angular resolution is used to collect contour measurements from a moving vehicle whose rear and right sides are visible. On the same dataset, we present the output of both the known and unknown shape approaches. For the known shape algorithm, the superelliptical exponent is assumed to better represent the target as an automobile, while the unknown shape algorithm is initialized again with for the prior of the state . The outputs of the algorithms are shown in Figure 6. Both algorithms successfully track the vehicle and estimate its velocity, and the algorithm with the adaptable target shape finds a suitable superellipse although initialized with an elliptical prior.
![Refer to caption](x13.png)
![Refer to caption](x14.png)
![Refer to caption](x15.png)
V Conclusion
In this paper, we have proposed a new ETT algorithm for contour measurements which represents the target shape as a family of curves known as superellipses. Through the variation of a single exponent parameter, the algorithm unifies a variety of shapes such as diamonds, ellipses and rectangles with varying degrees of rounded corners in a single, implicit measurement equation. We have used this model in a Rao-Blackwellized particle filter, with scaling constraints to prevent particles from reaching local minima of the implicit measurement model proposed and analytical sensor-object geometry relations by modifying the Kalman filter equations. Using both simulations and real-world data, we have shown that the algorithm has high tracking performance, as well as being able to estimate the unknown shape of the target.
References
- [1] K. Granstrom, M. Baum, and S. Reuter, “Extended object tracking: Introduction, overview and applications,” arXiv preprint arXiv:1604.00970, 2016.
- [2] K. Granström and M. Baum, “A tutorial on multiple extended object tracking,” Authorea Preprints, 2023.
- [3] M. Feldmann, M. Feldmann, D. Fränken, D. Franken, W. Koch, and W. Koch, “Tracking of extended objects and group targets using random matrices,” IEEE Transactions on Signal Processing, 2011.
- [4] S. Yang and M. Baum, “Tracking the orientation and axes lengths of an elliptical extended object,” IEEE Transactions on Signal Processing, vol. 67, no. 18, pp. 4720–4729, 2019.
- [5] S. Steuernagel, K. Thormann, and M. Baum, “Improved extended object tracking with efficient particle-based orientation estimation,” in 2023 26th International Conference on Information Fusion (FUSION). IEEE, 2023, pp. 1–8.
- [6] K. Granström, S. Reuter, D. Meissner, and A. Scheel, “A multiple model PHD approach to tracking of cars under an assumed rectangular shape,” in 17th International Conference on Information Fusion (FUSION). IEEE, 2014, pp. 1–8.
- [7] N. Wahlström and E. Özkan, “Extended target tracking using Gaussian processes,” IEEE Transactions on Signal Processing, vol. 63, no. 16, pp. 4165–4178, 2015.
- [8] M. Baum and U. D. Hanebeck, “Extended object tracking with random hypersurface models,” IEEE Transactions on Aerospace and Electronic systems, vol. 50, no. 1, pp. 149–159, 2014.
- [9] J. W. Koch, “Bayesian approach to extended object and cluster tracking using random matrices,” IEEE Transactions on Aerospace and Electronic Systems, vol. 44, no. 3, pp. 1042–1059, 2008.
- [10] K. Granström and U. Orguner, “New prediction for extended targets with random matrices,” IEEE Transactions on Aerospace and Electronic Systems, vol. 50, no. 2, pp. 1577–1589, 2014.
- [11] B. Tuncer and E. Özkan, “Random matrix based extended target tracking with orientation: A new model and inference,” IEEE Transactions on Signal Processing, vol. 69, pp. 1910–1923, 2021.
- [12] J. S. Fowdur, M. Baum, and F. Heymann, “An elliptical principal axes-based model for extended target tracking with marine radar data,” in 2021 IEEE 24th International Conference on Information Fusion (FUSION). IEEE, 2021, pp. 1–8.
- [13] M. Kumru and E. Özkan, “3D extended object tracking using recursive Gaussian processes,” in 2018 21st International Conference on Information Fusion (FUSION). IEEE, 2018, pp. 1–8.
- [14] M. Baum and U. D. Hanebeck, “Random hypersurface models for extended object tracking,” in 2009 IEEE International Symposium on Signal Processing and Information Technology (ISSPIT). IEEE, 2009, pp. 178–183.
- [15] B. Tuncer, U. Orguner, and E. Özkan, “Multi-ellipsoidal extended target tracking with variational Bayes inference,” IEEE Transactions on Signal Processing, vol. 70, pp. 3921–3934, 2022.
- [16] K. Granström, C. Lundquist, and U. Orguner, “Tracking rectangular and elliptical extended targets using laser measurements,” in 14th International Conference on Information Fusion. IEEE, 2011, pp. 1–8.
- [17] M. Michaelis, P. Berthold, T. Luettel, and H.-J. Wuensche, “Extended target tracking with a particle filter using state dependent target measurement models,” in 2022 25th International Conference on Information Fusion (FUSION). IEEE, 2022, pp. 1–8.
- [18] P. Hoher, S. Wirtensohn, T. Baur, J. Reuter, F. Govaers, and W. Koch, “Extended target tracking with a lidar sensor using random matrices and a virtual measurement model,” IEEE Transactions on Signal Processing, vol. 70, pp. 228–239, 2021.
- [19] P. Hoher, J. Reuter, D. Dold, D. Griesser, F. Govaers, and W. Koch, “Extended target tracking with a lidar sensor using random matrices and a Gaussian processes regression model,” in 2023 26th International Conference on Information Fusion (FUSION), 2023, pp. 1–8.
- [20] N. T. Gridgeman, “Lamé Ovals,” The Mathematical Gazette, vol. 54, no. 387, pp. 31–37, 1970. [Online]. Available: http://www.jstor.org/stable/3613154
- [21] A. Doucet, S. Godsill, and C. Andrieu, “On sequential Monte Carlo sampling methods for Bayesian filtering,” Statistics and computing, vol. 10, pp. 197–208, 2000.
- [22] L. Gong, S. D. Pathak, D. R. Haynor, P. S. Cho, and Y. Kim, “Parametric shape modeling using deformable superellipses for prostate segmentation,” IEEE transactions on medical imaging, vol. 23, no. 3, pp. 340–349, 2004.
- [23] D. Köhntopp, B. Lehmann, D. Kraus, and A. Birk, “Classification and localization of naval mines with superellipse active contours,” IEEE Journal of Oceanic Engineering, vol. 44, no. 3, pp. 767–782, 2018.
- [24] M. Duchemin, C. Tugui, and V. Collee, “Optimization of contact profiles using super-ellipse,” SAE International Journal of Materials and Manufacturing, vol. 10, no. 2, pp. 234–244, 2017.
- [25] J. Allen, N. Kundtz, D. A. Roberts, S. A. Cummer, and D. R. Smith, “Electromagnetic source transformations using superellipse equations,” Applied Physics Letters, vol. 94, no. 19, 2009.
- [26] R. E. Kalman, “A New Approach to Linear Filtering and Prediction Problems,” Journal of Basic Engineering, vol. 82, no. 1, pp. 35–45, 03 1960. [Online]. Available: https://doi.org/10.1115/1.3662552
- [27] T. Schon, F. Gustafsson, and P.-J. Nordlund, “Marginalized particle filters for mixed linear/nonlinear state-space models,” IEEE Transactions on signal processing, vol. 53, no. 7, pp. 2279–2289, 2005.
- [28] B. Alexe, T. Deselaers, and V. Ferrari, “Measuring the objectness of image windows,” IEEE transactions on pattern analysis and machine intelligence, vol. 34, no. 11, pp. 2189–2202, 2012.