A CTIVE L EARNING FOR O BJECT D ETECTION WITH E VIDENTIAL D EEP L EARNING AND H IERARCHICAL U NCERTAINTY A GGREGATION
A BSTRACT
Despite the huge success of object detection, the training process still requires an immense amount of labeled data.
Although various active learning solutions for object detection have been proposed, most existing works do not take advantage of epistemic uncertainty, which is an important metric for capturing the usefulness of a sample.
Also, previous works pay little attention to the attributes of each bounding box (e.g., nearest object, box size) when computing the informativeness of an image.
In this paper, we propose a new active learning strategy for object detection that overcomes the shortcomings of prior works.
To make use of epistemic uncertainty, we adopt evidential deep learning (EDL) and propose a new module termed model evidence head (MEH), that makes EDL highly compatible with object detection.
Based on the computed epistemic uncertainty of each bounding box, we propose hierarchical uncertainty aggregation (HUA) for obtaining the informativeness of an image.
HUA realigns all bounding boxes into multiple levels based on the attributes and aggregates uncertainties in a bottom-up order, to effectively capture the context within the image.
Experimental results show that our solution outperforms existing state-of-the-art methods by a considerable margin.
I NTRODUCTION
Deep learning contributes to huge success in computer vision problems such as semantic segmentation (Long et al., 2015; Ronneberger et al., 2015;
Chen et al., 2018) and object detection (Liu et al., 2016; Lin et al., 2017; Redmon et al., 2016).
However, training a deep neural network typically comes with a cost of large labeled datasets.
Labeling data for complex vision problems requires intensive labor of human experts, which makes preparing for practical application challenging.
Active learning, which gradually labels a set of samples based on the informativeness (e.g., uncertainty), is a promising solution for this problem due to its simplicity and high performance.
Although active learning has been extensively studied on image classification, only a few prior works focused on object detection (Yuan et al., 2021;
Su et al., 2020; Haussmann et al., 2020; Yu et al., 2021) despite its practical importance.
Furthermore, existing works on active learning for object detection have two limitations.
First, when computing the informativeness of an image, most previous works only use the aleatoric uncertainty , not taking the epistemic uncertainty into account.
Epistemic uncertainty, also known as knowledge uncertainty, captures the lack of knowledge of a model (caused by a lack of data) and can be reduced when large amounts of data are available.
Aleatoric uncertainty, on the other hand, captures the noise inherent in the observed data and is irreducible.
As stated in (Nguyen et al., 2022; Hafner et al., 2018; H¨ullermeier & Waegeman, 2021), epistemic uncertainty can reflect the usefulness of samples and support active learning better than aleatoric uncertainty.
Secondly, previous works on active learning for object detection generally ignore the attributes of bounding boxes (e.g., nearest object, box size) when computing the informativeness of an image: informativeness is often defined as the maximum or mean of the uncertainty values of all bounding boxes in the image.
This can be a problem because a cluttered image with many objects belonging to various categories can be enforced to have a similar uncertainty value relative to just a simple image with only a few objects belonging to a single category.
Goal and challenge.
The general goal of this paper is to propose an active learning strategy for object detection, that handles the above limitations of existing works.
First, we aim to build an algorithm that can compute epistemic uncertainty quickly yet correctly in object detection.
Some existing works on object detection (Haussmann et al., 2020; Feng et al., 2019) calculate epistemic uncertainty using multi-model based methods (e.g., model ensemble (Beluch et al., 2018), Monte
Carlo (MC) dropout (Gal & Ghahramani, 2016)).
However, these methods require multiple models or repetitive forward propagations for MC integration, consequently making practical applications difficult.
Secondly, we aim to design an uncertainty aggregation scheme which can consider attributes of bounding boxes and understand the context within images.
This goal is meaningful since aggregation schemes of previous works (Yuan et al., 2021; Roy et al., 2018; Choi et al., 2021), which simply rely on the maximum/mean of all bounding boxes, are hard to reflect the context in images.
Main contributions.
Our first key idea is to adopt Evidential Deep Learning (EDL) to effectively compute epistemic uncertainty, and to propose a new module that makes EDL highly compatible with object detection.
Introduced by (Sensoy et al., 2018; Amini et al., 2020), EDL is a useful tool to compute epistemic uncertainty for detecting unfamiliar data (e.g., unseen unlabeled data) since it samples model ensembles almost instantly.
However, previous works on EDL have mainly focused on image classification and induce unconfident prediction and unstable training when simply applied to object detection.
To this end, we propose a new module named as Model Evidence Head (MEH) to enable confident prediction and stable training of EDL.
Specifically, MEH predicts the expected difficulty, or model evidence, and is optimized independently of the object detector.
To our knowledge, this is the first research to make EDL compatible with object detection on 2D images.
Another key ingredient for our solution is Hierarchical Uncertainty Aggregation (HUA), which makes use of attributes in bounding boxes for computing the informativeness of an image.
According to the attributes, HUA realigns boxes into multiple levels and aggregates uncertainties in a bottom-up order.
This helps to capture the context within the image and improves the quality of the expected informativeness of images.
Overall, our main contributions are summarized as follows:
• We make use of EDL to effectively compute epistemic uncertainty in object detection, and design a new module termed Model Evidence Head (MEH) which solely predicts the model evidence independently of the class confidence to make EDL adaptable to object detection.
• We propose Hierarchical Uncertainty Aggregation (HUA) , which reorganizes all bounding boxes into several levels and aggregate uncertainties of each level in a bottom-up manner, to better capture the context within the image.
We validate the efficacy of proposed methods using RetinaNet and SSD as
base models
on well
- known datasets:
PASCAL VOC, MS-COCO.
Extensive experiments demonstrate that the proposed methods significantly improve performance achieving new state-of-the-art results.
R ELATED W ORKS
Active learning for object detection.
Active learning (Sinha et al., 2019; Yoo & Kweon, 2019;
Sener & Savarese, 2017; Gal et al., 2017; Wang et al., 2016) aims to select a small subset of informative unlabeled samples which is expected to be most effective.
Although active learning has been extensively studied for the classification problem, only a few works focus on object detection (Yuan et al., 2021;
Su et al., 2020; Haussmann et al., 2020; Yu et al., 2021), despite its practical importance.
(Yuan et al., 2021; Su et al., 2020) train discriminators using unlabeled data to predict whether an image is from the labeled set or the unlabeled set.
LL4AL (Yoo & Kweon, 2019) trains an auxiliary module to predict loss of data samples, where a sample with high predicted loss is considered as the one with high informativeness.
The concurrent work CDAL (Agarwal et al., 2020) introduced a distance measure to select diverse samples in semantic and spatial context.
Recently, (Choi et al., 2021) predict the parameters of Gaussian mixture models and computes epistemic uncertainty as the variance of Gaussian modes.
However, there is no guarantee that model uncertainty leads to variance in GMM, and the size of the model ensemble is fixed to be small since the number of Gaussian modes is not allowed to change after training.
Overall, existing approaches do not fully take advantage of epistemic uncertainty and pay little attention to attributes of bounding boxes during uncertainty aggregation.
Our work resolves both limitations using novel and effective methods.
Bayesian deep learning.
The goal of Bayesian deep learning is to build a credible machine that measures uncertainty on its decision as well.
In contrast to the frequentist approach where model parameters are not random variables but fixed quantities, the Bayesian approach assumes prior distribution on model parameters and estimates posterior distributions based on given data.
Recently, many computer vision works have proposed the use of uncertainty through Bayesian deep learning.
(Kendall & Gal, 2017) investigates the benefits of modeling epistemic/aleatoric uncertainty in vision tasks, while (Simon et al., 2022) and (Guo et al., 2022) use the reparameterization trick to estimate uncertainty for neural architecture search and action recognition, respectively.
However, previous
Bayesian deep learning methods such as Bayes by Backprop (Gal & Ghahramani, 2015; Blundell et al., 2015), MC dropout (Gal & Ghahramani, 2016) and Variational Inference (Blei et al., 2017;
Hoffman et al., 2013; Kingma & Welling, 2013) often require costly sampling-based MC approximation to estimate the posterior distribution.
Instead, our work adopts Evidential Deep Learning (EDL) (Sensoy et al., 2018;
Amini et al., 2020) to compute epistemic uncertainty through a single forward propagation.
Different from typical Bayesian approaches, EDL tries to predict prior distributions of posterior models without repetitive model sampling.
However, EDL has hardly been applied to complicated computer vision problems such as object detection yet, and it usually requires adversarial learning using a regularization loss which makes training unstable.
In this work, we propose MEH to overcome aforementioned limitations of previous works.
P ROPOSED M ETHOD 3. P ROBLEM S ETUP AND O RGANIZATION
Active learning consists of multiple learning cycles.
At the first cycle, a large unlabeled dataset Uand a small labeled dataset L 0 are given.
Once an object detector is trained using L 0 , the network selects the most informative images I 0 from U 0 , based on some measures.
Human oracles then manually label I 0 and construct new labeled/unlabeled sets for the next cycle as L 1 =
L 0 ∪ I 0 and
U 1 = U 0 \ I 0 .
The same process is repeated until the annotation budget is exhausted.
In this section, we describe our method for computing the informativeness of an image for active learning in object detection.
Section 3.2 describes how we estimate epistemic uncertainty of a bounding box using EDL and the proposed MEH.
Based on the results in Section 3.2, we describe how we compute the informativeness of an image using HUA, in Section 3.3.
E VIDENTIAL D EEP L EARNING FOR E PISTEMIC U NCERTAINTY
Overview of our approach.
Fig.
1 shows the high-level descriptions of our approach under the EDL framework for computing epistemic uncertainty of a bounding box.
MEH in Fig.
1a is our new module that makes EDL adabtable to object detection.
Under the Dirichlet-Categorical Bayesian framework, we propose to predict class confidences β
= { β k } Kk =1 and model evidence λ , to obtain the concentration parameter α
= { α k } Kk =1 and construct the prior Dirichlet distribution Dir ( θ | α ) for each bounding box.
Epistemic uncertainty is then computed using model ensembles θ ∼ Dir ( θ | α ) as the mutual information between predictions and model posterior as
[Equation]
where H denotes Shannon entropy and θ parameterizes categorical likelihood Cat ( θ ) ; p ( y | θ ) and p ( θ | α ) are probability functions of categorical and Dirichlet distributions.
Fig.
1b shows an example on how the 3 types of uncertainty in (1) change according to λ predicted from our MEH module.
Prior works on active learning without model sampling fail to take advantage of epistemic uncertainty U epi since p ( θ | α ) becomes a point estimate, enforcing U tot = U ale and U epi = 0 .
Our two-stage Bayesian framework based on EDL enables us to sample θ almost instantly, which is significantly faster than multi-model-based Bayesian methods such as MC-dropout (Gal & Ghahramani, 2016) and model ensemble (Haussmann et al., 2020).
In the following, we start by applying
EDL to object detection to describe our algorithm in details.
Applying EDL to object detection.
Inspired by EDL (Sensoy et al., 2018;
Amini et al., 2020;
Zhao et al., 2020), unlike typical object detectors where the classification head predicts parameters θ of the categorical distribution, we predict a high-order Dirichlet distribution Dir ( θ | α ) which is a conjugate prior of the lower-order categorical likelihood Cat ( θ ) .
To fit our evidential model to data, we maximize the marginal likelihood, also known as maximum likelihood Type II.
The marginal p r o b a b ili t y
C1 C2 C
C1 C2 C C1 C2 C
Dissimilar models ( 𝜃 ) → High Epistemic Uncertainty p r o b a b ili t y
C1 C2 C
Similar models ( 𝜃 ) → Low Epistemic Uncertainty
C1 C2 C3 C1 C2 C
Unlabeled Image
Backbone
Cls.
Head
MEH
Reg.
Head 𝛽 𝜆 𝛼
[Figure]
K
[Figure]
C
[Figure]
Obtaining concentration parameter α p r o b a b ili t y
C1 C2 C
C1 C2 C C1 C2 C
Dissimilar models ( 𝜃 ) →
[Figure]
Epistemic Uncertainty p r o b a b ili t y
C1 C2 C
Similar models ( 𝜃 ) →
[Figure]
Epistemic Uncertainty
[Figure]
Unlabeled Image
Backbone
Cls.
Head
MEH
Reg.
Head 𝛽 𝜆 𝛼 𝛼 = 𝛼 𝑘 = 𝜆 exp 𝛽 i σ jK exp 𝛽 j | 𝑘 ∈ {1,2,...,K}
[Figure]
sampling likelihood for a bounding box y i can be obtained by marginalizing over the likelihood parameter θ :
[Equation]
The first term in the integral comes from the categorical distribution, while the second term is from the Dirichlet distribution; the integration can be interpreted as marginalization over every possible categorical model θ ∼ Dir ( θ | α ) .
Thanks to the Dirichlet-Categorical conjugacy, the integral can be written in a closed-form:
[Equation]
The network is then optimized to minimize negative log marginal likelihood L cls = −
[Equation]
k
[Equation]
p ( y = k | α )) , where ¯ y is an one-hot label vector.
At inference, the expected probability for the k -th category is computed as ˆ p k
= α k /S , where
S =
(cid:80) k
α k is the Dirichlet strength.
Issues.
When the EDL method used for image classification is directly applied to object detection, several problems arise: (i) training becomes unstable due to the adversarial regularization loss, and (ii) the prediction becomes unconfident due to the use of ReLU as a hypothesis function.
Given class confidences β = { β i } Ki =1 , previous works (Sensoy et al., 2018; Zhao et al., 2020; Hemmer et al., 2022) adopt ReLU to compute the concentration parameter as α ′ k = ReLU ( β k )+1 .
Although
ReLU performs well in image classification with simple datasets, it noticeably decreases the mAP score in object detection where the number of classes is large and confident prediction is important.
For example, in the case of 80-way classification, β k should be at least 7820 to achieve ˆ p k
=
.even
when β i =
for all i ̸
= k , according to α ′ k = ReLU ( β k ) + 1 and (3).
MEH module.
To overcome the aforementioned issues, we propose MEH and introduce several techniques.
As a first step, we apply softmax as α k = exp( β k ) (cid:80) c exp( β c ) instead of ReLU to produce sufficiently confident prediction.
However, although softmax enables confident prediction, it forces the Dirichlet strength S =
(cid:80) k
α k to be 1.
This makes the entropy of Dirichlet distribution unchangeable and deprives the model of the ability to predict model evidence.
To tackle this issue, we propose a new module termed MEH, which solely predicts model evidence λ .
When λ is obtained from MEH and class confidences { β k } Kk =1 are obtained from the classification head, we propose to re-scale the concentration parameter α as
[Equation]
where β c is the class confidence for class c .
The above re-scaling process with λ transforms
Dir ( θ | α ) to be either concentrated or flat.
Epistemic uncertainty is then computed based on the re-scaled Dir ( θ | α ) via (1).
Fig.1b illustrates the effect of λ on epistemic uncertainty.
The effects of softmax and λ are validated via experiments in Section 4.
MEH loss function.
We separate the training of MEH from the primary object detector with a novel loss function for stability.
For a Bayesian model to work properly, it must be able to recognize the “I don’t know” state, i.e., the model should generate low model evidence for difficult samples.
To this end, previous works on EDL (Hemmer et al., 2022; Sensoy et al., 2018) shrink the model evidence for singletons which do not correspond to target category, via the following regularization loss
[Equation]
where ˆ α = ¯ y
+ ( 1 − ¯ y ) ⊙ α is the concentration parameter
where the element for target category is disregarded.
However, this loss enforces the non-target elements to be one, no matter how large an error is detected, since the uniform Dirichlet distribution Dir ( θ | < 1 , ..., 1 > ) is considered as a constant supervisory signal.
Furthermore, this adversarial regularization loss suppresses the overall model evidence and makes training unstable.
To overcome these issues, we propose a simple yet effective solution: we utilize the continuous loss score from the classification head as a target value.
Specifically, when classification loss l s,i is given for bounding box i at scale s , the proposed loss function for MEH is defined as
[Equation]
This loss encourages MEH to decrease model evidence
λ s,i when the classfication loss l s,i is large; similarly, MEH is guided to increase λ s,i for small l s,i .
L MEH has the same effect as the previous regularization loss (5) in that it prevents overconfidence and helps the model to predict uncertainty better; however, by using the target loss as a changeable supervisory signal, L MEH is able to dynamically penalize the model to better predict the model evidence.
Stability.
Note that we deliberately supervise only model evidence λ independently of Dirichlet parameters { α k } Kk =1 to train MEH.
λ is utilized only for re-scaling the Dirichlet strength and λ has no effect on p ( x | θ ) or ˆ p k , since λ will naturally vanish when division α k /S occurs.
Hence, the
MEH network Φ MEH and remaining network Φ \ Φ MEH can be updated in a disjoint manner.
We in turn optimize Φ MEH and Φ \ Φ MEH , thus training of Φ MEH never affects the performance of the primary object detector; this resolves the stability issue of previous EDL approaches.
Applications to recent models.
Our method is not limited to softmax-based object detectors like
SSD (Liu et al., 2016), but also applicable to sigmoid-based detectors like RetinaNet (Lin et al., 2017).
Given sigmoid-based prediction p ∈ [0 , 1] , focal loss can be defined as FL ( p t ) =
− (1 − p t ) γ log( p t ) where p t = p if y = 1 , p t = 1 − p otherwise.
To enable evidential learning, we replace sigmoid-based binary prediction p by Dirichlet-Categorical multiclass prediction p ( x | α ) of equation (3).
We train RetinaNet with the modified evidential focal loss and report its performance in Fig.
3. H IERARCHICAL U NCERTAINTY A GGREGATION
Based on the epistemic uncertainty of a bounding box, we propose HUA: a new method for computing the final informativeness score of an image.
Recent active learning methods for object detection (Yuan et al., 2021; Choi et al., 2021; Yu et al., 2021) typically compute the informativeness of an image as the mean or maximum of all bounding boxes.
To this end, we propose a generalized framework that can apply different kinds of aggregation functions based on the level of information (predicted category, box size, nearby object) of each bounding box, as described in Fig.
Filtering bounding boxes.
Single-stage object detectors such as RetinaNet and SSD generate bounding boxes at every scale, pixel and anchor.
Since most boxes correspond to background, we first filter out background boxes whose max k
(ˆ p k ) is lower than threshold γ score .
Realigning bounding boxes.
Besides the uncertainty score, each bounding box contains much more information: predicted object, scale and category it belongs to.
Based on these information, we propose to realign the boxes into a hierarchical structure.
First, boxes are matched to the nearest object depending on the IoU score.
For example, in Fig.
2a, the blue box is matched to “object”, but the green box is ignored since IoU is lower than the threshold γ IoU .
Secondly, boxes are further grouped based on the scale to which they belong.
Lastly, boxes are divided based on the category ( argmax k ˆ p k ) to which they belong.
See Fig.
2b for the resulting hierarchy of bounding boxes.
Uncertainty aggregation.
Once the bounding boxes are fully realigned, individual uncertainty scores are unified in a hierarchical order.
As shown in Fig.
2b, uncertainties of the boxes in a lower level (e.g., class) are aggregated into a single value through a predefined aggregation function;
[Figure]
human human cow horse cow cow horse
[Figure]
object
[Figure]
object
[Figure]
𝑏𝑜𝑥 𝑏𝑜𝑥 𝑏𝑜𝑥 𝑏𝑜𝑥 𝑏𝑜𝑥 𝑏𝑜𝑥 𝑏𝑜𝑥 𝑏𝑜𝑥
[Figure]
the aggregated value is then passed to an upper level (e.g, scale).
This aggregation repeats from the “class level” to “object level”.
Note that HUA can be viewed as a generalized scheme where different types of aggregation functions can be adopted at different levels of information; it can cover aggregation methods of previous works (Choi et al., 2021; Yuan et al., 2021) where identical functions (max or mean) are applied to the every level.
As discussed in Appendix, we specifically propose to adopt sum operation when aggregating the uncertainties at the object level, since this reflects the number of objects in final informativeness of the image.
We also propose to apply sum operation to the class level.
Given that ambiguous bounding boxes are often classified into more than one class, considering these classes altogether helps to increase the uncertainty score of the difficult object.
By considering the attributes of each bounding box, HUA effectively captures the context within the image in the informativeness score.
3. S ELECTION OF I NFORMATIVE I MAGE
Now the epistemic uncertainties of all unlabeled images can be computed at each active learning cycle.
While previous works typically select the top- k uncertain images, we propose to select from the filtered-out images as well.
These images are also valuable since the model was incapable of sensing any objects due to lack of knowledge.
Composing a certain fraction of selections with these images further improves the performance, where the detailed analysis is provided in Appendix.
E XPERIMENTS 4. E XPERIMENTAL S ETUP
Dataset.
Our work is validated on PASCAL VOC (Everingham et al., 2010) and MS-COCO (Lin et al., 2014).
As to PASCAL VOC, VOC07+12 trainval is used for training and VOC07 test is used for evaluation.
While mAP50 is used for evaluation metric in PASCAL VOC, mean of mAP at IOU=.50:.05:.95 is used for MS-COCO.
In the first cycle for PASCAL VOC, 5% from 16, samples are randomly selected.
Then 2.5% of samples are labeled at each cycle until reaching 20%.
As for MS-COCO with 117,267 samples, labeled sets increased from 2% to 10% in steps of 2%.
Implementation details.
For fair comparisons with previous works (Yuan et al., 2021; Agarwal et al., 2020; Yoo & Kweon, 2019), we adopt RetinaNet (Lin et al., 2017) and SSD (Liu et al., 2016) which use ResNet50 and VGG16 as backbones, respectively.
The structure of MEH is the same as the regression head except that the dimension of the output layer is one.
As for RetinaNet, we follow the setting of (Yuan et al., 2021) and set epoch to be 26, batch size to be 2; also, we follow the setting of (Yoo & Kweon, 2019; Agarwal et al., 2020) for SSD and set epoch to be 300, batch size to be 32.
In both experiments on PASCAL VOC and MS-COCO, the SGD optimizer is used with a learning rate of 0 .
001 ; a ℓ 2 weight decay rate of 0 . 0001 ; a momentum of 0 . 9 ; and a warm-up strategy for the early 500 steps.
As for the backbones, ResNet50 and VGG16 are pretrained on
ImageNet.
Note that at every cycle, the object detector is trained from scratch with the pretrained backbone.
We implement our framework using Pytorch 1.5.0 and mmdetection 2.13.0.
In every experiment, single NVIDIA GeForce RTX 2080Ti is used.
We set the confidence threshold for box
[Figure]
[Figure]
filtering γ score to 0.3, and IoU threshold for object grouping γ IoU to 0.5.
500 ensemble members θ are sampled for uncertainty estimation.
For our scheme, (sum, max, sum) operations are adopted for each information level (object, scale, class) when applying HUA.
Ablation studies with different operations are provided in Appendix.
We compose 15% of selections with filtered images, where the results with various selection ratios are reported in Appendix.
Baselines.
We compare our work with state-of-the-art works: MI-AOD (Yuan et al., 2021), CDAL (Agarwal et al., 2020), LL4AL (Yoo & Kweon, 2019), Core-set (Sener & Savarese, 2017), CALD (Yu et al., 2022).
Also, as basic baselines, Entropy Sampling (Roy et al., 2018), Random Sampling with vanilla RetinaNet/SSD are considered as well.
For Core-set, features from backbone (ResNet50/VGG16) are used following the setting of (Yoo & Kweon, 2019; Choi et al., 2021).
While Random Sampling randomly selects unlabeled samples to be labeled for the next cycle, Entropy Sampling selects the samples based on Shannon entropy; the average value of Shannon entropy among all of bounding boxes is regarded as the informativeness as done in (Yuan et al., 2021; Choi et al., 2021; Roy et al., 2018).
Five independent networks are trained with different seeds and the averaged performance is reported with 95% confidence intervals.
4. M AIN E XPERIMENTAL R ESULTS
Comparison with state-of-the-arts.
Fig.
3 shows mAP scores under various settings.
Our proposed method exhibits a clear edge over other baselines, giving the best performance.
In Figs.
3a and 3c, the proposed method shows superiority in early cycles, proving that the proposed evidential focal loss is effective when labeled data is extremely limited.
It is noteworthy to consider our method is trained only on a labeled set, while MI-AOD benefits from semi-supervised learning using an unlabeled set as well.
In Fig.
3b, the reason SSD has lower initial performance than RetinaNet comes from the difference of the loss function.
Unlike RetinaNet trained with the focal loss, SSD is trained with the CE loss.
As in equation (3), since we calculated α using a softmax, the loss expression does not change at all in the case of SSD.
Still, it is remarkable that our methods (MEH,
HUA) gradually increase the performance by selecting informative images even with the same loss.
Comparison with multi-model based Bayesian methods.
In recent active learning works, multimodel based methods are used to compute epistemic uncertainty.
Here, we compare existing meth- ods and our work.
As for baselines, we first consider MC-dropout; following (Beluch et al., 2018;
Choi et al., 2021), dropout layers whose rate is 0.1 are added to the convolutional layers andforward passes are sampled to compute epistemic uncertainty using equation (1).
Secondly, model ensemble is considered, where we follow (Haussmann et al., 2020; Choi et al., 2021) and train three independent models to construct an ensemble.
Lastly, we consider the work of (Choi et al., 2021); it predicts the parameter of a Gaussian mixture model (GMM) and regards the variance of Gaussian modes as the epistemic uncertainty.
In every baseline, we average uncertainties from all bounding boxes for the computation of the final image-level uncertainty, as done in (Yuan et al., 2021; Choi et al., 2021; Roy et al., 2018).
In Fig.
4, we first observe that multi-model based methods yield higher scores than random sampling, underlying the importance of epistemic uncertainty.
Still, the large performance gap between the baselines and our work confirms the benefits of the proposed techniques: MEH and HUA.
It is noteworthy that baselines ignore the hierarchy between bounding boxes, which implies that there possibly exist important yet underestimated boxes.
4. A BLATION S TUDIES AND D ISCUSSIONS
Effects of MEH and HUA.
Table 7 shows ablation studies on the proposed methods.
While Random and Entropy use vanilla RetinaNet with focal loss, proposed methods use the proposed evidential focal loss.
While ReLU computes α with ReLU as done in (Sensoy et al., 2018;
Zhao et al., 2020;
Hemmer et al., 2022), Softmax uses softmax.
All methods except HUA compute uncertainty of an image as the mean of all boxes, as in (Yuan et al., 2021; Choi et al., 2021).
At every cycle, the proposed methods (Softmax, MEH, HUA) increase performances by a large margin.
The results indicates that informativeness can be captured much better via our proposed methods.
Interestingly, it can be seen that the ReLU method performs even worse than Random .
This result confirms that previous works of EDL for image classification cannot be simply applied to object detection.
[Table]
Comparison of computing costs.
In Table 2, we compare computing costs of the proposed method with multi-model based baselines (MC-dropout, Ensemble ) and Entropy baseline.
Specifically, the first column indicates the latency time for uncertainty calculation of a single image with a 95% confidence interval and the second column shows the size of all parameters in the models.
It can be seen that the proposed method has a similar model size to that of MC-dropout and Ensemble baselines; a slight increase is due to the MEH module.
With respect to the latency time, multi-model based methods require a significant computing cost because of multiple forward propagations for MC integration.
Specifically, MC-dropout and Ensemble take 25 and 3 forward passes, respectively.
In comparison, our method requires much less latency time, although it samples 500 ensemble members
Cat ( θ ) .
This is because our method requires only single forward propagation and each ensemble member is sampled virtually (i.e., θ ∼ Dir ( θ | α ) ).
Note that Entropy has the lowest latency, since this method dose not use epistemic uncertainty and there is no need to sample ensemble members.
Visualization analysis of model evidence λ .
To clearly understand the role of model evidence λ , we illustrate the effect of λ in Fig.
In the 2 nd column, when model evidence λ is not considered, uncertain regions are observed over a wide area inside the object.
On the other hand, in the 3 rd column, MEH estimates high λ only around the center of objects.
This is because the difficulty of classification increases as the center of the bounding box moves away from the center of objects due to ambiguity.
The MEH output, λ , is used to readjust the prior Dirichlet distribution Dir ( θ | α ) .
As noted in Section 3.2, a high λ increases Dirichlet strength and make D ( θ | α ) concentrated.
It makes similar θ ∼ Dir ( θ | α ) be sampled and decreases the epistemic uncertainty according to equation (1).
[Table]
For example, in the 4 th column, it can be seen that the uncertainty of the central part of the object is reduced due to high λ ; conversely, the uncertainty of the edge part of the object is preserved high.
[Figure]
Examples of easy and hard samples.
In Fig.
6, for intuitive understanding, we compare uncertainties of easy and hard samples with Entropy Sampling (Roy et al., 2018), and MI-AOD
(Yuan et al., 2021) which represent SoTA.
In all methods, easy examples tend to have just one unoccluded object; however, hard examples tend to have numerous objects, which are unclear or heavily occluded.
One interesting thing to note is that difference in informativeness scores between the easy samples and the hard samples.
In previous methods, the range of scores between images is not wide, since these methods average the uncertainties of all bounding boxes to get a single score.
On the other hand, in the proposed method, it can be seen that the informativeness scores of the hard examples are more than 50 times higher than those of the easy examples.
This is because the proposed uncertainty aggregation scheme (HUA) adds up the informativeness scores of meaningful bonding boxes with consideration for additional information (e
[Figure]
g., the number of objects, categories in images)
[Figure]
C ONCLUSION
We proposed a new active learning strategy for object detection.
With the evidential deep learning framework and model evidence head, our scheme effectively estimates the epistemic uncertainty of a bounding box.
Also, our hierarchical uncertainty aggregation provides a new guideline for computing informativeness of an image.
Our scheme presents an up-and-coming direction for active object detection
[Figure]
where estimating epistemic uncertainty accurately yet quickly is of crucial importance
[Figure]
R EPRODUCIBILITY S TATEMENT
To ensure reproducibility we provide theoretical proof and implementation details in Appendix.
Specifically, our experimental setup including hyperparameter setting is described in Appendix, and our source codes are included in Supplementary Material.
The code is available at https: //github.com/MoonLab-YH/AOD_MEH_HUA
A CKNOWLEDGEMENT
This work was conducted by Center for Applied Research in Artificial Intelligence (CARAI) grant funded by DAPA and ADD (UD190031RD).
Also, this work was supported by Institute of Information & Communications Technology Planning & Evaluation(IITP) funds from MSIT of Korea (No.
A A PPENDIX B A DDITIONAL E XPERIMENTAL S ETUP
In this section, we provide more detailed information on our experimental setting.
Following previous works, two well-known object detectors are chosen as base models: RetinaNet (Lin et al., 2017) and SSD (Liu et al., 2016).
First, RetinaNet adopts ResNet-50 which is pretrained on ImageNet as backbone.
ResNet-50 has four stages and output of each stage is passed to the neck structure.
The output dimension of each stage is 256, 512, 1024, 2048, respectively.
The parameters of the backbone in the first stages are frozen during training.
As for the neck structure, Feature Pyramid
Network (FPN) with five stages is adopted.
FPN takes the output of ResNet as input and returns 256-dimensional features at the every stage.
The output of FPN then goes through a classification head and a regression head, independently.
The structure of the classification head and the regression head are the same as that of vanilla RetinaNet (Lin et al., 2017) which consists of three convolutional modules and one last convolutional layer for classification and regression, respectively.
Following (Lin et al., 2017; Yuan et al., 2021), we utilize a bounding box generator whose octave base scale is 4; scales per octave is 3; ratios of bounding boxes are (0.5, 1.0, 2.0).
The base sizes of boxes are (8, 16, 32, 64, 128) from the lowest scale to the highest scale.
The coordinates of bounding boxes and ground truth are encoded to the delta and considered as the target value of the regression head.
Specifically, given central coordinates ( x, y ), width ( w ) and height ( h ) are given for the bounding box ( px, py, pw, ph ) and the ground truth ( gx, gy, gw, gh ), values of delta is computed as:
[Equation]
Regression heads are supervised to predict the encoded value ( gx, gy, gw, gh ) and these values are decoded to the coordinates of the uppermost left corner and the lowermost right corner to make the final prediction.
Regarding the loss, as stated in the main paper, focal loss (Lin et al., 2017) with marginal likelihood is adopted.
Hyperparameters for focal loss ( γ, α ) are set to 2.0 and 0.25, respectively.
As to assigning positive/negative boxes, bounding boxes whose IoU scores with the ground truth are higher than 0.5 are assigned to the positive box; boxes with iou scores greater than 0.4 and less than 0.5 are assigned as negative boxes and all remaining boxes are ignored.
As for evaluation, 1000 bounding boxes with the highest confidence are used for Non-Maximum Suppression (NMS).
The IoU threshold for NMS is set to 0.5 and the number of boxes after NMS is at most 100.
Secondly, SSD adopts VGG-16 which is pretrained on ImageNet as backbone.
Following (Liu et al., 2016), the backbone of SSD has extra layers to generate multi-scale feature maps.
After these layers, six levels of features are created with dimensions of (512, 1024, 512, 256, 256, 256) and these features are taken as inputs to the classification head and the regression head.
The same encoding/decoding scheme as RetinaNet was used.
Different from RetinaNet, hard negative mining is applied to assign positive/negative boxes; specifically, three times more negative boxes than positive boxes are assigned.
As for evaluation, the same NMS scheme is adopted as RetinaNet except the number of bounding boxes after NMS is set to 200.
For both models, the SGD optimizer is used with learning rate of 0.001; a momentum of 0. and a weight decay of 0.0001.
C A LGORITHM D ESCRIPTION
The detailed procedure of the proposed active object detection method is given in Algorithm 1.
D P ROOF OF M ARGINAL L IKELIHOOD L OSS
We consider the Dirichlet-Categorical Bayesian framework where the low-order predictive distribution parameters θ are sampled from the high-order prior distribution (i.e., θ ∼ Dir ( θ | α ) and θ parameterizes categorical likelihood Cat ( θ ) .)
To train our evidential model, we maximize the marginal likelihood which is given as below:
[Equation]
do do 2:
Predict class confidence β
= { β β β K } and model evidence λ 3:
Compute concentration parameter α = { α k = λexp ( β k ) (cid:80) i exp ( β i ) | k
= 1 , 2 , ..., K } 4:
Compute predictive probability p ( y = k | α )
=
α k (cid:80) i α i 5:
Compute classification loss L cls ( y p ( y = k | α )) 6:
Update object detector Φ tD using L cls 7:
Compute MEH loss L MEH ( λ
L cls ) 8:
Update MEH Φ tMEH
L MEH
Φ tD fixed 9: end for
Step
Compute via
1: for data u
U do 2: for bounding box do 3:
Predict class confidence β
= { β 1 , β 2 , ..., β K } and model evidence λ 4:
Compute concentration parameter α = { α k = λexp ( β k ) (cid:80) i exp ( β i ) | k
= 1 , 2 , ..., K } 5:
Sample categorical distributions as θ ∼ Dir ( θ | α ) 6:
Compute epistemic uncertainty I
[ y, θ ]
= H (cid:2)
E p ( θ | α )
[ p ( y | θ )] (cid:3) − E p ( θ | α ) (cid:2) H [ p ( y | θ )] (cid:3) 7: end for which is equivalent to
[Equation]
The above integral can be simplified using a useful trick.
Given Dir ( θ | α ) , integrating probability density function over θ results in one:
[Equation]
Dividing the above equation by a constant part
[Equation]
which is not involved in the integration
[Equation]
gives the following result:
[Equation]
Now we can regard equation (8) as the integration over ”shifted parameter” (i.e., α j + 1 { y j = i } ).
This shift of parameter changes the equation (10) to:
[Equation]
Considering the case when y j = i separately, the above equation (11) can be written as:
[Equation]
Now we can use the fact that Γ( x + 1) =
x Γ( x ) to simplify the above equation as:
[Equation]
By substituting the result of the above equation in the integral of equation (8), we obtain:
[Equation]
E A NALYSIS OF T HRESHOLD V ALUES
To understand HUA better, we investigate the effect of each threshold value: γ score , γ IoU .
We vary the thresholds and report the results in Table 3.
Note that aggregate functions (Sum, Max, Sum) are applied to information levels (Object, Scale, Class) in all tries for fair comparison.
It can be seen that higher γ IoU increases the performance.
This is because excessively low thresholds cause too many unnecessary bounding boxes to be taken into account, thus deteriorating the quality of estimated informativeness.
Otherwise, a too high or too low γ score impairs the quality of filtering and degrades the final performance.
Thresholds
Ratio (%) of Labeled Samples
[Table]
γ
[Table] [Table]
[Table]
F A NALYSIS OF HUA
Fig.
7 visualizes how HUA works in detail.
Note that aggregation functions (Sum, Max, Sum) are chosen for (Object, Scale, Class).
First, it can be seen that unclear or occluded objects have high uncertainty.
For example in the left image, occluded boats in the left side show higher uncertainty than the one in the right side.
It can also be seen that uncertainties of numerous objects within the left image are summed and contribute to the higher informativeness score.
Note that since the aggregation function for ”Class” is ”Sum”, bounding box which is classified into multiple classes simultaneously is supposed to have high uncertainty.
For example in the right image, the head of horse in the right image is classified into the horse and the cow at the same time; bounding boxes which predict either category are separated into independent groups, and uncertainty for each category group is computed independently and summed together.
G C OMPARISON WITH OTHER L OSS T YPE FOR MEH
For the training of MEH, the mean Square Error ( MSE ) loss is adopted for regression of the target loss.
In this section, we further investigate performances of other typical losses which are wellknown for the regression problem.
Table 4 compares the mAP50 score on PASCAL VOC with
RetinaNet as the base model.
The method MAE indicates the mean absolute error; given prediction ˆ y and ground truth y , Loss ( y, ˆ y )
=
| y − ˆ y | .
The MSLE method refers to the mean squared logarithmic error, (i.e., Loss ( y, ˆ y ) =
( log ( y + 1) − log (ˆ y + 1)) 2 ).
Experimental results show that general losses for regression are all effective, yet MSE loss shows the best result.
[Figure]
[Table]
[Figure]
[Table]
H A NALYSIS ABOUT THE S ELECTION R ATIO OF F ILTERED S AMPLES
In this section
[Table]
we compare performance of the proposed method varying the selection ratio of filtered samples.
Note that the bounding boxes whose confidence value are lower than γ score are filtered out to ignore predictions for background.
This scheme naturally makes the informativeness score of any image to zero, if all bounding boxes in the image are unconfident.
We stress that these images with zero informativeness are important since the object detector is unable to find any object due to its lack of knowledge.
In Table 5, it can be seen that taking some portion of these images of zero informativeness into account helps to improve the performance.
This is because training with filtered images provides the model to experience difficult objects from unfamiliar categories.
However, including too much filtered images into the labeled set of the next stage rather impair the performance.
This is because some objcets with low confidence are too small/large or heavily occluded and interfere with training of the detector.
[Table]
) of Labeled Samples
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
I C OMPARISON OF AGGREGATION FUNCTION
An obvious question in HUA is: which combination of aggregation functions yields the best performance?
In Table 6, we compare three types of functions: Avg, Max, Sum.
In all tries, γ score , γ IoU is set to 0.3, 0.9, respectively.
We try various combinations applying different aggregation functions in each level of information.
It can be seen (Sum, Max, Sum) gives the best performance exhibiting a large margin over naive aggregation schemes of (Choi et al., 2021; Yuan et al., 2021).
Besides, sum operation is observed to perform the best for “Object”, since it reflects the number of objects inside an image.
Likewise, sum operation shows the best result for “Class”.
It confirms the importance of considering uncertainties from ambiguous classes together.
Otherwise, max operation prevails for “Scale”.
Since predictions can appear at various scales depending on the shape of the object regardless of the amount of information, it seems that the quality of uncertainty is rather impaired if all scales are considered.
Function
Ratio (%) of Labeled Samples (Obj, Scale, Class) 5.
(Avg, Avg, Avg) (Yuan et al., 2021) 52.
(Avg, Avg, Sum) 52. (Avg, Sum, Sum) 52.
(Avg, Max, Sum) 52. (Max, Avg, Sum) 52.
(Max, Max, Max) (Choi et al., 2021)
[Table] [Table]
, Sum)
[Table] [Table]
J D ETECTION OF N OISY L ABELS
[Figure]
In active learning work, generally, it is assumed that a perfect oracle generates 100% accurate labels to unlabeled samples, so that the main focus of existing active learning works (including our work) becomes effectively seleting the informative samples under the assumption with no label noise.
In this section, we deal with an interesting question: how should we handle the label noise issue when human annotations are not 100% accurate?
Orthogonal to the existing active learning methods, a simple way is to utilize a new module that identifies the noisy labels and re-label them or exclude them during training.
As illustrated in works of (Bernhardt et al., 2022; Younesian et al., 2021), cross-entropy is commonly adopted to detect the mislabeled samples in noise-robust learning literatures.
To demonstrate the capability our model for detecting noisy labels, we present simple toy examples in Figure 8.
To reproduce the noisy label, we change labels ( y ) of some bounding boxes to other similiar categories (e.g., horse → cow), and compute the cross entropy using the probability distribution ( ˆ y ) of Equation 3 for comparison.
As can be seen in Figure 8, bounding boxes with wrong labels show much higher cross entropy, ensuring opportunities to cope with noisy labels.
[Figure]
K A NALYSIS ON HUA WITH RESPECT TO THE NUMBER OF OBJECTS
As can be seen in Figure 6, HUA calculates image informativeness considering various information inside the image (number of objects, difficulty of each object), which makes the uncertainty scores of hard examples significantly higher than those of easy examples.
However, this characteristic of HUA raises the following question: Does HUA favor only complex images with many objects?
Wouldn’t it be possible to replace HUA by simply selecting images in which many objects appear?
To answer these question, we investigated the relationship between the number of objects in the image and the uncertainty score, and found that a large number of objects does not always lead to a high uncertainty score, as shown in Figure 9.
This is because the resulting uncertainty of an image depends not only on the number of objects but also on the uncertainty value of each object .
For example, in the image on the left of Figure 9, there exists two objects, but the total uncertainty is higher than that of the right image, because the bounding boxes corresponding to the objects are classified as people and motorbikes at the same time.
On the other hand, since the objects in the right image are easy objects that are clearly classified only as people, the uncertainty of all five objects is not too large.
L A NALYSIS ON THE ORDER OF HUA
In this section, we investigate the effect of the aggregation order in HUA to provide additional insight.
Under the same conditions as the experiment conducted in the main manuscript, we only changed the order of scale and class, i.e., the aggregation hierarchy in HUA is changed from ”object (sum) ← scale (max) ← class (sum)” to ”object (sum) ← class (sum) ← scale (max)”.
As shown in the table below, the performance is degraded compared to the proposed scheme when we switch the order.
[Table]
We interpret this result as follows: 1) Since most objects do not span multiple scales (they mostly span 1 ∼ 2 scales), there is little difference even if the order of class and scale have changed.
If the order is changed, uncertainty values from multiple scales may be mixed into a single informativeness score of one object.
Rather, it seems more appropriate to focus on a single informative scale to compute the score of the image.
Overall, both our intuition and experimental results indicate that the original order is more powerful and makes more sense.
M A NALYSIS ON GENERALITY OF HUA
To demonstrate the generality of the proposed HUA, we conducted additional experiments.
We combined traditional methods (Roy et al., 2018; Yuan et al., 2021) with HUA and reported the performance gain in Table 8.
[Table]
As shown in the table, HUA generally improves the performance in both cases.
This is because
HUA helps to better understand the context within images by considering the information hierarchy (e.g., class, object, scale), than just taking the maximum value of the uncertainties of all bounding boxes as done in (Roy et al., 2018; Yuan et al., 2021).
This result confirms the intuition that existing works can be enhanced by adopting HUA for uncertainty aggregation.
N A PPLICATION TO A TWO - STAGE DETECTOR
[Figure]
[Table]
a) Obtaining concentration parameter α using a two- stage detector
[Figure]
Although we have verified our algorithms (MEH, HUA) only for single-stage detectors (RetinaNet,
SSD) in the main paper, these algorithms can be easily extended to two-stage detectors as well.
Fig.
10a shows the high-level descriptions of our approach under the EDL framework in a twostage detector.
For each proposal, the ROI pooled features are taken as input by the classification head and MEH.
Then class confidence β
= { β k } Kk =1 and model evidence λ are generated, and concentration parameter α of the prior Dirichlet distribution Dir ( θ | α ) can be computed via equation (4).
Given θ ∼ Dir ( θ | α ) , epistemic uncertainty of every proposal is computed via equation (1).
Afterwards, epistemic uncertainty of all proposals are aggregated considering information hierarchy (e.g., predicted class, scale, number of detected object) through HUA to get total uncertainty of the image.
The only difference from the single-stage detector is that it computes uncertainty only for proposed bounding boxes by RPN rather than all bounding boxes.
Fig.10b shows the performance in PASCAL VOC 2012 when the proposed methods are applied to FasterRCNN (Ren et al., 2015).
It can be seen that the proposed method performs better than various baselines (SSM(Wang et al., 2018), LS+C and LT/C(Kao et al., 2018), VAAL(Sinha et al., 2019), LL4AL(Yoo & Kweon, 2019)), which confirms that our scheme can give general performance gain even in the two-stage detector.