D IVE D EEPER I NTO I NTEGRAL P OSE R EGRESSION
A BSTRACT
Integral pose regression combines an implicit heatmap with end-to-end training for human body and hand pose estimation.
Unlike detection-based heatmap methods, which decode final joint positions from the heatmap with a non-differentiable argmax operation, integral regression methods apply a differentiable expectation operation.
This paper offers a deep dive on the inference and back-propagation of integral pose regression to better understand the differences in performance and training compared to detection-based methods.
For inference, we give theoretical support why expectation should always be better than the argmax operation, i.e. integral regression should always outperform detection. Yet in practice, this is observed only in hard cases because the heatmap activation for regression shrinks in easy cases.
We then experimentally show that the activation shrinkage is one of the leading causes for integral regression’s inferior performance.
For backpropagation, we theoretically and empirically analyze the gradients to explain the slow training speed for integral regression.
Based on these findings, we incorporate the supervision of spatial prior to speed up training and improve performance.
I NTRODUCTION
2D human pose estimation aims to detect the image coordinates of the body and or the hand.
In the last few years, detection-based methods (Newell et al., 2016; Xiao et al., 2018; Sun et al., 2019; Li et al., 2019) and integral pose regression methods (Sun et al., 2018; Iqbal et al., 2018) have emerged as two common paradigms for human pose estimation.
Both detection-based and integral regression methods learn a heatmap representing pixel-wise likelihoods of the joint positions. The heatmap is learned explicitly for detection-based methods but remains implicit or “latent” for integral regression methods.
To decode the heatmaps to joint coordinates, detection methods use an argmax, while integral regression methods take the expected value.
As the expectation operation is differentiable, integral regression has the benefit of being end-to-end learnable, even though detection methods seem more competitive accuracy-wise (COC, 2020).
Gu et al.
(2021)’s
recent work showed a curious performance difference between integral regression and detection methods.
With a nuanced split over the evaluation set, integral regression outperforms detection when test samples are “harder”, i.e. with fewer keypoints present in the scene, under higher occlusion, and lower bounding box resolutions.
Given that both detection and integral regression methods work with fully convolutional feed-forward architectures, this naturally begs the question - what are the reasons behind this performance difference?
Why is it that integral regression, which otherwise seems to lag behind detection methods in performance (COC, 2020), can excel on these hard cases?
This serves as the motivation for our closer study and analysis of integral pose regression.
Detection and integral regression methods differ in two aspects in both the forwards and backwards pass (see Fig. 1 for an overview).
In the forwards pass, the heatmap is decoded with an argmax for detection versus a softmax normalization and expectation for integral regression.
In the backwards pass, detection methods are supervised with an explicitly defined Gaussian heatmap centered on the ground truth joint, while integral regression is supervised by the joint coordinates directly.
Through detailed theoretical analysis and experimentation on the decoding and back-propagation process, we make the following findings and contributions:
We propose a unified model of the heatmap to interpret and compare detection and integral regression methods.
We verify the model experimentally and show that as samples progress from hard to easy, the activation region on the heatmap shrinks for both detection and integral regression methods.
We demonstrate experimentally that degenerately small regions of activation degrade the accuracy of both detection and integral regression methods.
Integral regression methods, which decode the heatmap with an expectation operation, should result in a lower expected end-point-error than detection methods that decode with an argmax operation.
In practice, this can be observed only for hard samples due to the shrinkage of the active region on the heatmap.
Direct supervision with the joint coordinates in integral regression, although end-to-end, suffers from gradient vanishing and provides less spatial cues for learning the heatmap than the explicit heatmap supervision of detection methods.
As a result, the training of integral regression is more inefficient and slower to converge than detection methods.
Our findings serve as insight to understand integral pose regression which has a better theoretical performance and show that density of heatmaps plays an important role on decoding to coordinates.
R ELATED W ORK
Since the concept of heatmaps were proposed in (Tompson et al., 2014), detection-based methods have been top performers for human pose estimation.
Existing detection-based works emphasize extracting high-quality multi-resolution features.
Xiao et al.
(2018) proposed to adapt ResNet with deconvolution layers while Hourglass (Newell et al., 2016) and Cascaded Pyramid Network (CPN) (Li et al., 2019) introduced cascaded network architectures with a coarse-to-fine design paradigm.
HighResolution Network (HRNet) (Sun et al., 2019) follows the coarse-to-fine paradigm and further improves performance by adopting more dense connections across different resolution representations.
The integral regression method of (Sun et al., 2018) introduced a competitive regression-based framework but is exceeded by more advanced detection-based works (Xiao et al., 2018; Sun et al., 2019).
Numerical regression-based methods, which directly regress the joint coordinates, are commonly used in facial land-mark detection (Feng et al., 2018; Zhu et al., 2020).
These methods however are not as accurate as detection-based methods on human pose estimation.
As a follow-up, integral regression tries to implicitly merge the spatial knowledge from the heatmap via a “latent” heatmap.
Integral regression methods are especially preferred in hand pose estimation (Spurr et al., 2020; Yang et al., 2021) but are still less common for human pose estimation (Sun et al., 2018; Nibali et al., 2018).
Recently, two parallel lines of work have been developed to explain the behaviour of detection and integral regression methods respectively.
For detection-based methods, Huang et al.
(2020) revealed a heatmap bias caused by the inconsistency of coordinate system transformations and solved it by redesigning the transformations during data processing.
Moreover, as the predicted heatmaps during inference may violate the Gaussian assumption and worsen the performance, Huang et al.
(2020) proposed to find the optimal Gaussian location based on the prediction while Zhang et al.
(2020) modulates the predicted Gaussian distribution for decoding.
For integral regression methods, Nibali et al.
(2018) experimentally compared different heatmap regularizers, heatmap normalization schemes and loss functions.
They find that using Jensen-Shannon regularization on the heatmap with a softmax normalization and L1 loss achieves the best performance for integral regression.
More recently, Gu et al.
(2021) performed a systematic comparison of detection versus integral regression methods with a common backbone and discovered the performance advantage of integral regression on the “hard” samples.
Furthermore, Gu et al.
(2021) revealed a bias that arises from taking the expectation after the softmax normalization. As such, they proposed a compensation scheme to mitigate the bias, which in turn improves the overall performance of integral regression, making them competitive with detection methods.
P RELIMINARIES ON H UMAN P OSE E STIMATION
In this work, we target the more commonly used ‘top-down’ form of pose estimation in which a person detector already provides a cropped image I of the person.
For simplicity, we focus our discussion on one given joint of the K total joints in the body.
The pose estimation model outputs a heatmap ˆ H ∈ R M × N where M and N are the dimensions of the spatial heatmap.
Typically, M and N are scaled down by a factor of 4 from the original input dimensions of I (Xiao et al., 2018;
Sun et al., 2019).
The heatmap ˆ H represents a (discrete) spatial likelihood P ( J | I ) , where J ∈ R 1 ×is the 2D coordinates of that joint.
In practice, all K heatmaps are predicted simultaneously by the same network, where each joint is one channel.
In both detection and integral regression methods, the coordinates ˆ J are decoded from ˆ H ; where the two methods differ is the manner of decoding (see Sec. 3.1) and the form of supervision applied (see Sec. 3.2).
3. H EATMAP D ECODING : M AX VERSUS E XPECTED V ALUE
Detection Methods apply an argmax on ˆ H indexed by p to estimate the joint coordinates ˆ J de :
[Equation]
Taking an argmax can be interpreted as taking a maximum likelihood on the heatmap ˆ H , assuming that ˆ H is proportional to the likelihood.
In practice, the final ˆ J de value is determined as a linear combination of the highest and second-highest response on ˆ H as a way to account for quantization effects in the discrete heatmap (Newell et al., 2016).
A more recent work, DARK (Zhang et al., 2020) approximates the true prediction by a Taylor series evaluated at the maximum activation of the heatmap and shows this to be more accurate.
Integral Pose Regression applies an expectation operation on ˆ H to give a mean estimate of the joint coordinates.
To do so, the heatmap must first be normalized to sum up to 1; the most common and effective (Nibali et al., 2018) approach is to apply a softmax normalization.
Afterwards, the predicted joint ˆ J re with x and y components ˆ J x and ˆ J y 1 is determined by taking the expectation on the normalized heatmap ˜ H with elements ˜
h p at location p :
[Equation]
where
[Equation]
Here, Ω is the domain of the heatmap and β is a scaling parameter used in the softmax normalization.
Note that softmax normalization assigns a non-zero value to all pixels in ˜ H , even if it was originally zero in ˆ H .
These values also contribute to the expected value, resulting in a center-biased estimated joint coordinate ˆ J re (Gu et al., 2021).
The smaller the β , the stronger the bias.
Although Gu et al.
(2021) proposed a compensation scheme, for the purposes of our analysis, we will assume that β is sufficiently large that the impact of the bias is negligible.
3. S UPERVISION : E XPLICIT H EATMAP VERSUS G ROUND T RUTH C OORDINATES
Detection Methods are learned by providing supervision on the heatmap.
The ground truth H is given as a circular Gaussian, with a mean centered at the ground truth joint coordinate (see Fig. 1).
The loss applied is a pixel-wise MSE between the ground truth h p and the predicted ˆ h p :
[Equation]
where Ω is the domain of the heatmap.
As the loss is defined in terms of the heatmap and not in terms of the predicted joint coordinates, detection-based methods are not end-to-end in their learning and this is often cited as a drawback (Sun et al., 2018; Zhang et al., 2020).
Integral Pose Regression defines a loss based on the difference between the prediction ˆ J re and ground truth joint location J .
The L1 loss empirically performs better than the L2 (Sun et al., 2018):
[Equation]
integral regression are end-to-end methods because they provide supervision at the joint level.
As ˆ H is learned only implicitly, some works refer to the heatmap as “latent” (Iqbal et al., 2018).
3. P ERFORMANCE D IFFERENCES
The performance of detection versus integral regression methods on human pose estimation was compared by Sun et al.
(2018) and Gu et al.
The former found that integral regression is either as competitive or better for 2D pose estimation, though results are not fully conclusive, as Sun et al.
(2018) used a different backbone for integral regression than detection.
The more systematic and detailed comparison of Gu et al.
(2021) used a common backbone. It also partitioned the data based on three factors of variation dominating current human pose estimation benchmarks (Ruggero Ronchi & Perona, 2017): the number of joints present in the scene, the percentage of occlusion and the largest dimension of the bounding box input.
The “hard” samples are those with either 1-5 joints present, > 50% occlusion, or 32-64 pixel input. For these samples, integral regression had, on average, 15% lower end-point-error (EPE) than detection.
The “easy” samples are defined by 11-17 joints, < 10% occlusion, or > 128 px input. In these cases, detection methods were marginally better.
The remaining cases with factors in between were considered to be “medium” ; performance of both methods were roughly equivalent on these samples.
A NALYSIS ON H EATMAP D ECODING 4. L OCALIZED H EATMAP M ODEL
To analyze the performance differences between detection and integral regression methods, we make the following assumptions on the heatmap itself.
First and foremost, we assume that a well-trained network will produce a heatmap ˆ h p with significant or large values in a localized support region around ˆ J .
We denote the region of support as Φ and assume that outside of Φ , the heatmap activation is approximately zero (see Fig.
2 (a)).
Specifically, we model that the normalized heatmap 2 ˜ H is modelled by some density distribution P .
We make no assumption on the form of P , but do assume that it fully captures the support for the joint location within Φ and that Φ is centered on the expected value of P on location µ = ( µ x , µ y ) with an area of (2 s + 1 , 2 s + 1) , i.e. ,
[Equation]
[Figure]
Experimental Verification of Localized Heatmap Model
The assumption of localized activations is reasonable for detection methods as their heatmaps are learned explicitly to match a (localized) ground truth Gaussian.
We empirically verify this model by tallying the activations in the normalized heatmap for both detection and integral regression methods based on the easy/medium/hard splits of Sec. 3.3.
Specifically, we train both a detection and an integral regression network for human pose estimation.
For both networks, we use a Simple Baseline (SBL) (Xiao et al., 2018) architecture with a ResNet50 (He et al., 2016) backbone.
The two networks differ only in their manner of decoding and form of supervision as outlined in Section 3.
We then apply the networks on the standard human pose estimation benchmark MSCOCO
(Lin et al., 2014) using the same easy, medium and hard splits as Gu et al.
For each produced heatmap of 64 × 48 , we estimate an activation sum A by summing the normalized heatmap activations within a (2 s +1) × (2 s +1) square around the ground truth coordinate, ( J x , J y ) :
[Equation]
From Fig.
2 (b), we show that with a sufficiently large s, e.g. 9, the activation sum has alrealy exceeded 95%.
4. E XTREME L OCALIZATION AS A C AUSE FOR P ERFORMANCE D IFFERENCES
Expected End-Point-Error (EPE) For detection methods that decode the heatmap with an argmax, we assume that each position p has some probability w ( µ , s ) of being the maximum activation, i.e. arg max( P ) Φ( µ,s ) ∼ w ( µ, s ) .
Here w ( µ, s ) represents a radially symmetric distribution centered on µ with all non-zero support contained within Φ .
It follows that the expected EPE for one sample of a detection method can be defined as:
[Equation]
where J gt denotes the ground truth coordinates.
For integral regression methods that decode the heatmap with an expectation, the estimated joint coordinate aligns with the center of Φ by definition, i.e. ˆ J = µ , which leads to the following expected EPE for a single sample:
[Equation]
It can be verified that the E de ≥ E re (see proof in Appendix A.2).
This result suggests that for some fixed ( µ , s ) , the expected EPE of regression should always be better than detection.
However, the results of Gu et al.
(2021) as summarized in Sec.
3.3 clearly show this not to be the case.
This inconsistency can be better understood with the following two experiments.
Experimental Verification : Integral Regression Methods Are “More” Localized than Detection
The plot in Fig.
2 (b) indicates that regression methods seem to be more localized than detec- [Table]
tion methods. In particular, for smaller s , A is smaller for detection methods than integral regression methods.
To further verify, we compare the heatmaps with an idealized Gaussian heatmap of varying standard deviations or σ . We plot the σ which gives the lowest Pearson Chi-square statistic, i.e. the highest similarity for the different levels of difficulty in Fig.
2 (c) (detailed values are provided in
Appendix A.3).
As expected, the optimal σ decreases as the samples progress from hard to easy; this result is in line with Fig.
2 (b) and shows that heatmaps are more localized for easy samples.
However, we also observe that the optimal σ for intgral regression is much smaller for the medium and easy cases.
We speculate that this difference arises due to detection methods being trained with a Gaussian heatmap of σ = 2 (Xiao et al., 2018; Sun et al., 2019).
Integral regression methods, on the other hand, have no such explicit supervision and therefore no restriction on the extent of localization.
Experimental Verification: Extremely Localized Heatmaps Degrade
Performance Is it possible that smaller hypothetical σ , i.e. more localized heatmaps cause poor performance?
We verify this for detection methods by applying ground truth Gaussian heatmaps of varying σ for training the networks as per Sec.
and evaluate network performance using Average Precision (AP).
A higher AP corresponds to more accurately located joints.
Fig.
2 (d) confirms that σ = 2 is optimal.
A smaller σ = 1 degrades the performance slightly, and there is a significant drop in the extreme case when σ =0 .
To verify the impact of σ on regression methods, we add a distribution prior to the heatmap to prevent the shrink of the support region Φ .
Specifically, we apply a Kullback-Leibler Divergence (KLDiv) loss between the non-normalized heatmap ˆ H and a Gaussian heatmap with varying σ .
From Table 1, we see that the regression baseline is worse than detection baseline.
However, when we add a prior on the heatmap distribution to encourage a sufficiently large Φ , i.e. ( + σ = 1 , 2 , 3 ), performance improves.
At the optimal + σ = 2 , integral regression becomes competitive with and even exceeds the detection baseline with the optimal σ .
Most importantly, extreme σ values cause performance to degenerate, especially in the case of the extremely small + σ =0 .
We also demonstrate the performance and localization extent of easy/medium and hard cases by AP and activation sum A ( s = 2) .
After adding priors ( e.g. , + σ = 2 ), A ( s = 2) e/m decreases from 0.65 to 0.54, indicating the heatmaps are less localized, and performance AP e/m is significantly improved.
We provide the details of training of combining the two losses and magnitudes of two gradients for better understanding in Supplementary A.4 and detailed changes in heatmap shrinkage Supplementary in A.5.
A NALYSIS OF S UPERVISION AND L EARNING
Given the same heatmap, the difference in heatmap decoding not only influences the final coordinates but also determines the the gradients that arise from the loss.
The gradients in turn influence the learning and thereby generation of the heatmaps themselves.
In this section, we explore the gradients of heatmaps with respect to the loss functions for detection integral regression methods and pinpoint the specific gradient components which slow down the learning of integral regression.
5. H EATMAP G RADIENTS
Detection Methods
The gradient of L de (see Eq. 3) with respect to each pixel in the estimated heatmap ˆ h p is straightforward:
[Equation]
The gradient in Eq.
9 features the predicted heatmap value ˆ h p and the ground truth heatmap value h p . It provides explicit supervision at every pixel.
False positives and false negatives are penalized by reducing wrong high likelihoods and raising incorrect low likelihoods respectively.
For each position p in the heatmap, if the predicted ˆ h p is smaller than the ground truth h p , the gradient is negative and proportionally increases the value in the next iteration. The vice versa is true for predicted ˆ h p larger than the ground truth.
Integral Regression
The gradient of the loss L re (see Eq. 4) with respect to ˆ h p can be estimated based on the chain rule as (see detailed derivation in Appendix A.6)
[Equation]
In Eq.
10, the gradient is split into two terms of interest: ∇ 1 and ∇ 2 which we name the value factor and location factor respectively.
The nature of the value and location factors gives strongly indication to the learning process for integral regression methods.
Firstly, the gradient ∇ p is proportional to the normalized predicted heatmap value ˜ h p as given in the value factor.
This factor makes it prone to gradient vanishing wherever locations p have small heatmap predictions.
Secondly, the location factor is a linear combination of i and j , which manifests itself as a linear plane; far away points like corners are more likely to have large magnitudes.
Though these two factors do not prevent the loss from decreasing during learning, they slow down training.
We further elaborate in Sec. 5.2.
5. A D ETAILED A NALYSIS OF THE INTEGRAL REGRESSION H EATMAP
We now conduct a detailed analysis of the gradients presented in Eq.
and show the typical cases that arise during learning.
When referring to Ω , the domain of the heatmap, we use a coordinate system with the origin at the upper-left corner and increasing coordinates moving to the lower right corner (see Fig.
2 (a)).
For simplicity of discussion, we assume that the ground truth joint J gt is located somewhere in the lower-right quadrant of Ω , though the analysis holds for J gt in all other quadrants of Ω analogously.
The four scenarios are visualized in Fig.
We base our analysis on the assumption that larger magnitude of ∇ p will have larger influence on the weights of the network and effect greater change at the location p in the next iteration, i.e.
[Equation]
where γ represents a step size or learning rate (see sketch of proof in Appendix A.7) and n denotes the update iteration.
As the value and location factors have different contributions on the gradient from Eq.
10 and these are in turn affected by the size s and location µ of Φ , we can define four characteristic cases of Φ and outline how learning is affected.
(1) s is large, Φ has uniformly random values , e.g. a randomly initialized heatmap at the start of training.
As ˜ h p is in a similar range for all p , the value factor ∇ 1 (see Eq. 10) of all pixels are approximately similar.
As such, the distinction between one pixel’s gradient versus another is determined by the location factor ∇ 2 .
Given that the ground truth coordinates ( J x , J y ) is in the lower-right quadrant, as long as ( J x , J y ) are both greater than the predicted coordinate ( ˆ J x , ˆ J y ) , ∇ p has a gradation that increases progressively towards the bottom right corner.
Generally speaking, the location factor pushes the activations of the heatmap towards the corner of the correct quadrant as corners own the largest gradients in the resultant linear plane of location factor ∇ 2 .
(2) s is small, µ is far from the ground truth J , e.g. when µ is in the upper left quadrant of Ω while the ground truth J gt is at the bottom right.
A typical scenario is if the heatmap activates around the left ankle for the ground truth right ankle.
In this case, for the pixels outside of Φ , the value factor approaches zero, i.e. ˜ h p −→ 0 , and pushes ∇ p towards zero and the pixels receive very limited gradient updates.
For pixels in Φ , the values decrease gradually until the value factor outside of Φ no longer dominates, i.e. they reach the same scale, at which point the heatmap returns to case (1).
(3) s is small, µ is in the corner of the same quadrant as the ground truth J gt .
Initially, this case is similar to case (2).
In the process of all the elements becoming the same scale, all the activations moves towards the diagonal side and the prediction approaches the ground truth coordinate.
(4) s is small, µ close to ground truth J gt .
This scenario occurs when the model is reasonably trained and can roughly localize the joint.
The gradient at the ground truth pixel J gt is always negative or zero, i.e. ,
[Equation]
This non-positive value guides the network to predict a large heatmap response at the ground truth location, i.e. a large ˜ h
( J x ,J y ) ; this, in turn, increases the gradient ∇ ( J x ,J y ) .
This property makes the network tend to predict few extraordinary large pixels, thus shrinking the support region Φ and leading to a very small σ in Sec. 4.1.
For detection-based methods, the ground truth heatmaps have already explicitly point out whether a pixel should be activated or not in the heatmap.
Combined with its effective pixel-wise loss, the network learns faster than regression based methods; this is validated in the next section.
5. E XPERIMENTAL V ERIFICATION
Idealized Sample We start by considering the case of one sample for a single joint and visualize the progression of a (64 , 48) heatmap as it gets updated by Eq.
While ∇ n p can be calculated mathematically by Eq.
10, we use autograd in Pytorch to obtain the gradients which we verify to be equivalent.
The four cases from Sec.
5.2 are initialized randomly, with a symmetric Gaussian ( σ = 2) centered at some upper left point, with a linear plane at lower right quadrant, and a symmetric Gaussian ( σ =2 ) centered at ground truth, respectively.
The progression of the differently initialized heatmaps can be observed in Fig.
Mathematically, it is clear from Eq.
2 that the resulting joint coordinate location from integral regression can align with the ground truth joint even if the underlying heatmap does not follow a localized heatmap model.
This is also illustrated in the first three cases of Sec.
However, we posit that during real-world training, over the thousands of training samples observed, over the many epochs, the network can only consistently lower the loss if the heatmap ˜ H is learned to represent P ( J | I ) .
As such, the activation region Φ will be correctly localized over the corresponding semantic region in the image.
Instead, what slows down convergence is the learning of Φ , or rather, the lack of direct guidance to yield a correct Φ as per detection methods.
For the same four cases, we visualize the updated heatmap of Eq.
11 from the gradients defined by the detection loss in Eq.
Unlike
[Figure]
regression, each case leads consistently to the same localized Φ centered on the ground truth joint coordinate, regardless of the initialization.
Real World Samples
The four cases listed in Sec.
5.2 are not so clearly observable in real-world training as the output is based on the optimization results of batch-wise training in a real network.
To verify the results, we compare the training progression for a detection and integral regression network under the experimental settings described in Sec. 4.1.
Fig. 4 (b) compares the training speed of both detection and integral regression and the slower convergence of integral regression.
We visualize for one sample from the MSCOCO validation set the heatmap at epochs 1 andto compare the training progression for detection and integral regression.
Fig.
4 (a) shows the heatmaps of the “left eye” and “right ankle” joints.
For regression, after one epoch, the activations are widespread over a quarter to half the heatmap.
This aligns roughly to a mix of Case 1 andwhere activations are in the correct quadrant but not yet localized to a small region of support.
After 10 epochs of training, predictions for the left eye (which is actually occluded, making it a “hard” sample) are localized with several activations of approximately the same value (all shown as red).
For the right ankle (which is clearly present and is an “easy” sample), the network predicts heatmap values that have collapsed spatially and dominate at on one or two pixel locations.
In contrast, the detection method presents a well-localized heatmap after only one epoch of training.
While there is some activation on the left ankle in epoch 1, this is unlikely to cause errors for the argmax decoding.
After 10 epochs, all activations are centered around the correct right ankle.
C ONCLUSION
In this paper, we dive deeper into integral pose regression method by comparing the differences in the heatmap activations and gradients between detection and integral regression methods.
Our theoretical analysis, paired with empirical verification shows that the shrinkage of heatmaps of regression method is the main cause for integral regression’s lower performance compared to detection.
Furthermore, we investigate the gradients of integral regression by giving theoretical evidence, toy problem and real world illustrations to show that the learning of integral regression is slower to converge as a result of only implicit supervisory signals.
To mitigate the slow convergence and poor performance, we propose a simple spatial supervision on top.
Acknowledgements
This research / project is supported by the Ministry of Education, Singapore, under its MOE Academic Research Fund Tier 2 (STEM RIE2025 MOE-T2EP20220-0015).
A A PPENDIX A. E XTENSION OF F IG . 2 ( B )
We provide a extended plot of Fig.
2 (b), which includes both detection and regression on easy, medium, and hard cases in Fig.
I.
A. P ROOF FOR E Q . 7 ⩾ E Q . 8 IN S EC . 4.
This subsection gives the detailed proof for Eq.
7 > Eq. 8, which means
[Equation]
where
[Equation]
In this way, this equation can be rewritten as
[Equation]
As the distribution of Φ( µ , s ) is radially symmetric, One sufficient but unnecessary proof is that for any symmetric pair { p 1 ( µ x − a, µ y − b ) , p 2 ( µ x + a, µ y + b ) } , it satisfies
[Equation]
According to Triangle Inequality in normed vector space, we arrive at
[Equation]
When we set u = ( x − µ x − a, y − µ y − b ) and v = ( x − µ x + a, y − µ y + b ) , Eq.
II is established.
A. P EARSON C HI - SQUARE VALUE ON DIFFERENT CASES
We report the detailed Pearson Chi-square value between heatmaps of both detection and regression in different scenarios and Gaussian templates with different σ s in Table.
I, II, III.
We set the threshold of A ( s ) > 0 .
8 to show that the support region can represent the whole heatmaps.
Therefore, for easy and medium case, s =4 ; while for hard cases, s =8 .
[Figure]
[Table] [Table]
A. D ETAILS OF COMBINING PRIOR LOSS AND INTEGRAL LOSS
This subsection demonstrates the influence of weights of prior loss and integral loss, and the magnitudes of two gradients regarding the shared heatmaps during the training.
Let assume w p and w i are weights of the prior and integral losses respectively.
We consider the ratio w p
L p : w i · λL i where λ
= 10 − 2 to set the two to approximately equal magnitudes at the initial epoch.
Table V shows that the AP heavily depends on the value of σ but for some fixed σ (within each row), the ratio of loss weightings has less impact and results are stable within 1 − 2% , with larger influence only at the extremely settings of 100 : 1 or 1 :
Table IV (b) provides the ratio of gradient magnitudes to help explain why the weightings do not have much impact.
Table IV (b) gives the mean ratio g p g i where g p and g i are the gradient magnitudes of the prior loss and integral loss respectively over the course of training.
A column of 2/1/0/-1/- means 10 2 / 10 1 / 10 0 / 10 − 1 / 10 − 2 ratio in gradients over the epochs.
The table shows that regardless of the weighting, the gradient ratio decreases over the course of training ( i.e. , dropoff with each column).
Apart from an extremely high weighting of the prior, i.e. 100:1 in column 1, the prior influences training only in the initial 10 epochs.
After epoch 10, the gradient ratio is around the same scale or smaller.
A. D ETAILS OF HEATMAP SHRINKAGE
In order to check whether the added prior influences the spread of the heatmaps, we investigate the shrinkage change of heatmaps by separating the samples.
We separate the samples based on easy/medium versus hard cases where the former generally produces localized heatmaps while the latter has dispersed heatmaps as per Fig.
I; the alternative is to separate by thresholding the activation sum in a fixed area (Eq. 6).
We consider activation sum A (1) > 0 .
34 of the integral regression baseline as localized, otherwise as dispersed (threshold 0.34 selected based on Fig.
I).
We tally the activation sums for the localized versus dispersed heatmaps, without and with the prior for different σ ’s. Activation sums A ( s ) e/m , A ( s ) h , A ( s ) l , A ( s ) h , and Average Precision AP e/m , AP h , AP l , AP h indicate the extent of localization and the performance of the two conditions respectively.
An increase in activation sum indicates further heatmap localization; a decrease indicates dispersion.
The “No prio” row is the baseline integral pose regression.
From Table IV (a), both separations present the same trend.
For localized cases, forcing σ to be degenerately small or too large ( σ = 0 .
5 or σ = 4 ) increases/decreases the activation sum over the baseline, thereby confirming the effect of the prior as intended.
The change in AP follows, i.e. decrease when σ incompatible from being too small or too large.
For dispersed cases, the prior shrinks the activation sum as expected.
Like the localized case, there is an optimal range for which improvement in AP is better; degenerate σ will decrease the AP from the baseline. [Table] [Table]
[Figure]
[Table]
2) e/m
AP [Table] [Table] A
[Table]
) e/
[Table]
AP
[Table]
A
[Table]
AP
[Table]
A
[Table]
prior
[Table]
[Table]
[Table]
[Table]
[Table]
+ σ
[Table]
[Table]
[Table]
[Table]
[Table]
A. D ERIVATION OF E Q .
Eq.
10 is obtained by the chain rule.
Therefore, we list the value of each term as follows
[Equation]
where q = ( u, v ) can be any pixel in Ω .
Combining the terms, gradients at p can be calculated by ∂L re
[Equation]
Eq.
10 can be obtained by rearranging the terms of Eq.
VI.
A. D ERIVATION OF E Q .
Let suppose H n = g ( w n | I ) where H n is the heatmap of given image I at the n -th iteration, g ( w n ) is the function of generating the heatmap, i.e. , backbone network, with weights at the n -th iteration.
According to a Taylor expansion, we arrive at
[Figure]
[Figure]
[Equation]
where w n +1 − w n = − α ∇ g ( w )= − α ∂L∂ H g ′ ( w n ) according to gradient descent, and ( w n +1 − w n ) i is negligible compared with ( w n +1 − w n ) when i> 1 .
Therefore, Eq.
VII can be rewritten as
[Equation]
(VIII)
Therefore, for each pixel h p in H , the update rule is
[Equation]
where γ represents α ∥ g ′ ( w n ) ∥ , h n p and ∇ n p denote heatmap and gradient value located at p in the heatmap at n th iteration.
h
p denotes the original heatmap.
A. R EALWORLD V ISUALIZATIONS
We show the visualization of training performance of integral pose regression and detection-based methods in Figs.
II and III on one single image, respectively.
We can see that the further training of integral pose regression is focused on narrowing down the activated region while the further training of detection-based methods makes the predictions more stable and precise.