DETR Breakdown Part 2: Methodologies and Algorithms Aritra Roy Gosthipaty and Ritwik Raha PyImageSearch

​

Table of Contents

DETR Breakdown Part 2: Methodologies and Algorithms

In this tutorial, we’ll learn about the methodologies applied in DETR.

This lesson is the 2nd of a 3-part series on DETR Breakdown:

DETR Breakdown Part 1: Introduction to DEtection TRansformersDETR Breakdown Part 2: Methodologies and Algorithms (this tutorial)DETR Breakdown Part 3: Architecture and Details

To learn about the methodologies applied in Detection Transformers, just keep reading.

DETR Breakdown Part 2: Methodologies and Algorithms

Welcome back to Part 2 of this tutorial series on Detection Transformers. In the previous tutorial DETR Breakdown Part 1: Introduction to DEtection TRansformers, we looked at what factors led to the birth of DETR, what components were added, and what really is the Chemical X that made DETR into the super object detector it is today.

As scientists, learners, and builders, we must seek to understand what causes excellent performances. One of the key components of DETR is a secret sauce called Set Prediction Loss. Today we place this “Set Prediction Loss” under the microscope.

The DETR Model 👁️

First, let us revisit the DETR Model architecture in the interactive diagram below.

Ironically, we will not be demystifying the above architecture in this tutorial. Set Prediction Loss comes in where this architecture ends and outputs predictions. The predictions are matched with the ground truth objects to yield a loss. The model iterates on that and gets better and better and eventually at par with past models like Faster R-CNN.

How? Just keep reading to find out!

Object Detection Set Prediction Loss 📉

DETR infers a fixed-size set of predictions. Note that is significantly larger than the typical number of objects in an image.

One of the main difficulties of training is to score predicted objects with respect to the ground truth. Therefore, Carion et al. (2020) propose the following algorithm.

A loss that produces an optimal bipartite matching between predicted and ground truth objectsOptimize the object-specific (bounding box) losses

Optimal Bipartite Matching 🔄

Here let us pause for a moment to understand what is bipartite matching.

Bipartite matching is a mathematical concept from graph theory, often used in computer science and optimization problems. It refers to the process of finding a subset of edges in a bipartite graph such that each vertex in the graph is incident to at most one edge in the subset and the size of the subset is maximized. This subset of edges is called a maximum cardinality matching or simply a maximum matching.

L.R. Ford and D.R. Fulkerson, “Maximal Flow Through a Network,” Canadian Journal of Mathematics, vol. 8, pp. 399-404, 1956.

A bipartite graph is a special type whose vertices can be divided into two disjoint sets such that all edges connect a vertex from one set to another. In other words, no edges connect vertices within the same set.

Bipartite matching is the process of pairing vertices from the two sets so that each vertex is paired with at most one vertex from the other set, and the total number of paired vertices is maximized.

Think of it as finding the best way to match items from two categories, like connecting workers to jobs or students to projects. Each item from one category is paired with at most one item from the other category, and the cost of the pairs is optimal. This is shown in the demo below.

Optimal Bipartite Matching for Objects 🌐

Before we go into how the above concept of bipartite matching can be applied to ground truth and predicted objects, let us first familiarize ourselves with the terminologies and nomenclature.

The process of finding matching pairs is similar to the methods used in previous object detectors for connecting proposed or anchor boxes to actual objects in an image. The main difference is that, for direct set prediction, we need to find unique one-to-one matches without any duplicates. Let’s look at the demo below to understand the pointers better.

is the ground truth set of objects. is the set of predictions.

We already know that is much larger than an image’s actual set of objects. So, to align the ground truth with predictions, we pad ground truth with (no objects token).

The task now is to find an optimal bipartite matching between these two sets: ground truth and prediction.

Let be a set of all the possible permutations of . If , . This means that our ground truth set and the prediction set have two elements each. To find the optimal bipartite matching between the two sets, we search for a specific permutation (order) of the predictions, which gives the lowest cost when matched with the ground truth.

This specific order is defined by , where . First, let’s define the cost function to find the optimal permutation ().

Let us see an example, the cost function would then look something like this:

For :
For :

We will now choose the that gives us the lowest cost. Upon selecting the permutation that lowers the cost, we eventually get the optimal bipartite matching of the ground truth and the predicted objects. Let’s see this in action in the video below.

Optimize Object Specific Losses 🔧

Now that we have matched the predictions to the ground truth, we apply the Hungarian loss for all matched pairs.

Notice the usage of here that denotes the already optimal permutation for the predictions.

Think of this as the only set of lines in the above diagram that will give you the lowest cost. For example, this is shown in the video below.

Note from the paper: In practice, we down-weight the log-probability term when by a factor of 10 to account for class imbalance. In the matching cost, we use probabilities instead of log-probabilities. This makes the class prediction term commensurable to , and we observed better empirical performances.

The L1 loss is commonly used in object detection to measure the difference between the predicted box coordinates and the ground-truth box coordinates. However, this loss can cause issues when dealing with boxes of different sizes.

For instance, consider two boxes with the same relative error (i.e., the same proportion of error compared to the box size) but different sizes. The larger box will have a larger absolute error, which could lead the model to prioritize larger boxes over smaller ones.

A combination of L1 loss and a scale-invariant loss called generalized Intersection over Union (IoU) can address this issue.

The generalized IoU loss measures the overlap between the predicted box and the ground-truth box, considering their sizes. The full expression is shown in the video below.

Using a linear combination of L1 loss and IoU loss, the model can balance the importance of accurately predicting box coordinates and their relative sizes, leading to more consistent performance across boxes of different sizes.

Quiz Time! 🤓

Summary

As we know by now, DETR has two main components — pièce de résistance, if you will: Set Prediction Loss and Transformer Architecture. In this tutorial, we primarily focused on Set Prediction Loss.

First, we familiarized ourselves with Object Detection Set Prediction Loss.Next, we took a detour to comprehend the workings of Optimal Bipartite Matching.Subsequently, we explored how Optimal Bipartite Matching applies to objects.Finally, we delved into the mathematics behind optimizing object-specific losses.

In the upcoming and final part of this series, we will reflect on the architecture and intricacies of DETR. We will examine its building blocks, connections, and what makes this model so efficient.

Stay tuned for the upcoming parts of this series, and be sure to share your thoughts and learnings from DETR and what you’re looking forward to. Connect with us on Twitter by tagging @pyimagesearch.

Citation Information

A. R. Gosthipaty and R. Raha. “DETR Breakdown Part 2: Introduction to DEtection TRansformers,” PyImageSearch, P. Chugh, S. Huot, K. Kidriavsteva, and A. Thanki, eds., 2023, https://pyimg.co/slx2k

@incollection{ARG-RR_2023_DETR-breakdown-part2,
author = {Aritra Roy Gosthipaty and Ritwik Raha},
title = {{DETR} Breakdown Part 2: Introduction to DEtection TRansformers},
booktitle = {PyImageSearch},
editor = {Puneet Chugh and Susan Huot and Kseniia Kidriavsteva and Abhishek Thanki},
year = {2023},
url = {https://pyimg.co/slx2k},
}

To download the source code to this post (and be notified when future tutorials are published here on PyImageSearch), simply enter your email address in the form below!

The post DETR Breakdown Part 2: Methodologies and Algorithms appeared first on PyImageSearch.

 Table of Contents DETR Breakdown Part 2: Methodologies and Algorithms The DETR Model 👁️ Object Detection Set Prediction Loss 📉 Optimal Bipartite Matching 🔄 Optimal Bipartite Matching for Objects 🌐 Optimize Object Specific Losses 🔧 Quiz Time! 🤓 Summary Citation…
The post DETR Breakdown Part 2: Methodologies and Algorithms appeared first on PyImageSearch.  Read More Deep Learning, Object Detection, Paper Review, Set Prediction, Transformers, Tutorial, detr, end-to-end object detection, object detection, object detection with transformers, set prediction loss, transformer, tutorial