Through the PRISm: Importance-Aware Scene Graphs for Semantic Image Retrieval
Organizations: National Technical University of Athens Athens, Greece
Abstract
Web-based visual search increasingly handles queries driven by semantic intent, focusing on what is happening in an image rather than merely what it looks like. While global embeddings from vision-language models capture overall appearance, they often blur fine-grained relational structure. Conversely, scene graphs make these relationships explicit, but generating them automatically for unannotated web images produces dense, noisy graphs that retain every detected entity and relation regardless of relevance. To address this, we introduce PRISm (Pruning-based Image Retrieval via Importance Prediction on Semantic Graphs), an image-to-image retrieval framework that learns to focus on what matters in a scene. PRISm employs an Importance Prediction Module, supervised by caption-derived relevance strictly during training, to score and prune non-essential objects and <subject, relation, object> triplets. An Edge-Aware Contextual GNN then aggregates relation conditioned neighbor messages, fusing the refined graph with global visual features into a single, index-ready embedding per image. On PSG, PRISm outperforms the strongest baseline by points in MAP@1. Furthermore, on Flickr with machine-generated scene graphs and captions, PRISm remains best on top-1 metrics, while adding only 33,ms per query beyond scene graph generation.
Figures & tables
| Method | PSG Dataset | Flickr Dataset | |||||||||||||
| NDCG | MAP | MRR | NDCG | MAP | MRR | ||||||||||
| @1 | @3 | @5 | @1 | @3 | @5 | - | @1 | @3 | @5 | @1 | @3 | @5 | - | ||
| Vision | DINOv3 | 78.27 | 78.35 | 78.56 | 73.81 | 80.41 | 78.98 | 83.00 | 50.84 | 49.56 | 49.22 | 55.10 | 64.32 | 63.19 | 68.33 |
| MambaVision | 75.32 | 75.77 | 76.18 | 68.88 | 76.28 | 75.46 | 79.14 | 50.32 | 50.80 | 50.78 | 56.20 | 64.20 | 60.24 | 68.76 | |
| VL | BLIP-2 | 80.66 | 80.86 | 80.90 | 77.47 | 83.78 | 82.25 | 85.81 | 52.34 | 52.17 | 51.87 | 60.23 | 68.92 | 67.57 | 72.57 |
| CLIP | 81.28 | 81.49 | 81.59 | 78.96 | 85.12 | 83.64 | 86.90 | 52.47 | 52.59 | 52.40 | 60.59 | 68.60 | 67.79 | 72.59 | |
| PSG | Flickr | |
| Graphs | 47,874 | 12,000 |
| Nodes | 535,441 | 139,577 |
| Edges | 268,114 | 139,811 |
| Obj. classes | 133 | 132 |
| Rel. types | 56 | 48 |
| Nodes / graph | 11.18 | 11.63 |
| Setting | NDCG@1 | MAP@1 | MRR | |
| With Vision | PRISm | 88.54 | 91.62 | 95.09 |
| - Importance Pruning | 88.05 | 90.57 | 94.40 | |
| - Edge-Aware Layer | 88.39 | 91.46 | 95.04 | |
| - Global Visual Stream | 83.21 | 82.09 | 88.70 | |
| Without Vision | PRISm* | 74.09 | 64.93 | 76.84 |
| - Edge-Aware Layer | 73.70 | 64.15 | 76.01 |
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
| Setting | NDCG@1 | MAP@1 | MRR |
| PRISm | 88.54 | 91.62 | 95.09 |
| No oversampling | 87.79 | 90.81 | 94.45 |
| No weighted loss | 88.02 | 91.25 | 94.74 |
Image Retrieval
Image retrieval finds and ranks images from visual, textual, or combined queries across large and specialized collections.
See all 85 papers