Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,89 @@
|
|
| 1 |
---
|
| 2 |
license: llama2
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: llama2
|
| 3 |
+
base_model:
|
| 4 |
+
- lmsys/vicuna-7b-v1.5
|
| 5 |
+
tags:
|
| 6 |
+
- vision-language-model
|
| 7 |
+
- TEM,microscopy
|
| 8 |
+
- materials-science
|
| 9 |
+
- llava
|
| 10 |
+
- scientific-VLM
|
| 11 |
+
language:
|
| 12 |
+
- en
|
| 13 |
+
pipeline_tag: image-text-to-text
|
| 14 |
---
|
| 15 |
+
|
| 16 |
+
|
| 17 |
+
# ATOMIC-LLaVA
|
| 18 |
+
|
| 19 |
+
ATOMIC-LLaVA is a domain-specific Vision-Language Model for Transmission Electron Microscopy (TEM), fine-tuned from LLaVA-v1.5-7B (Vicuna-v1.5-7B) using a two-stage training pipeline on 32,564 TEM subfigures collected from Nature portfolio journals.
|
| 20 |
+
|
| 21 |
+
This model is introduced in the ECCV 2026 paper:
|
| 22 |
+
|
| 23 |
+
> **ATOMIC: A Domain-Specific Vision-Language Model for Transmission Electron Microscopy**
|
| 24 |
+
|
| 25 |
+
For code, evaluation scripts, and dataset, please refer to our GitHub repository:
|
| 26 |
+
๐ [https://github.com/SemiMRTLab-NCKU/ATOMIC](https://github.com/SemiMRTLab-NCKU/ATOMIC)
|
| 27 |
+
|
| 28 |
+
---
|
| 29 |
+
|
| 30 |
+
## Model Details
|
| 31 |
+
|
| 32 |
+
| | |
|
| 33 |
+
|---|---|
|
| 34 |
+
| **Base Model** | LLaVA-v1.5-7B (Vicuna-v1.5-7B) |
|
| 35 |
+
| **Training Stage** | Stage 1 (alignment) + Stage 2 (instruction tuning) |
|
| 36 |
+
| **Training Data** | 120K Stage 1 pairs + 60K Stage 2 conversations |
|
| 37 |
+
| **Domain** | Transmission Electron Microscopy (TEM) |
|
| 38 |
+
| **Modalities** | CTEM, HR-TEM, STEM, Diffraction |
|
| 39 |
+
|
| 40 |
+
---
|
| 41 |
+
|
| 42 |
+
## Important: Inference Requirements
|
| 43 |
+
|
| 44 |
+
ATOMIC-LLaVA is built on LLaVA and **cannot be loaded directly via `transformers`**. Inference requires the LLaVA repository.
|
| 45 |
+
|
| 46 |
+
**Step 1 โ Clone LLaVA:**
|
| 47 |
+
```bash
|
| 48 |
+
git clone https://github.com/haotian-liu/LLaVA.git
|
| 49 |
+
cd LLaVA
|
| 50 |
+
pip install -e .
|
| 51 |
+
```
|
| 52 |
+
|
| 53 |
+
**Step 2 โ Download weights:**
|
| 54 |
+
```python
|
| 55 |
+
from huggingface_hub import snapshot_download
|
| 56 |
+
snapshot_download(repo_id="LabSmart/ATOMIC-LLaVA", local_dir="./ATOMIC-LLaVA")
|
| 57 |
+
```
|
| 58 |
+
|
| 59 |
+
**Step 3 โ Run inference using our evaluation scripts:**
|
| 60 |
+
|
| 61 |
+
Please refer to `evaluation/` in our GitHub repository for inference and evaluation scripts.
|
| 62 |
+
|
| 63 |
+
---
|
| 64 |
+
|
| 65 |
+
## Training Data
|
| 66 |
+
|
| 67 |
+
Training data is available on HuggingFace:
|
| 68 |
+
๐ [https://huggingface.co/datasets/LabSmart/ATOMIC_dataset](https://huggingface.co/datasets/LabSmart/ATOMIC_dataset)
|
| 69 |
+
|
| 70 |
+
---
|
| 71 |
+
|
| 72 |
+
## Citation
|
| 73 |
+
|
| 74 |
+
```bibtex
|
| 75 |
+
@inproceedings{atomic2026eccv,
|
| 76 |
+
title = {ATOMIC: A Domain-Specific Vision-Language Model
|
| 77 |
+
for Transmission Electron Microscopy},
|
| 78 |
+
author = {Tu, C. and Hsu, Shu-han and others},
|
| 79 |
+
booktitle = {Proceedings of ECCV 2026},
|
| 80 |
+
year = {2026},
|
| 81 |
+
note = {BibTeX will be updated upon publication}
|
| 82 |
+
}
|
| 83 |
+
```
|
| 84 |
+
|
| 85 |
+
---
|
| 86 |
+
|
| 87 |
+
## License
|
| 88 |
+
|
| 89 |
+
This model is released under the [LLaMA 2 Community License](https://ai.meta.com/llama/license/). It is intended for academic research purposes only and may not be used for commercial purposes.
|