581 B
581 B
kNN OpenCL
Modulo futuro per costruire il grafo dei vicini degli embedding SFace tramite OpenCL.
Input previsto
embedding float32[N][128]
Gli embedding saranno già L2-normalizzati.
Di conseguenza:
cosine_similarity(a,b) = dot(a,b)
Output previsto
indices[N][32]
scores[N][32]
Obiettivo iniziale
embedding float32[N][128]
|
v
brute-force dot product
|
v
TOPK=32
|
+-- neighbor index
`-- similarity score
La prima implementazione privilegerà la correttezza e la validazione CPU/GPU.