Skip to content

[BUG] Low recall with CAGRA when sparsity or dimensionality is high #761

Description

@lokeshwer

Describe the bug
Low recall with CAGRA when sparsity or dimensionality (n_cols) is high.

Steps/Code to reproduce bug
run_filtered_search_test
Execute this test against n_cols 1024 and sparsity > 0.95

Expected behavior
The recall should be > 0.7

Environment details (please complete the following information):

  • Environment location: Docker, Azure A100
  • Method of RAFT install: conda, Docker
    • docker pull rapidsai/base:25.04a-cuda12.8-py3.12

Additional context
I created a 3 Million vector 1024 dimension embedding from texts for building a RAG system.
I created a bitset to ensure that RAG queries are relevant. However, it started to perform poorly on recall. There were cases where without bitset the passages were relevant.
I think it is due to high dimensionality and high sparsity. Any ways to circumvent this?

P.S The recall is better in 25.04 over 25.02 as far as I tested.

Image

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions