Disponible via
Choisissez les modèles à comparer
Publications (1)
Aperçu
Jina Embedding B v1 est un modèle d'intégration de texte spécialisé conçu pour transformer le texte anglais en représentations numériques de grande dimension tout en préservant le sens sémantique. Le modèle répond au besoin critique d'intégrations de texte efficaces et précises dans les environnements de production, particulièrement utile pour les organisations nécessitant un équilibre entre efficacité de calcul et qualité d'intégration. Avec ses 110 millions de paramètres générant des intégrations de 768 dimensions, il constitue une solution pratique pour les équipes mettant en œuvre des systèmes de recherche sémantique, de regroupement de documents ou de recommandation de contenu sans nécessiter de ressources de calcul importantes.
Méthodes
Le modèle utilise une architecture basée sur un encodeur T5 améliorée avec un pooling moyen pour générer des représentations de longueur fixe. Formé sur l'ensemble de données Linnaeus-Clean soigneusement organisé, qui contient 385 millions de paires de phrases de haute qualité filtrées à partir de 1,6 milliard de paires initiales, le modèle a subi un processus de formation en deux phases. La première phase a utilisé l'apprentissage contrastif avec perte InfoNCE sur les paires de textes, tandis que la deuxième phase a incorporé la formation de triplets pour affiner la capacité du modèle à distinguer les contenus similaires des contenus différents. Cette approche de formation innovante, combinée à un filtrage rigoureux des données, y compris la détection de la langue et la vérification de la cohérence, permet au modèle de capturer efficacement des relations sémantiques nuancées.
Performance
Lors d'évaluations en conditions réelles, Jina Embedding B v1 démontre des capacités impressionnantes, notamment dans les tâches de similarité textuelle sémantique. Le modèle atteint des performances de pointe sur STS12 avec un score de 0,751, surpassant les modèles établis comme all-mpnet-base-v2 et all-minilm-l6-v2. Il affiche de solides performances dans divers tests tout en maintenant des temps d'inférence efficaces. Cependant, les utilisateurs doivent noter que le modèle est spécifiquement optimisé pour le contenu en anglais et peut ne pas fonctionner de manière optimale sur des tâches multilingues ou spécifiques au code. Le modèle a depuis été remplacé par jina-embeddings-v2-base-en et jina-embeddings-v3, qui offrent des performances améliorées dans une gamme plus large de cas d'utilisation.
Conseils
Pour un déploiement optimal, le modèle nécessite un GPU compatible CUDA, bien que sa taille modérée permette une inférence efficace sur du matériel standard. Le modèle accepte des séquences d'entrée d'une longueur maximale de 512 jetons et est particulièrement adapté aux environnements de production où une génération d'intégration cohérente et fiable est cruciale. Il fonctionne mieux sur le contenu en anglais et est idéal pour des applications telles que la recherche sémantique, la comparaison de similarité de documents et les systèmes de recommandation de contenu. Les équipes doivent envisager d'utiliser les nouvelles versions v2 ou v3 pour les nouveaux projets, car elles offrent des performances améliorées et une prise en charge linguistique plus large. Le modèle n'est pas recommandé pour les tâches nécessitant une compréhension multilingue ou des connaissances spécialisées dans un domaine en dehors du texte anglais général.