anindyanag58 commited on
Commit
a600aad
·
verified ·
1 Parent(s): 06f5a72

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +36 -0
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).