Skip to content

JPQ etc#47

Draft
cmacdonald wants to merge 525 commits into
mainfrom
no_epochs
Draft

JPQ etc#47
cmacdonald wants to merge 525 commits into
mainfrom
no_epochs

Conversation

@cmacdonald
Copy link
Copy Markdown
Collaborator

@cmacdonald cmacdonald commented Apr 9, 2026

This is the PR corresponding to changes made for the JPQ reproducibility paper: A Replicability Study of Joint Product Quantisation for Effective Space-Efficient Dense Retrieval (Macdonald, Tonellotto & Shen. SIGIR 2026).

Functionality:

  • a Trainer for making a JPQ model and index
  • more models (STAR, Lion, RepLLama etc); note that STAR and Lion involved copying in code from the upstream repos.
  • a encode_queries_torch() method added to all models, to allow differentiation/training of query encoders

@cmacdonald cmacdonald marked this pull request as draft April 9, 2026 12:32
Copilot AI and others added 14 commits April 9, 2026 14:52
* Fix all ruff issues in jpq module (F401, E401, F541, E713, E402, E741, F403, F405, F811, E731, E701)

Agent-Logs-Url: https://github.com/terrierteam/pyterrier_dr/sessions/9a61345a-fe00-41eb-83b1-7ca3f786e057

Co-authored-by: cmacdonald <620938+cmacdonald@users.noreply.github.com>


---------

Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>
Co-authored-by: cmacdonald <620938+cmacdonald@users.noreply.github.com>
Co-authored-by: Craig Macdonald <craig.macdonald@glasgow.ac.uk>
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.

4 participants