OBB-PBD: Extending Parallel Box Decoding to Oriented Bounding Boxes for Referring Expression Grounding in Aerial Imagery
Organizations: Department of Computer Science & Engineering, Chungnam National University, Daejeon 34134, Republic of Korea
Abstract
Referring expression grounding in overhead imagery requires oriented bounding boxes rather than horizontal ones, since aerial objects appear at arbitrary orientations. Most grounding models emit box coordinates sequentially; Parallel Box Decoding (PBD) instead predicts all geometry slots in one forward pass, but only for four-coordinate horizontal boxes. We extend this block-aligned interface to a six-slot oriented representation using a double-angle parameterization. Rather than assuming correctness from a decreasing loss curve, we verify the extension with image/prompt-shuffle controls and visibility/leakage probes, which show that the corrected model depends on both the image and the prompt rather than on prompt statistics alone. On the aerial vehicle visual grounding (AVVG) benchmark, whose targets are genuinely oriented, the extended model improves rotated intersection over union (rotIoU) from 0.1253 for the zero-shot baseline to 0.303 +/- 0.012 (mean +/- standard deviation over three training seeds) across the complete 6139-record test split spanning all 120 images. Learning oriented boxes rather than their axis-aligned envelopes adds 0.055 +/- 0.008 rotIoU beyond a matched control (per-seed 95% confidence intervals all excluding zero), whereas block expansion alone yields no rotIoU gain distinguishable from zero.
Figures & tables
| Setting | Value |
| Dataset | AVVG subset of the RefGeo-derived data ( Zhou et al., 2024 ) |
| Training samples | 26,451 usable records after removing 31 out-of-frame annotations, split into a 421-image training set (22,495 records) and a 74-image held-out validation set (3956 records) disjoint from the test split (Section 3.5 ) |
| Image source | 495 DJI UAV aerial images (421 training + 74 validation, disjoint from the 120-image test split), pixels |
| Optimizer | AdamW (weight decay 0.01), constant learning rate (no schedule, no warmup) |
| Batch and precision | 1 per GPU, gradient accumulation 4, 3 GPUs (global batch size 12); bfloat16 (auxiliary losses computed in float32) |
| Loss weights | , , |
| Method | Inference Path | aabbIoU | rotIoU | [email protected] | [email protected] | Angle MAE ( ∘ ) | Decoded |
| GT HBB envelope (ref.) | GT-derived | 1.0000 | 0.6037 | 78.6% | 16.7% | 32.33 | – |
| LocateAnything (zero-shot) | Upstream generate() | 0.1512 | 0.1253 | 15.1% | 3.3% | – | 99.9% |
| OBB-PBD (ours) | Direct PBD | 0.334 0.012 | 0.303 0.012 | 37.1 1.5% | 22.0 0.8% | 9.98 0.34 | 99.9% |
| Step | aabbIoU | rotIoU |
| Baseline HBB-PBD, 4-slot (adaptation + decoding path) † | ||
| HBB-PBD HBB-envelope, 6-slot (format-switch effect) | ||
| HBB-envelope ours (orientation-learning effect) | ||
| Baseline ours (total, Table 2 ) |
| Condition | rotIoU | Per-Seed 95% CIs | [email protected] | Angle MAE ( ∘ ) |
| Matched image and prompt | 0.303 0.012 | – | 22.0 0.8% | 9.98 0.34 |
| Image shuffled | 0.006 0.001 | – | 0.2 0.0% | 39.46 0.19 |
| Prompt shuffled | 0.044 0.002 | – | 3.1 0.3% | 19.18 0.28 |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| ID | Symptom | Cause and Correction |
| U1 | Eight-token block cannot be constructed. | Block size capped by a hard-coded min(4, block_size) ; now read from the configuration. |
| U2 | Crash in training mode. | Forward pass returned a tuple instead of CausalLMOutputWithPast ; return type corrected. |
| U3 | Backward pass fails (in-place write into a leaf view). | Embeddings reshaped for image-token injection remained a leaf view; .clone() added. |
| U4 | Evaluation loss crashes when input_ids=None . | Mask selection keyed only on self.training ; input_ids=None calls now use the training mask. |
| U5 | No crash; loss decreases, but coordinate slots see no image or prompt tokens (on one batch, the first coordinate slot attended to 7 of 894 keys; 894 of 894 after the fix). | Causal-prefix length derived from position-ID drops that training never produces ( for every sample); now recovered from the mask-token region, as at inference. |
| Arm | Seed | Epoch | rotIoU [95% CI] | aabbIoU [95% CI] | [email protected] | Angle MAE ( ∘ ) |
| OBB-PBD | 1 | 5 | 0.3124 | 0.3441 | 22.8% | 9.59 |
| 2 | 5 | 0.3062 | 0.3378 | 22.1% | 10.16 | |
| 3 | 4 | 0.2898 | 0.3207 | 21.2% | 10.20 | |
| HBB-envelope | 1 | 5 | 0.2527 | 0.3465 | 3.1% | 22.01 |
| 2 | 5 | 0.2476 | 0.3432 | 2.9% | 22.31 | |
| 3 | 5 | 0.2442 | 0.3420 | 2.3% | 22.09 |
| Step | Seed | rotIoU [95% CI] | aabbIoU [95% CI] |
| Baseline HBB-PBD (adaptation + decoding path) † | 1 | ||
| 2 | |||
| 3 | |||
| HBB-PBD HBB-envelope (format switch) | 1 | ||
| 2 | |||
| 3 |
Visual Grounding
Visual grounding connects language to the image regions, objects, or pixels that support a description, question, or action.
See all 155 papers