TAROT: Targeted Data Selection via Optimal Transport

Lan Feng, Fan Nie, Yuejiang Liu, Alexandre Alahi

Visual Intelligence for Transportation Lab, EPFL

Stanford University

Paper

---

🛠 Environment Setup

To get started with TAROT, follow these steps:

• Install GeomLoss.
• Install Trak.

---

🚀 Quick Start

🚗Motion Prediction

📖Instruction Tuning

🏞️Semantic Segmentation

📊Qualitative Results of WFD

Acknowledgement

This repo relies on the TRAK implementation to compute the projected gradients. We are immensely grateful to the authors of that project.

For citation:

@article{feng2024tarot,
  title={TAROT: Targeted Data Selection via Optimal Transport},
  author={Feng, Lan and Nie, Fan and Liu, Yuejiang and Alahi, Alexandre},
  journal={arXiv preprint arXiv:2412.00420},
  year={2024}
}