Upload README.md with huggingface_hub
Browse files
README.md
ADDED
|
@@ -0,0 +1,36 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: cc-by-4.0
|
| 3 |
+
tags:
|
| 4 |
+
- multimodal-retrieval
|
| 5 |
+
- text-video-retrieval
|
| 6 |
+
- audio-visual
|
| 7 |
+
- hypergraph
|
| 8 |
+
---
|
| 9 |
+
|
| 10 |
+
# HyVol++: Weighted and Relational Hypergraph-Regularized Gramian Volumes
|
| 11 |
+
|
| 12 |
+
Pretrained weights by Anindya Nag, Ambuj Mehrish and Sebastiano Vascon.
|
| 13 |
+
|
| 14 |
+
HyVol++ extends HyVol ([CVML-CFU/HyVol](https://huggingface.co/CVML-CFU/HyVol)) with two
|
| 15 |
+
hypergraph variants that add semantic relations between training samples before the
|
| 16 |
+
volume loss is computed:
|
| 17 |
+
|
| 18 |
+
- **wHyVol**: weighted hypergraph with learned per-edge weights and attention.
|
| 19 |
+
- **wRHG**: weighted relational hypergraph with four edge families
|
| 20 |
+
(document, same-modality, cross-modality and friend).
|
| 21 |
+
|
| 22 |
+
The hypergraph is removed after training, so the backbone, scoring function and
|
| 23 |
+
retrieval cost are unchanged.
|
| 24 |
+
|
| 25 |
+
## Checkpoints
|
| 26 |
+
|
| 27 |
+
| File | Backbone |
|
| 28 |
+
|---|---|
|
| 29 |
+
| `wHyVol/GRAM_wHyVol/wHyVol_GRAM.pt` | GRAM + wHyVol |
|
| 30 |
+
| `wHyVol/HyperGRAM_wHyVol/wHyVol_HyperGRAM.pt` | Hyperbolic GRAM + wHyVol |
|
| 31 |
+
| `wRHG/GRAM_wRHG/wRHG_GRAM.pt` | GRAM + wRHG |
|
| 32 |
+
| `wRHG/HyperGRAM_wRHG/wRHG_HyperGRAM.pt` | Hyperbolic GRAM + wRHG |
|
| 33 |
+
|
| 34 |
+
All models are trained on a 150K-clip subset of VAST-27M and evaluated zero-shot.
|
| 35 |
+
Each checkpoint is the ITM-best one on MSR-VTT (T-VAS).
|
| 36 |
+
Load each checkpoint with the codebase of its backbone (GRAM or Hyperbolic GRAM).
|