Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,30 @@
|
|
1 |
-
|
2 |
-
|
3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# LVM
|
2 |
+
|
3 |
+
This is the model implementation of the CVPR 2024 'Sequential Modeling Enables Scalable Learning for Large Vision Models'. (https://arxiv.org/abs/2312.00785)
|
4 |
+
|
5 |
+
LVM is a vision pretraining model that converts various kinds of visual data into visual sentences and performs next-token prediction autoregressively. It is compatible with both GPU and TPU.
|
6 |
+
|
7 |
+
LVM is built on top of [OpenLLaMA](https://github.com/openlm-research/open_llama) (an autoregressive model) and [OpenMuse](https://github.com/huggingface/open-muse) (a VQGAN that converts images into visual tokens).
|
8 |
+
|
9 |
+
This was trained in collaboration with HuggingFace. Thanks [Victor Sanh](https://huggingface.co/VictorSanh) for the support in this project.
|
10 |
+
|
11 |
+
## Key Differences from the Original Paper Version
|
12 |
+
1. We are currently releasing the 7B model (previously 3B). Additional model size variants will be available soon.
|
13 |
+
2. Deep filtering (including quality filters, deduplication, and known CSAM content removal) has been applied to the LAION dataset, reducing the dataset size from 1.5B to 1.2B images.
|
14 |
+
|
15 |
+
3. The tokenizer has been improved for better performance.
|
16 |
+
|
17 |
+
## License
|
18 |
+
LVM is licensed under the Apache 2.0 License.
|
19 |
+
|
20 |
+
|
21 |
+
## Citation
|
22 |
+
If you found LVM useful in your research or applications, please cite our work using the following BibTeX:
|
23 |
+
\`\`\`bibtex
|
24 |
+
@article{bai2023sequential,
|
25 |
+
title={Sequential modeling enables scalable learning for large vision models},
|
26 |
+
author={Bai, Yutong and Geng, Xinyang and Mangalam, Karttikeya and Bar, Amir and Yuille, Alan and Darrell, Trevor and Malik, Jitendra and Efros, Alexei A},
|
27 |
+
journal={arXiv preprint arXiv:2312.00785},
|
28 |
+
year={2023}
|
29 |
+
}
|
30 |
+
\`\`\`
|