feat: support LAION large TopK benchmarks - #834
Conversation
|
[APPROVALNOTIFIER] This PR is NOT APPROVED This pull-request has been approved by: jamesgao-jpg The full list of commands accepted by this bot can be found here. DetailsNeeds approval from an approver in each of these files:Approvers can indicate their approval by writing |
|
Please take a look at this PR to see if it's needed. |
|
@jamesgao-jpg |
@frankleaf It should be ready for testing, and I did some preliminary testing on Zilliz Cloud already and got some results (not published yet), and the end to end result looks reasonable. |
Signed-off-by: jamesgao-jpg <james.gao@zilliz.com>
Signed-off-by: jamesgao-jpg <james.gao@zilliz.com>
Signed-off-by: jamesgao-jpg <james.gao@zilliz.com>
Signed-off-by: jamesgao-jpg <james.gao@zilliz.com>
Signed-off-by: jamesgao-jpg <james.gao@zilliz.com>
Signed-off-by: jamesgao-jpg <james.gao@zilliz.com>
Signed-off-by: jamesgao-jpg <james.gao@zilliz.com>
Signed-off-by: jamesgao-jpg <james.gao@zilliz.com>
Signed-off-by: jamesgao-jpg <james.gao@zilliz.com>
Signed-off-by: jamesgao-jpg <james.gao@zilliz.com>
Select query_mode=large_topk for Zilliz Cloud performance runs above the default TopK limit and validate reused collections. Signed-off-by: jamesgao-jpg <james.gao@zilliz.com>
|
@jamesgao-jpg Sure, I'll give it a try as well. By the way, could you let me know when the Zilliz test results will be available? I'd like to take a look and compare the performance differences between Zilliz and open-source Milvus. |
* fix query_mode supoort milvus type * fix query_mode supoort milvus type Signed-off-by: jamesgao-jpg <james.gao@zilliz.com>
13e8d41 to
f4ad773
Compare
|
@jamesgao-jpg Also, I have another question: Is the current 1M GT obtained by performing FLAT brute-force search on the LAION 100M dataset? |
@frankleaf Conceptually yes. In practice we used GPU to parallelize computation and did it streamingly to avoid memory problem, so not exactly flat brute force in CPU as in faiss. I will upload the gt generation script later to the bucket for reproductibility. |
|
@jamesgao-jpg I've tested it, and the 1M recall is working perfectly—great job! Also, I was wondering if there are any plans to provide a 1M test set with filters along with ground truth? |
@frankleaf Yes that will be the plan next. I will pin you once that's released. |
|
@jamesgao-jpg thanks! Also, will we be launching a separate leaderboard for the Top1M scenario on |
@frankleaf Yes that will be the plan, but might not be released as soon, since we need to decide what's the best leaderboard format to introduce this and what backends can provide meaningful results (many backends support very limited topks right out of the box). |
Related to #826.
Summary
Performance768D100Mto select LAION query and ground-truth artifacts for K through 1,000,000.recall_atacross supported depth cutoffs.PerformanceCase, with separate IDs-only and vector runs.query_mode=large_topkbefore vector index creation and log the automatic mode selection.Metrics
qpsremains the highest successful QPS observed among the configured concurrency levels.Compatibility
custom_case.payload_profileinput remains supported.query_mode=large_topk; other backends must already permit the requested K.Verification
make lint,make unittest, 108 focused dataset, runner, CLI, backend-translation, result-compatibility, REST, and frontend tests, CLI--dry-runvalidation, and Level 3 impact-map validation.mainat5da3ad1: 60 focused large-TopK and affected FTS metric/CLI/result tests passed.git diff --check origin/main...HEADpassed after the rebase.Scope Notes