Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,17 @@
|
|
| 1 |
---
|
| 2 |
license: mit
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
+
datasets:
|
| 4 |
+
- imagenet-1k
|
| 5 |
+
metrics:
|
| 6 |
+
- accuracy
|
| 7 |
+
pipeline_tag: image-classification
|
| 8 |
---
|
| 9 |
+
# mrl-convnext-tiny
|
| 10 |
+
|
| 11 |
+
We provide a ConvNeXt-Tiny model trained with [Matryoshka Representation Learning](https://github.com/RAIVNLab/MRL) (MRL) from scratch on ImageNet-1K,
|
| 12 |
+
adapted from [the original repository](https://github.com/facebookresearch/ConvNeXt). We train with a Matryoshka Loss at [d] = {48, 96, 192, 384, 786},
|
| 13 |
+
|
| 14 |
+
<p align="center">
|
| 15 |
+
<img src="https://cdn-uploads.huggingface.co/production/uploads/64af72d4a609b29cc7b5919b/eGorc5hWdriNcEJkwP7Lz.png" alt="drawing" width="600"/>
|
| 16 |
+
Top-1 Retrieval Accuracy (1-nearest neighbor) of mrl-convnext-tiny on ImageNet-1K with Exact search and IVF search across vector dimensionality.
|
| 17 |
+
</p>
|