Visually Similar Pair Alignment for Robust Cross-Domain Object Detection
Onkar Krishna, Hiroki Ohashi
TL;DR
This work tackles the challenge of unsupervised domain adaptation for object detection by showing that aligning visually similar pairs across domains—instead of pairing arbitrary source and target instances—improves transfer. It introduces a memory-augmented framework with separate foreground and background memories that retrieve visually similar source features for alignment, coupled with a targeted foreground triplet-like alignment and a background adversarial module. Across adverse weather, synthetic-to-real, and real-to-artistic shifts, the approach achieves state-of-the-art results (e.g., 53.1% mAP on Foggy Cityscapes and 62.3% mAP on Sim10k) and demonstrates the benefits of memory-based, visually aware domain alignment. The work also provides a customized cross-domain dataset with controlled visual attributes and analyzes memory design choices, demonstrating robust gains and practical efficiency improvements through memory subsampling.
Abstract
Domain gaps between training data (source) and real-world environments (target) often degrade the performance of object detection models. Most existing methods aim to bridge this gap by aligning features across source and target domains but often fail to account for visual differences, such as color or orientation, in alignment pairs. This limitation leads to less effective domain adaptation, as the model struggles to manage both domain-specific shifts (e.g., fog) and visual variations simultaneously. In this work, we demonstrate for the first time, using a custom-built dataset, that aligning visually similar pairs significantly improves domain adaptation. Based on this insight, we propose a novel memory-based system to enhance domain alignment. This system stores precomputed features of foreground objects and background areas from the source domain, which are periodically updated during training. By retrieving visually similar source features for alignment with target foreground and background features, the model effectively addresses domain-specific differences while reducing the impact of visual variations. Extensive experiments across diverse domain shift scenarios validate our method's effectiveness, achieving 53.1 mAP on Foggy Cityscapes and 62.3 on Sim10k, surpassing prior state-of-the-art methods by 1.2 and 4.1 mAP, respectively.
