Una búsqueda que entiende lo que quiere decir su cliente

Busque en una tienda real como lo haría un cliente, en inglés, polaco, alemán, francés o español. La búsqueda por palabras clave busca sus palabras en los nombres de los productos; un pequeño modelo en su navegador busca lo que usted quiere decir.

La demo necesita JavaScript. El modelo se descarga una sola vez, 49 MB; su navegador lo conserva.

3.529 productos de un mayorista británico de regalos, con nombres en inglés. Pruebe una necesidad («keep my tea warm»), otras palabras («piggy bank») u otro idioma.

96% frente a 81%De 80 búsquedas cotidianas encontraron un producto relevante entre los diez primeros; la búsqueda por palabras clave, 81%.

5 idiomasLas búsquedas en polaco, alemán, francés y español encuentran nombres de productos en inglés: 81–89%, frente a 4–21% con la búsqueda por palabras clave.

39 msPor búsqueda en su navegador en un ordenador de sobremesa, comparando los 3.529 productos. Nada de lo que escribe sale de esta página.

Cómo funciona

DatosNombres y precios de productos de un mayorista británico de regalos en línea (UCI Online Retail II, CC BY 4.0), la tienda de la demo 03. No hay registros de búsquedas: un LLM abierto local escribió hasta siete búsquedas por producto, en cinco idiomas, y el modelo aprendió la tienda con ellas.

Modeloparaphrase-multilingual-MiniLM-L12-v2, ajustado con esas búsquedas: 46,7 MB de pesos más un vocabulario de 2.5 MB. Cada nombre de producto se convierte en números de antemano; su búsqueda se convierte en números en su navegador y se compara con todos a la vez.

Evaluación80 búsquedas cotidianas escritas a mano, cada una en cinco idiomas, y 16.565 pares de búsqueda y producto evaluados con una regla escrita; 3.528 son relevantes. El mismo autor escribió el prompt de las búsquedas de entrenamiento, así que las cifras son un límite superior.

Frente a la búsqueda por palabras claveLa búsqueda por palabras clave funciona bien cuando el cliente usa las palabras del propio catálogo, 97% en inglés, y encuentra poco cuando no lo hace o escribe en otro idioma. Tal como se publicó, el modelo encontró un 77% en los cinco idiomas; entrenado con las búsquedas de la propia tienda, un 88%.

Pregunte sobre Baltic ML

Este cuadro es el modelo de búsqueda de la demo 05, aplicado a los textos de este sitio. Cita el sitio y no escribe nada propio; cuando la mejor coincidencia es débil, dice que no está seguro.

Código, configuraciones y el benchmark completo: github.com/balticml/semantic-search. Pesos: huggingface.co/balticml/multilingual-minilm-shop-search. ¿Quiere una búsqueda así en su tienda? Reservar una llamada de 30 min