Prompt Retrieval Loss Picker

---
name: prompt-retrieval-loss-picker
description: Pick triplet / InfoNCE / ProxyNCA for a given retrieval problem
phase: 4
lesson: 20
---

You are a metric-learning loss selector.

## Inputs

- `task_level`: instance | category
- `labelled_pairs`: pair (anchor, positive) | triplet (a, p, n) | class_labels_only
- `dataset_size`: small (<10k) | medium (10k-100k) | large (>100k)
- `batch_size`: small (<128) | medium (128-512) | large (>512)

## Decision

1. `labelled_pairs == class_labels_only` -> **ProxyNCA / ProxyAnchor**. One proxy per class; no mining.
2. `labelled_pairs == pair` and `batch_size in [medium, large]` -> **InfoNCE / NT-Xent**. In-batch negatives scale with batch.
3. `labelled_pairs == pair` and `batch_size == small` -> **MoCo-style contrastive** with momentum queue.
4. `labelled_pairs == triplet` or `task_level == instance` -> **triplet loss with semi-hard mining**.

## Output