HNSW_SQ
HNSW_SQ combina los gráficos Hierarchical Navigable Small World (HNSW) con la cuantificación escalar (SQ), creando un método avanzado de indexación vectorial que ofrece un equilibrio controlable entre tamaño y precisión. Comparado con el HNSW estándar, este tipo de índice mantiene una alta velocidad de procesamiento de consultas, aunque introduce un ligero incremento en el tiempo de construcción del índice.
Resumen
HNSW_SQ combina dos técnicas de indexación: HNSW para una navegación rápida basada en grafos y SQ para una compresión vectorial eficaz.
HNSW
HNSW construye un grafo multicapa en el que cada nodo corresponde a un vector del conjunto de datos. En este grafo, los nodos se conectan en función de su similitud, lo que permite recorrer rápidamente el espacio de datos. La estructura jerárquica permite al algoritmo de búsqueda reducir el número de vecinos candidatos, lo que acelera considerablemente el proceso de búsqueda en espacios de gran dimensión.
Para más información, consulte HNSW.
SQ
SQ es un método para comprimir vectores representándolos con menos bits. Por ejemplo:
SQ8 utiliza 8 bits, mapeando los valores en 256 niveles. Para más información, consulte IVF_SQ8.
SQ6 utiliza 6 bits para representar cada valor de coma flotante, dando como resultado 64 niveles discretos.
Hnsw Sq
Esta reducción en la precisión disminuye drásticamente la huella de memoria y acelera el cálculo al tiempo que conserva la estructura esencial de los datos.
SQ4UCompatible with Milvus 2.6.8+
Para escenarios que exigen una velocidad de consulta extrema y un uso mínimo de memoria, Milvus introduce SQ4U, una cuantificación escalar uniforme de 4 bits. Se trata de una forma agresiva de cuantificación escalar que comprime el valor de punto flotante de cada dimensión en un entero sin signo de 4 bits.
La "U" de SQ4U significa Uniforme. A diferencia de la cuantización escalar no uniforme, que suele calcular los valores mínimo y máximo de forma independiente para cada dimensión (cuantización por dimensión), SQ4U aplica una estrategia de cuantización uniforme global:
Estadística Global: El sistema calcula un único valor mínimo
vminy un único rango de valoresvdiffque se aplica a todas las dimensiones del vector (o a todo el segmento del vector).Mapeado uniforme: El rango de valores global se divide en 16 intervalos iguales. Cada valor de coma flotante del vector, independientemente de la dimensión a la que pertenezca, se asigna a un entero de 4 bits (0-15) utilizando estos parámetros compartidos.
Ventajas de rendimiento:
Tasa de compresión 8x: Reduce el tamaño 8 veces en comparación con
FP32y 2 veces en comparación conSQ8, lo que disminuye significativamente la presión sobre el ancho de banda de la memoria, que suele ser el cuello de botella en la búsqueda vectorial.Optimización SIMD: La estructura compacta permite a las CPU modernas (AVX2/AVX-512) procesar más dimensiones por ciclo. Además, el uso de parámetros globales elimina la necesidad de cargar valores de escala/desplazamiento variables durante el cálculo de la distancia, lo que mantiene el canal de instrucciones totalmente saturado.
Eficiencia de la caché: Los vectores de menor tamaño hacen que quepan más datos en la caché de la CPU, lo que reduce la latencia causada por el acceso a la memoria.
Debido a la compartición global de parámetros, SQ4U funciona mejor con datos normalizados o conjuntos de datos con distribuciones de valores coherentes en todas las dimensiones.
HNSW + SQ
HNSW_SQ combina los puntos fuertes de HNSW y SQ para permitir una búsqueda aproximada eficiente del vecino más próximo. El proceso funciona de la siguiente manera
Compresión de datos: SQ comprime los vectores utilizando
sq_type(por ejemplo, SQ6 o SQ8), lo que reduce el uso de memoria. Esta compresión puede disminuir la precisión, pero permite al sistema manejar conjuntos de datos más grandes.Construcción de gráficos: Los vectores comprimidos se utilizan para construir un gráfico HNSW. Como los datos están comprimidos, el gráfico resultante es más pequeño y más rápido de buscar.
Búsqueda de candidatos: Cuando se proporciona un vector de consulta, el algoritmo utiliza los datos comprimidos para identificar rápidamente un grupo de vecinos candidatos a partir del grafo HNSW.
(Opcional) Perfeccionamiento de resultados: Los resultados candidatos iniciales pueden ser refinados para una mayor precisión, basándose en los siguientes parámetros:
refine: Controla si se activa este paso de refinamiento. Cuando se establece entrue, el sistema recalcula las distancias utilizando representaciones de mayor precisión o sin comprimir.refine_type: Especifica el nivel de precisión de los datos utilizados durante el refinamiento (por ejemplo, SQ6, SQ8, BF16). Una elección de mayor precisión, comoFP32, puede producir resultados más precisos, pero requiere más memoria. Debe superar la precisión del conjunto de datos comprimido original ensq_type.refine_k: Actúa como factor de ampliación. Por ejemplo, si su k superior es 100 yrefine_kes 2, el sistema vuelve a clasificar los 200 candidatos superiores y devuelve los 100 mejores, mejorando la precisión general.
Para obtener una lista completa de parámetros y valores válidos, consulte Parámetros del índice.
Crear un índice
Para construir un índice HNSW_SQ sobre un campo vectorial en Milvus, utilice el método add_index(), especificando los parámetros index_type, metric_type, y parámetros adicionales para el índice.
from pymilvus import MilvusClient
# Prepare index building params
index_params = MilvusClient.prepare_index_params()
index_params.add_index(
field_name="your_vector_field_name", # Name of the vector field to be indexed
index_type="HNSW_SQ", # Type of the index to create
index_name="vector_index", # Name of the index to create
metric_type="L2", # Metric type used to measure similarity
params={
"M": 64, # Maximum number of neighbors each node can connect to in the graph
"efConstruction": 100, # Number of candidate neighbors considered for connection during index construction
"sq_type": "SQ6", # Scalar quantizer type
"refine": true, # Whether to enable the refinement step
"refine_type": "SQ8" # Precision level of data used for refinement
} # Index building params
)
En esta configuración:
index_type: El tipo de índice a construir. En este ejemplo, establezca el valorHNSW_SQ.metric_type: El método utilizado para calcular la distancia entre vectores. Los valores soportados incluyenCOSINE,L2, yIP. Para más detalles, consulte Tipos de métricas.params: Opciones de configuración adicionales para construir el índice. Para más información, consulte Parámetros de creación de índices.
Una vez configurados los parámetros del índice, puede crear el índice utilizando el método create_index() directamente o pasando los parámetros del índice en el método create_collection. Para más detalles, consulte Crear colección.
Búsqueda en el índice
Una vez creado el índice e insertadas las entidades, puede realizar búsquedas de similitud en el índice.
search_params = {
"params": {
"ef": 10, # Parameter controlling query time/accuracy trade-off
"refine_k": 1 # The magnification factor
}
}
res = MilvusClient.search(
collection_name="your_collection_name", # Collection name
anns_field="vector_field", # Vector field name
data=[[0.1, 0.2, 0.3, 0.4, 0.5]], # Query vector
limit=3, # TopK results to return
search_params=search_params
)
En esta configuración:
params: Opciones de configuración adicionales para la búsqueda en el índice. Para obtener más información, consulte Parámetros de búsqueda específicos del índice.
Parámetros del índice
Esta sección proporciona una visión general de los parámetros utilizados para construir un índice y realizar búsquedas en el índice.
Parámetros de creación de índices
La siguiente tabla enumera los parámetros que pueden configurarse en params al crear un índice.
Parámetro |
Descripción |
Rango de valores |
Sugerencia de ajuste |
|
|---|---|---|---|---|
HNSW |
|
Número máximo de conexiones (o aristas) que puede tener cada nodo en el grafo, incluyendo tanto las aristas salientes como las entrantes. Este parámetro afecta directamente tanto a la construcción como a la búsqueda de índices. |
Tipo: Entero Rango: [2, 2048] Valor por defecto: |
Un valor mayor de Considere la posibilidad de aumentar Considere la posibilidad de reducir En la mayoría de los casos, se recomienda establecer un valor dentro de este intervalo: [5, 100]. |
|
Número de vecinos candidatos considerados para la conexión durante la construcción del índice. Para cada nuevo elemento se evalúa un conjunto mayor de candidatos, pero el número máximo de conexiones realmente establecidas sigue estando limitado por |
Tipo: Entero Rango: [1, int_max] Valor por defecto: |
Un valor más alto de Considere aumentar Considere la posibilidad de reducir En la mayoría de los casos, se recomienda establecer un valor dentro de este intervalo: [50, 500]. |
|
SQ |
|
Especifica el método de cuantificación escalar para comprimir vectores. Cada opción ofrece un equilibrio diferente entre compresión y precisión:
|
Tipo Cadena Rango: [ Valor por defecto: |
La elección de |
|
Un indicador booleano que controla si se aplica un paso de refinamiento durante la búsqueda. El refinamiento consiste en volver a clasificar los resultados iniciales calculando las distancias exactas entre el vector de consulta y los candidatos. |
Tipo: Booleano Rango: [ Valor por defecto: |
Establezca |
|
|
Determina la precisión de los datos utilizados para el refinamiento. Esta precisión debe ser mayor que la de los vectores comprimidos (como se establece en |
Tipo: Cadena Rango:[ Valor por defecto: Ninguno |
Utilice |
Parámetros de búsqueda específicos del índice
La siguiente tabla enumera los parámetros que pueden configurarse en search_params.params cuando se realizan búsquedas en el índice.
Parámetro |
Descripción |
Rango de valores |
Sugerencia de ajuste |
|
|---|---|---|---|---|
HNSW |
|
Controla la amplitud de la búsqueda durante la recuperación del vecino más cercano. Determina cuántos nodos son visitados y evaluados como vecinos más cercanos potenciales. Este parámetro sólo afecta al proceso de búsqueda y se aplica exclusivamente a la capa inferior del gráfico. |
Tipo: Entero Rango: [1, int_max] Valor por defecto: limit (TopK vecinos más cercanos a devolver) |
Un valor mayor de Considere la posibilidad de aumentar Considere la posibilidad de reducir En la mayoría de los casos, se recomienda establecer un valor dentro de este rango: [K, 10K]. |
SQ |
|
El factor de ampliación que controla cuántos candidatos adicionales se examinan durante la etapa de refinamiento, en relación con los K resultados principales solicitados. |
Tipo: Float Rango: [1, float_max) Valor por defecto: 1 |
Los valores más altos de |