Search-SWE
GPU
lightweight-query-retrieval
View on GitHub

Lightweight Query Retrieval

Reduce complete-batch retrieval time while preserving useful results across three collections.

Domain
latency-constrained retrieval
Primary focus
Lightweight Query Retrieval
Primary metric
Retrieval runtime after a macro nDCG@10 quality gate
latency-constrained retrievalsearch-swe
On this page

Runtime and requirements

  • Agent budget: 6 hours
  • Compute: 16 CPUs, 32 GiB RAM, 100 GiB storage; 1 GPU(s)

Offline A100 40 GB with supplied Llama-3.2-1B. Build: 1,200 seconds; service load: 120 seconds; public warmup: 60 seconds. Each complete hidden batch has a 120-second hard limit. Three fresh service runs are measured.

Public input and output contract

Inputs

/task/data/corpus.jsonl, labeled /task/data/examples.jsonl, and /task/models/Llama-3.2-1B. Examples are absent in verification. Evaluation has 1,271 queries across three collections.

Outputs

Submit /app/build.sh --corpus PATH --index-dir DIR and /app/run.sh --index-dir DIR --serve under /app. Index becomes read-only; use supplied temporary storage during serving. Emit {"status":"ready"} after loading.

Each stdin request contains queries with query_id, collection, text, instruction, plus top_k:10. Return a results array covering every query exactly once, each with at most ten unique collection-valid doc_id/finite descending score results. Flush one JSON response per request.

Evaluation

Every run must reach collection-macro linear-gain nDCG@10 >=0.45. Let T be median complete-batch retrieval seconds across three fresh runs, excluding build/load/warmup. Reward is clip((0.8-T)/0.8,0,1) after all gates. At or above 0.8 seconds reward is zero; full credit is the theoretical zero-runtime limit, not a positive-time plateau. Invalid output, execution or resource failures zero the task.

Final reward requires a healthy, accepting independent trajectory and submission integrity audit. Audit failures are reported separately from retrieval quality and cannot produce positive credit.

See the full instructions, task documentation and runtime configuration for the complete contract at the synchronized benchmark revision.