HNSWFlat index with JensenShannon metric brings program stuck

[qwevdb]

Summary

When using HNSWFlat index with JensenShannon metric and processed data > 1.0, the program will get stuck.

Platform

OS: Ubuntu 24.04 LTS

Faiss version: faiss-gpu 1.8.0

Installed from: Anaconda

Faiss compilation options:

Running on:

• [✔] CPU
• GPU

Interface:

• C++
• [✔] Python

Reproduction instructions

When using HNSWFlat index with JensenShannon metric and handling data greater than 1.0, the Python script will get stuck.
However, the Python script runs normally, when handling data between 0 and 1.0 .

import faiss
import numpy as np

d = 2
metric_type = faiss.METRIC_JensenShannon
M = 10
index = faiss.IndexHNSWFlat(d, M, metric_type)

np.random.seed(0)
insert_data = np.random.randint(0, 10, (10, d))  # stuck
# insert_data = np.random.randn(10, d)           # stuck
# insert_data = np.random.random((10, d))        # not stuck

index.train(insert_data)
index.add(insert_data)

query_data = np.random.randint(0, 10, (1, d))
# query_data = np.random.randn(1, d)         
# query_data = np.random.random((1, d))

k = 5

#program stuck
D, I = index.search(query_data, k)
print(D)
print(I)

[qwevdb]

Any results for this bug? @asadoughi @kuarora

[asadoughi]

I am able to reproduce the issue with the code provided. However, I'm not intimately familiar with Jensen-Shannon metric and will not be able to prioritize looking into this bug at this time. Normalizing your vectors should be a sufficient workaround for now?