MMCL: Correcting Content Query Distributions for Improved Anti-Overlapping X-Ray Object Detection
Fuente:
arXiv
Gespeichert in:
| Hauptverfasser: | , , , , , , , , , |
|---|---|
| Format: | Preprint |
| Veröffentlicht: |
2024
|
| Schlagworte: | |
| Online-Zugang: | |
| Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
| _version_ | 1866912699872444416 |
|---|---|
| author | Li, Mingyuan Jia, Tong Lu, Hui Wang, Hao Ma, Bowen Guo, Shiyi Lin, Shuyang Chen, Dongyue Wang, Haoran Yu, Baosheng |
| author_facet | Li, Mingyuan Jia, Tong Lu, Hui Wang, Hao Ma, Bowen Guo, Shiyi Lin, Shuyang Chen, Dongyue Wang, Haoran Yu, Baosheng |
| contents | Unlike natural images with occlusion-based overlap, X-ray images exhibit depth-induced superimposition and semi-transparent appearances, where objects at different depths overlap and their features blend together. These characteristics demand specialized mechanisms to disentangle mixed representations between target objects (e.g., prohibited items) and irrelevant backgrounds. While recent studies have explored adapting detection transformers (DETR) for anti-overlapping object detection, the importance of well-distributed content queries that represent object hypotheses remains underexplored. In this paper, we introduce a multi-class min-margin contrastive learning (MMCL) framework to correct the distribution of content queries, achieving balanced intra-class diversity and inter-class separability. The framework first groups content queries by object category and then applies two proposed complementary loss components: a multi-class exclusion loss to enhance inter-class separability, and a min-margin clustering loss to encourage intra-class diversity. We evaluate the proposed method on three widely used X-ray prohibited-item detection datasets, PIXray, OPIXray, and PIDray, using two backbone networks and four DETR variants. Experimental results demonstrate that MMCL effectively enhances anti-overlapping object detection and achieves state-of-the-art performance on both datasets. Code will be made publicly available on GitHub. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2406_03176 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | MMCL: Correcting Content Query Distributions for Improved Anti-Overlapping X-Ray Object Detection Li, Mingyuan Jia, Tong Lu, Hui Wang, Hao Ma, Bowen Guo, Shiyi Lin, Shuyang Chen, Dongyue Wang, Haoran Yu, Baosheng Computer Vision and Pattern Recognition Unlike natural images with occlusion-based overlap, X-ray images exhibit depth-induced superimposition and semi-transparent appearances, where objects at different depths overlap and their features blend together. These characteristics demand specialized mechanisms to disentangle mixed representations between target objects (e.g., prohibited items) and irrelevant backgrounds. While recent studies have explored adapting detection transformers (DETR) for anti-overlapping object detection, the importance of well-distributed content queries that represent object hypotheses remains underexplored. In this paper, we introduce a multi-class min-margin contrastive learning (MMCL) framework to correct the distribution of content queries, achieving balanced intra-class diversity and inter-class separability. The framework first groups content queries by object category and then applies two proposed complementary loss components: a multi-class exclusion loss to enhance inter-class separability, and a min-margin clustering loss to encourage intra-class diversity. We evaluate the proposed method on three widely used X-ray prohibited-item detection datasets, PIXray, OPIXray, and PIDray, using two backbone networks and four DETR variants. Experimental results demonstrate that MMCL effectively enhances anti-overlapping object detection and achieves state-of-the-art performance on both datasets. Code will be made publicly available on GitHub. |
| title | MMCL: Correcting Content Query Distributions for Improved Anti-Overlapping X-Ray Object Detection |
| topic | Computer Vision and Pattern Recognition |
| url | https://arxiv.org/abs/2406.03176 |