---
title: 'Center-DETR: Center-Aware DETR for Cytology'
url: https://www.emergentmind.com/topics/center-detr
type: topic
---

# Center-DETR: Center-Aware DETR for Cytology

Center-DETR denotes a center-aware reformulation of a Co-DINO / Co-DETR detector for cervical cytology, introduced for the RIVA Cervical Cytology Challenge and built around the observation that the dataset’s annotations are rigid fixed-size bounding boxes. In that setting, the principal learning problem is not free-form box extent estimation but accurate localization of cell centers and, for Track A, Bethesda-category classification. The method therefore retains a strong DETR-family pipeline—specifically Co-DINO with a Swin-Large backbone—but reorganizes target definition, augmentation, post-processing, and loss emphasis around center prediction. Under this formulation, the reported system achieved 1st place in Track B and 2nd place in Track A [2604.02090].

## 1. Problem definition and representational shift

Center-DETR is motivated by a dataset protocol in which cell instances are annotated with boxes described as rigidly fixed at \(100 \times 100\) pixels. That annotation regime changes the semantics of detection. In ordinary object detection, regressing width and height is meaningful because object extent varies and the box is intended to capture object geometry. In the RIVA setting, the box size is largely annotation-imposed rather than object-specific, so the critical variable is the cell center rather than a freely varying \((w,h)\) pair [2604.02090].

The method therefore models detection as prediction of object centers together with class probabilities. For Track A, the class space comprises 8 Bethesda categories plus 1 background class; for Track B, localization is the primary objective. The central rationale is that standard IoU-driven box regression can become counterproductive under fixed-size annotations: a detector with a slightly shifted center may compensate by predicting larger-than-needed boxes to recover overlap, which introduces size-regression noise and degrades mAP [2604.02090].

This formulation makes Center-DETR best understood not as a wholly new DETR-family architecture from scratch, but as a task-specific adaptation of an existing end-to-end detector to a center-point prediction problem. The resulting detector is intended to answer where each cell center is, and, when required, which Bethesda category it belongs to.

## 2. Architectural basis in Co-DINO and Co-DETR

The underlying detection framework is Co-DINO within the broader Co-DETR collaborative hybrid assignment training paradigm. The backbone is Swin-Large, chosen in preference to ViT-Large because the target domain contains tiny cells, dense local structures, and subtle spatial morphology; the paper attributes to Swin hierarchical multi-scale representations with localized attention and progressively built hierarchical semantics that are better suited to such imagery [2604.02090].

The overall pipeline consists of an input image, a Swin-Large backbone for hierarchical multi-scale feature extraction, and a Co-DINO detector with object queries and denoising queries during training. In addition to the main DETR-style branch, the training graph includes one-to-many auxiliary branches: Faster R-CNN, ATSS, RetinaNet, and FCOS. These heads provide dense supervision during training and are discarded at inference time [2604.02090].

Among the auxiliary branches, FCOS is especially aligned with the center-point formulation. That alignment is not used to redefine the main DETR decoding mechanism, but it reinforces the center-aware training signal inside the collaborative hybrid assignment framework. Inference uses the main Co-DINO-Swin detector, with decoder outputs interpreted as center coordinates and class scores rather than unconstrained object boxes [2604.02090].

The architecture is therefore hybrid in a specific sense: the principal prediction pathway remains one-to-one DETR-style set prediction, while auxiliary one-to-many detectors enrich supervision during training. Center-DETR does not discard DETR matching or decoder-based prediction; it changes what the detector is expected to predict.

## 3. Center-aware augmentation and supervision

A central customization is center-preserving crop augmentation. Standard random cropping in natural-image detection can retain any object whose box overlaps the crop, but under the cytology formulation this can preserve truncated cells whose centers fall outside the visible region. The method instead keeps an object after cropping if and only if its center lies inside the crop region:
\[
x_{\min} \le c_{x_i} \le x_{\max}, \qquad y_{\min} \le c_{y_i} \le y_{\max}.
\]
Objects whose centers fall outside the crop are discarded [2604.02090].

The significance of this rule is domain-specific. The paper argues that Pap-smear interpretation depends on morphological integrity, including nucleus appearance and nucleus-to-cytoplasm ratio. If a crop removes the nucleus center, the remaining fragment may cease to be a valid positive example. Center-preserving cropping therefore aligns augmentation with the center-based target definition and reduces supervision noise from partial cells [2604.02090].

Its standalone quantitative effect is modest but consistent. On Track B, Co-DINO-Swin improves from \(0.609\) to \(0.611\), and Co-DINO-ViT improves from \(0.604\) to \(0.606\), when center-preserving cropping is added alone [2604.02090]. This suggests that the crop policy is not the dominant source of performance gain, but it regularizes training in a direction consistent with the task formulation.

## 4. Analytical geometric box optimization

The most distinctive component is an analytical post-processing rule that reconstructs a fixed-size square from a predicted center. Because ground-truth boxes are \(100 \times 100\), the paper asks a specific question: given center localization jitter, what fixed square side length \(S\) should be used at inference to maximize expected IoU with the ground truth? The reported answer is
\[
S = 101.5.
\]
The final box is therefore reconstructed from the predicted center using a fixed side length of \(101.5 \times 101.5\) [2604.02090].

The derivation is based on overlap geometry. Let \(\Delta_x = |x_c - x_{gt}|\) and \(\Delta_y = |y_c - y_{gt}|\) be center offsets between the predicted and ground-truth boxes. The overlap width and height are defined as
\[
W_{int} = \min\left(100,\ \max\left(0,\ \frac{100 + S}{2} - \Delta_x\right)\right),
\]
\[
H_{int} = \min\left(100,\ \max\left(0,\ \frac{100 + S}{2} - \Delta_y\right)\right).
\]
With intersection area \(A_{int} = W_{int}H_{int}\) and union \(100^2 + S^2 - A_{int}\), the paper optimizes the expected IoU over the empirical localization jitter distribution \(\epsilon \sim (\Delta_x,\Delta_y)\) [2604.02090].

The logic is a tradeoff. If \(S=100\), any center error immediately reduces overlap. If \(S>100\), there is a tolerance margin
\[
\gamma = \frac{S-100}{2},
\]
so small center errors can be absorbed without immediate intersection loss. But if \(S\) becomes too large, the union area grows and IoU falls. Under typical jitter of about \(1 \sim 1.5\) pixels, the optimum reported in the paper is \(S=101.5\) [2604.02090].

This geometric prior is not merely a post hoc adjustment for the proposed detector. The paper reports model-agnostic gains on Track B when the same \(101.5 \times 101.5\) reconstruction rule is applied to several baselines, indicating that the benefit is tightly linked to the fixed-size annotation regime rather than to Swin or Co-DINO alone [2604.02090].

| Detector | Track B before \(\rightarrow\) after box optimization |
|---|---|
| YOLO | \(0.460 \rightarrow 0.478\) |
| RetinaNet | \(0.507 \rightarrow 0.523\) |
| CenterNet | \(0.519 \rightarrow 0.525\) |
| Co-Deformable-DETR | \(0.586 \rightarrow 0.601\) |
| Co-DINO-ViT | \(0.604 \rightarrow 0.618\) |

A plausible implication is that, when annotation geometry is largely artificial, analytically matching that geometry can be more important than learning width and height as free semantic variables.

## 5. Optimization, assignment structure, and empirical performance

Center-DETR inherits the Co-DETR training structure: a main one-to-one DETR branch and auxiliary one-to-many heads. The total loss is given as
\[
\mathcal{L}_{Total} = \lambda_{cls}\mathcal{L}_{focal} + \lambda_{L1}\mathcal{L}_{L1} + \lambda_{IoU}\mathcal{L}_{GIoU} + \lambda_{Aux}\mathcal{L}_{Aux}.
\]
Here \(\mathcal{L}_{focal}\) is the classification loss on the main branch, \(\mathcal{L}_{L1}\) is the box regression loss, \(\mathcal{L}_{GIoU}\) is generalized IoU loss, and \(\mathcal{L}_{Aux}\) aggregates auxiliary-head losses over one-to-many assignments [2604.02090].

A central design choice is to downweight IoU-based supervision because the center-plus-fixed-size reconstruction prior reduces the need for network-driven size regression. The paper sets
\[
\lambda_{IoU} = 0.5.
\]
Track-specific reweighting then emphasizes the dominant difficulty of each challenge track. For Track A, where the principal bottleneck is classification among similar Bethesda categories, the paper uses
\[
\lambda_{cls}=3.5, \qquad \lambda_{L1}=1.5.
\]
For Track B, where localization is the objective, it uses
\[
\lambda_{cls}=1.5, \qquad \lambda_{L1}=3.5.
\]
Within the center-based formulation, the Track B choice effectively emphasizes center precision [2604.02090].

Implementation details reported in the paper include PyTorch and MMDetection, initialization from COCO pre-trained weights, training on the official training split with validation-set evaluation, and final challenge submission after merging training and validation data. Training used an NVIDIA RTX 4090 with 48 GB VRAM, and inference is reported at 12 GB VRAM. Aside from the explicit loss reweighting, the authors state that default hyperparameters were used [2604.02090].

The main comparative results show the strength of the underlying Co-DINO-Swin detector and the incremental value of the center-aware modifications. Reported baselines are: YOLO, Track A \(0.120\), Track B \(0.460\); RetinaNet, \(0.137\) and \(0.507\); CenterNet, \(0.063\) and \(0.519\); Co-Deformable-DETR, \(0.207\) and \(0.586\); Co-DINO-ViT, \(0.238\) and \(0.604\); and Co-DINO-Swin, \(0.237\) and \(0.609\) [2604.02090]. On Track B, the Co-DINO-Swin ablation proceeds as follows:

| Configuration | Track B |
|---|---|
| Baseline | \(0.609\) |
| \(+\) Center-preserving crop | \(0.611\) |
| \(+\) Crop \(+\) box optimization | \(0.634\) |
| \(+\) Crop \(+\) box optimization \(+\) loss tuning | \(0.635\) |

The ablation shows that the largest gain comes from analytical box optimization, not from cropping or loss tuning alone. This supports the paper’s claim that the decisive issue is the match between prediction geometry and the fixed-size annotation regime [2604.02090].

## 6. Relation to adjacent center-based DETR research, assumptions, and scope

Within arXiv literature, Center-DETR should be distinguished from several adjacent but non-identical uses of centers in detection and retrieval. In overhead imagery, centerpoint-only supervision has been studied through adaptations of RetinaNet and Faster R-CNN that predict object centers rather than boxes, showing that centerpoint labels can preserve most center-localization performance on xView, DOTA 1.5, and FAIR1M; however, those models are anchor-based CNN detectors rather than DETR-style set predictors [2210.01857]. In video moment retrieval, DETR-based models already predict temporal spans in \((center,\text{length})\) form, and length-aware matching has been proposed to improve short-moment center prediction indirectly through length-specialized queries rather than through a new center head [2412.20816].

At the level of DETR query semantics, SAP-DETR offers an explicit counterpoint to center-prior designs. It argues that centralizing query reference points can reduce query saliency and create redundant predictions, and proposes query-specific salient points with \([\ell,t,r,b]\) box parameterization instead of center-based semantics [2211.02006]. In dense pedestrian detection, PED analyzes why vanilla DETR and deformable DETR underperform in crowd scenes and responds with dense queries, rectified attention fields, visible-part supervision, and faster matching; that work is query-based and box-guided rather than center-supervised, but it is relevant to any Center-DETR-like design intended for dense scenes [2012.06785].

Against that background, the most precise characterization of Center-DETR is narrow and task-specific. It is a center-aware Co-DINO / Co-DETR detector specialized for a fixed-size annotation regime in cervical cytology, not a universal redefinition of DETR around center-only prediction [2604.02090]. Its assumptions are explicit: fixed-size annotations must be meaningful, center localization must carry most of the information needed for detection, localization jitter should be small enough that \(S=101.5\) remains near-optimal, and center-preserving cropping should be appropriate for the domain [2604.02090].

The paper also states that the method is largely engineering-driven to maximize mAP, with less emphasis on deeper clinical or biological modeling. This suggests that the design is highly effective for challenge conditions yet somewhat specialized. A plausible implication is that the transferable insight is conditional rather than universal: if a dataset’s annotation protocol makes box size largely artificial, then a DETR-family detector can profitably be reformulated around center prediction and an analytically chosen geometric prior; if object size and shape are semantically meaningful, the same reformulation may become suboptimal [2604.02090].

Source: https://www.emergentmind.com/topics/center-detr