Observação
O acesso a essa página exige autorização. Você pode tentar entrar ou alterar diretórios.
O acesso a essa página exige autorização. Você pode tentar alterar os diretórios.
Este artigo fornece uma visão geral da Pesquisa de Vetores de IA do Mosaico, incluindo o que ela é e como ela funciona.
O que é o Mosaic AI Vector Search?
O Mosaic AI Vector Search é uma solução de pesquisa de vetor que é integrada à Plataforma de Inteligência de Dados do Databricks e integrada às ferramentas de governança e produtividade. A pesquisa de vetor é um tipo de pesquisa otimizada para recuperar inserções. Incorporações são representações matemáticas do conteúdo semântico de dados, de modo geral dados de texto ou de imagens. As inserções são geradas por um modelo de linguagem grande e são um componente fundamental de muitos aplicativos de IA generativos que dependem da localização de documentos ou imagens semelhantes entre si. Alguns exemplos são os sistemas RAG, sistemas de recomendação e reconhecimentos de imagem e vídeo.
Com o Mosaic AI Vector Search, você cria um índice de busca em vetores a partir de uma tabela Delta. O índice inclui dados incorporados com metadados. Em seguida, você pode consultar o índice usando uma API REST para identificar os vetores mais semelhantes e retornar os documentos associados. Você pode estruturar o índice para sincronizar automaticamente sempre que a tabela Delta subjacente for atualizada.
O Mosaic AI Vector Search suporta o seguinte:
- Pesquisa híbrida de similaridade de palavras-chave.
- Filtragem.
- Listas de controle de acesso (ACLs) para gerenciar endpoints de pesquisa de vetores.
- Sincronize apenas as colunas selecionadas.
- Salve e sincronize as incorporações geradas.
Como funciona o Mosaic AI Vector Search?
Mosaic AI Vector Search usa o algoritmo HNSW (Hierarchical Navigable Small World) para pesquisas aproximadas de vizinhos mais próximos e a métrica de distância L2 para medir a similaridade de vetores de incorporação. Se quiser usar a similaridade por cosseno, você precisará normalizar suas incorporações de ponto de dados antes de usá-las para alimentar a busca em vetores. Quando os pontos de dados são normalizados, a classificação produzida pela distância euclidiana é a mesma que a classificação produzida pela similaridade por cosseno.
O Mosaic AI Vector Search também dá suporte à pesquisa híbrida por similaridade de palavras-chave, que combina a pesquisa de inserção baseada em vetores com técnicas tradicionais de pesquisa baseada em palavras-chave. Essa abordagem faz a correspondência das palavras exatas da consulta, além de usar uma pesquisa de similaridade baseada em vetores para capturar as relações semânticas e o contexto da consulta.
Ao integrar essas duas técnicas, a pesquisa híbrida por similaridade de palavras-chave recupera documentos que contêm não apenas as palavras-chave exatas, mas também aquelas que são conceitualmente semelhantes, fornecendo resultados de pesquisa mais abrangentes e relevantes. Esse método é particularmente útil em aplicativos de RAG em que os dados de origem têm palavras-chave exclusivas, como SKUs ou identificadores, que não são adequados para a simples pesquisa por similaridade.
Para obter detalhes sobre a API, veja a referência do SDK do Python e Consultar um endpoint de pesquisa vetorial.
Cálculo de pesquisa de similaridade
O cálculo de pesquisa de similaridade usa a seguinte fórmula:
em que dist
é a distância euclidiana entre a consulta q
e a entrada de índice x
:
Algoritmo de pesquisa por palavra-chave
As pontuações de relevância são calculadas usando o Okapi BM25. Todas as colunas de texto ou cadeia de caracteres são pesquisadas, incluindo as colunas de inserção do texto de origem e de metadados em formato de texto ou cadeia de caracteres. A função de geração de tokens divide os limites das palavras, remove a pontuação e converte todo o texto em letras minúsculas.
Como as pesquisas por similaridade e por palavra-chave são combinadas
Os resultados das pesquisas por similaridade e por palavra-chave são combinados usando a função RRF (Reciprocal Rank Fusion).
A RRF reclassifica cada documento com base em cada método usando a pontuação:
Na equação acima, a classificação começa em 0, soma as pontuações de cada documento e retorna os documentos com pontuação mais alta.
O rrf_param
controla a importância relativa dos documentos de classificação superior e inferior. Com base na documentação, rrf_param
está definido como 60.
As pontuações são normalizadas para que a pontuação mais alta seja 1 e a pontuação mais baixa seja 0 usando a seguinte equação:
Opções para fornecer inserções de vetores
Para criar um índice de pesquisa de vetor no Databricks, primeiro você deve decidir como fornecer inserções de vetor. O Databricks dá suporte para três opções:
Opção 1: Delta Sync Index com inserções calculadas pelo Databriks Você fornece uma tabela Delta de origem que contém dados em formato de texto. O Databricks calcula as inserções usando um modelo especificado e, opcionalmente, salva as inserções em uma tabela no Catálogo do Unity. À medida que a tabela Delta é atualizada, o índice permanece sincronizado com a tabela Delta.
O diagrama a seguir ilustra o processo:
- Calcular inserções de consulta. A consulta pode incluir filtros de metadados.
- Realize pesquisas por similaridade para identificar os documentos mais relevantes.
- Retorne os documentos mais relevantes e anexe-os à consulta.
Opção 2: Delta Sync Index com inserções autogerenciadas Você fornece uma tabela Delta de origem que contém inserções pré-calculadas. À medida que a tabela Delta é atualizada, o índice permanece sincronizado com a tabela Delta.
O diagrama a seguir ilustra o processo:
- A consulta consiste em incorporações e pode incluir filtros de metadados.
- Realize pesquisas por similaridade para identificar os documentos mais relevantes. Retorne os documentos mais relevantes e anexe-os à consulta.
Opção 3: Direct Vector Access Index Você precisa atualizar o índice manualmente usando a API REST quando a tabela de inserções for alterada.
O diagrama a seguir ilustra o processo:
Opções de ponto de extremidade
A Pesquisa de Vetor de IA do Mosaico fornece as seguintes opções para que você possa selecionar a configuração de ponto de extremidade que atenda às necessidades do aplicativo.
Observação
Os pontos de extremidade com otimização de armazenamento estão em Visualização Pública e estão disponíveis nas seguintes regiões: eastus
, eastus2
, westus
, westus2
, westeurope
.
- Os pontos de extremidade padrão têm uma capacidade de 320 milhões de vetores na dimensão 768.
- Pontos de extremidade otimizados para armazenamento têm uma capacidade maior (mais de um bilhão de vetores na dimensão 768) e proporcionam uma indexação 10 a 20 vezes mais rápida. As consultas em pontos de extremidade otimizados para armazenamento têm uma latência ligeiramente maior de cerca de 250 milissegundos. O preço dessa opção é otimizado para o maior número de vetores. Para obter detalhes de preços, consulte a página de preços da busca vetorial. Para obter informações sobre como gerenciar custos de pesquisa de vetor, consulte Mosaic AI Vector Search: Guia de gerenciamento de custos.
Especifique o tipo de ponto de extremidade ao criar o ponto de extremidade.
Veja também limitações de endpoints otimizados para armazenamento.
Como configurar o Mosaic AI Vector Search
Para usar o Mosaic AI Vector Search, crie o seguinte:
Um ponto de extremidade de busca em vetores. Este endpoint atende ao índice de busca de vetores. Você pode consultar e atualizar o endpoint usando a API REST ou o SDK. Confira Criar um ponto de extremidade de busca em vetores para obter instruções.
Os pontos de extremidade são escalados verticalmente de maneira automática para dar suporte ao tamanho do índice ou ao número de solicitações simultâneas. Os endpoints otimizados para armazenamento são reduzidos de forma automática e gradativa quando um índice é excluído. Os pontos de extremidade padrão não são reduzidos automaticamente.
Um índice de busca em vetores. O índice de busca em vetores é criado a partir de uma tabela Delta e é otimizado para fornecer pesquisas aproximadas de vizinho mais próximo em tempo real. O objetivo da pesquisa é identificar os documentos que sejam semelhantes à consulta. Os índices de busca em vetores aparecem no Catálogo do Unity e são regidos por ele. Confira Criar um índice de busca em vetores para obter instruções.
Além disso, se você optar por ter o Databricks computando as inserções, poderá usar um ponto de extremidade de API do Modelo de Fundação pré-configurado ou criar um ponto de extremidade de serviço de modelo para atender ao modelo de inserção de sua escolha. Consulte APIs de Modelo de Base de pagamento por token ou Criar modelo de base que atende os pontos de extremidade para obter instruções.
Para consultar o ponto de extremidade atenda a modelos, use a API REST ou o SDK do Python. Sua consulta pode definir os filtros com base em qualquer coluna na tabela Delta. Para obter detalhes, consulte Usar filtros em consultas, a Referência da API ou a Referência do SDK do Python.
Requisitos
- Workspace habilitado para Catálogo do Unity.
- Computação sem servidor habilitada. Para obter instruções, consulte Conectar para realizar computação sem servidor.
- Para pontos de extremidade padrão, a tabela de origem deve ter o Feed de Dados de Alterações habilitado. Confira Usar o feed de dados de alterações do Delta Lake no Azure Databricks.
- Para criar um índice de pesquisa de vetor, você deve ter CREATE TABLE privilégios no esquema de catálogo em que o índice será criado.
A permissão para criar e gerenciar endpoints de busca vetorial é configurada usando listas de controle de acesso. Consulte ACLs do endpoint de pesquisa vetorial.
Proteção e autenticação de dados
O Databricks implementa os seguintes controles de segurança para proteger seus dados:
- Cada solicitação do cliente ao Mosaic AI Vector Search é logicamente isolada, autenticada e autorizada.
- Mosaic AI Vector Search criptografa todos os dados em repouso (AES-256) e em trânsito (TLS 1.2+).
O Mosaic AI Vector Search dá suporte a dois modos de autenticação, entidades de serviço e PATs (tokens de acesso pessoal). Para aplicativos de produção, o Databricks recomenda que você use entidades de serviço, que podem ter um desempenho por consulta até 100 msec mais rápido em relação aos tokens de acesso pessoal.
Token da entidade de serviço. Um administrador pode gerar um token de entidade de serviço e passá-lo para o SDK ou a API. Consulte usar entidades de serviço. Para casos de uso de produção, o Databricks recomenda usar um token de entidade de serviço.
# Pass in a service principal vsc = VectorSearchClient(workspace_url="...", service_principal_client_id="...", service_principal_client_secret="..." )
Token de acesso pessoal. Você pode usar um token de acesso pessoal para autenticar com o Mosaic AI Vector Search. Consulte token de autenticação de acesso pessoal. Se você usar o SDK em um ambiente de notebook, o SDK gerará automaticamente um token PAT para autenticação.
# Pass in the PAT token client = VectorSearchClient(workspace_url="...", personal_access_token="...")
As Chaves gerenciadas pelo cliente (CMK) têm suporte em pontos de extremidade criados em 8 de maio de 2024 ou após 8 de maio de 2024.
Monitorar o uso e os custos
A tabela do sistema de uso faturável permite monitorar o uso e os custos associados aos índices de busca em vetores e pontos de extremidade. Veja um exemplo de consulta:
WITH all_vector_search_usage (
SELECT *,
CASE WHEN usage_metadata.endpoint_name IS NULL THEN 'ingest'
WHEN usage_type = "STORAGE_SPACE" THEN 'storage'
ELSE 'serving'
END as workload_type
FROM system.billing.usage
WHERE billing_origin_product = 'VECTOR_SEARCH'
),
daily_dbus AS (
SELECT workspace_id,
cloud,
usage_date,
workload_type,
usage_metadata.endpoint_name as vector_search_endpoint,
CASE WHEN workload_type = 'serving' THEN SUM(usage_quantity)
WHEN workload_type = 'ingest' THEN SUM(usage_quantity)
ELSE null
END as dbus,
CASE WHEN workload_type = 'storage' THEN SUM(usage_quantity)
ELSE null
END as dsus
FROM all_vector_search_usage
GROUP BY all
ORDER BY 1,2,3,4,5 DESC
)
SELECT * FROM daily_dbus
Você também pode consultar o uso por política de orçamento. Consulte a Busca Vetorial de IA do Mosaico: políticas orçamentárias.
Para obter detalhes sobre o conteúdo da tabela de uso de cobrança, confira Referência de tabela do sistema de uso faturável. Há consultas adicionais no notebook de exemplo a seguir.
Notebook de consulta de tabelas do sistema de busca em vetores
Limites de tamanho de dados e recursos
A tabela a seguir resume os limites de tamanho de dados e recursos para pontos de extremidade e índices de busca em vetores:
Recurso | Granularidade | Limite |
---|---|---|
Pontos de extremidade de busca em vetores | Por espaço de trabalho | 100 |
Inserções | Por ponto de extremidade | ~ 320.000.000 na dimensão de inserção 768 ~ 160.000.000 na dimensão de incorporação vetorial 1536 ~ 80.000.000 na dimensão de inserção 3072 (escalona aproximadamente de forma linear) |
Dimensão de inserção | Por índice | 4096 |
Índices | Por ponto de extremidade | 50 |
Colunas | Por índice | 50 |
Colunas | Tipos com suporte: bytes, curto, inteiro, longo, float, duplo, booliano, cadeia de caracteres, carimbo de data/hora, data | |
Campos de metadados | Por índice | 50 |
Nome do índice | Por índice | 128 caracteres |
Os limites a seguir se aplicam à criação e atualização de índices de busca em vetores:
Recurso | Granularidade | Limite |
---|---|---|
Tamanho da linha do Índice de Sincronização Delta | Por índice | 100 KB |
Tamanho da coluna de inserção de origem do Índice de Sincronização Delta | Por índice | 32764 bytes |
Limite de tamanho da solicitação de upsert em massa do Índice de Vetor Direto | Por índice | 10 MB |
Limite de tamanho da solicitação de exclusão em massa do Índice de Vetor Direto | Por índice | 10 MB |
Os limites a seguir se aplicam à API de consulta.
Recurso | Granularidade | Limite |
---|---|---|
Tamanho do texto de consulta | Por consulta | 32764 bytes |
Número máximo de resultados retornados (pesquisa aproximada do vizinho mais próximo) | Por consulta | 10.000 |
Número máximo de resultados retornados (pesquisa de similaridade de palavra-chave híbrida) | Por consulta | 200 |
Limitações
- O nome
_id
da coluna é reservado. Se a tabela de origem tiver uma coluna nomeada_id
, renomeie-a antes de criar um índice de pesquisa de vetor. - Não há suporte para permissões no nível de linha e de coluna. No entanto, você pode implementar suas próprias ACLs de nível de aplicação usando a API de filtro.
Limitações de pontos de extremidade otimizados para armazenamento
As limitações nesta seção se aplicam somente a pontos de extremidade otimizados para armazenamento. Os pontos de extremidade otimizados para armazenamento estão na Visualização Pública.
- Não há suporte para o modo de sincronização contínua.
- Não há suporte para colunas a serem sincronizadas.
- A dimensão de inserção deve ser divisível por 16.
- Não há suporte para atualização incremental. Cada sincronização recria totalmente o índice de pesquisa de vetor.
- Você deve prever uma redução significativa de ponta a ponta no tempo necessário para uma sincronização em comparação com os pontos de extremidade padrão. Conjuntos de dados com 1 bilhão de inserções devem concluir uma sincronização em menos de 8 horas. Conjuntos de dados menores levarão menos tempo para serem sincronizados.
- Não há suporte para workspaces compatíveis com HIPAA, PCI e FedRAMP.
- Não há suporte para CMK (chaves gerenciadas pelo cliente).
- Os pontos de extremidade otimizados para armazenamento dão suporte a até 1 bilhão de incorporações de vetores de 768 dimensões. Se você tiver um caso de uso de escala maior, entre em contato com sua equipe de conta.