license:cc-by-sa-4.0 size_categories:10K<n<100K format:parquet modality:image modality:text library:datasets library:pandas library:mlcroissant library:polars region:us
---
license: cc-by-sa-4.0
configs:
- config_name: default
  data_files:
  - split: train
    path: data/train-*
dataset_info:
  features:
  - name: image
    dtype: string
  - name: text
    dtype: string
  - name: paper_id
    dtype: string
  - name: figure_idx
    dtype: int64
  splits:
  - name: train
    num_bytes: 7047341
    num_examples: 17094
  download_size: 3047908
  dataset_size: 7047341
---
# ArXiv AI Figure-Caption Dataset

This dataset contains figures and their captions extracted from recent AI-related papers on arXiv.

## Dataset Information
- Total figures: 17094
- Source: arXiv papers from 2023.12 onwards
- Categories: CS.AI, CS.LG, CS.CV, CS.CL, CS.NE, STAT.ML

## Content Format
Each entry contains:
- `image`: URL of the figure image
- `text`: Caption text
- `paper_id`: arXiv paper ID
- `figure_idx`: Index of the figure in the paper

## Data Quality
- All captions have passed quality filters for length and content
- Images are verified for appropriate dimensions and format

## License
This dataset is released under CC BY-SA 4.0 License. This means you are free to:
- Share: Copy and redistribute the material in any medium or format
- Adapt: Remix, transform, and build upon the material for any purpose

Under the following terms:
- Attribution: You must give appropriate credit, provide a link to the license, and indicate if changes were made
- ShareAlike: If you remix, transform, or build upon the material, you must distribute your contributions under the same license

## Citation
If you use this dataset in your research, please cite it as:
```bibtex
@dataset{arxiv_figures,
  author       = {Yiming Liu},
  title        = {ArXiv AI Figure-Caption Dataset},
  year         = {2025},
  publisher    = {HuggingFace},
  url          = {https://huggingface.co/datasets/Leonardo6/ArXiv-AI-Figure-Caption}
}
```
## Acknowledgments
All figures and captions are sourced from papers on arXiv.org. We acknowledge the authors of all papers included in this dataset.
