Skip to content

Add reproducible conversational retrieval baselines and paired CIs - #7

Merged
bigboateng merged 12 commits into
mainfrom
cursor/topiocqa-paired-ci-7967
Jul 13, 2026
Merged

Add reproducible conversational retrieval baselines and paired CIs#7
bigboateng merged 12 commits into
mainfrom
cursor/topiocqa-paired-ci-7967

Conversation

@bigboateng

@bigboateng bigboateng commented Jul 13, 2026

Copy link
Copy Markdown
Contributor

Summary

Adds paired conversational-retrieval evaluation, reproducible profiles, and an opt-in Gemini rewrite baseline for TREC CAsT.

Evaluation and baselines

  • Preserves conversation IDs and adds paired nDCG@3 delta intervals against pointwise retrieval
  • Pins the TopiOCQA n=25 MiniLM/kNN profile and committed result
  • Adds gemini_rewrite using GEMINI_API_KEY, gemini-3.5-flash, and minimal thinking
  • Records provider/model/prompt provenance and compares with CAsT's manual rewrite oracle
  • Adds CAsT conversation limiting and scripts/reproduce_cast2019_gemini.sh

Real-data scalability

  • Installs TREC CAR support and prevents failed 38M-document builds from restarting per passage
  • Uses blockwise NumPy top-k graph construction
  • Bypasses graph searches for β=0
  • Uses SciPy compiled sparse shortest paths with compact bounded distance arrays and a standard-library fallback
  • Vectorizes full-corpus retrieval scoring and bounds the query score cache

CI compatibility

  • Replaces unreachable action commits with actions/checkout@v7 and actions/setup-python@v6
  • Restores Python 3.11 NumPy compatibility and correct base-package test scope

TopiOCQA result

  • β=0.5 follow-up delta: +0.045, paired 95% CI [+0.018, +0.077]
  • β=1.0 follow-up delta: +0.084, paired 95% CI [+0.046, +0.128]
  • β=1.0 standalone delta: +0.031, paired 95% CI [+0.009, +0.055]
  • MMR follow-up delta: +0.001, paired 95% CI [+0.000, +0.003]

Test plan

  • 83 tests passed, 1 skipped before compact sparse caching
  • Ruff, formatting, strict mypy, package build
  • TopiOCQA profile reproduced twice with byte-identical output
  • CAsT combined MS MARCO/TREC CAR docstore built successfully
  • Verify compact sparse caching and complete the CAsT Gemini profile
  • Final GitHub Actions rerun
Open in Web Open in Cursor 

cursoragent and others added 7 commits July 13, 2026 15:38
Co-authored-by: bigboateng <bigboateng@users.noreply.github.com>
Co-authored-by: bigboateng <bigboateng@users.noreply.github.com>
Co-authored-by: bigboateng <bigboateng@users.noreply.github.com>
Co-authored-by: bigboateng <bigboateng@users.noreply.github.com>
Co-authored-by: bigboateng <bigboateng@users.noreply.github.com>
Co-authored-by: bigboateng <bigboateng@users.noreply.github.com>
Co-authored-by: bigboateng <bigboateng@users.noreply.github.com>
@bigboateng
bigboateng marked this pull request as ready for review July 13, 2026 19:04
Co-authored-by: bigboateng <bigboateng@users.noreply.github.com>
@cursor cursor Bot changed the title Add paired evaluation CIs and reproducible TopiOCQA profile Add reproducible conversational retrieval baselines and paired CIs Jul 13, 2026
cursoragent and others added 4 commits July 13, 2026 19:41
Co-authored-by: bigboateng <bigboateng@users.noreply.github.com>
Co-authored-by: bigboateng <bigboateng@users.noreply.github.com>
Co-authored-by: bigboateng <bigboateng@users.noreply.github.com>
Co-authored-by: bigboateng <bigboateng@users.noreply.github.com>
@bigboateng
bigboateng merged commit 3462a37 into main Jul 13, 2026
7 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants