Un modèle de langage apprend la recherche et la génération de réponses
En termes simples
Un modèle de langage peut transformer une requête en une représentation qui aide à trouver les documents pertinents, mais il utilise aussi cette requête pour prédire une réponse. ULME est une façon d’entraîner un même modèle pour ces deux tâches : la méthode ajoute des emplacements appris pour la recherche et permet à la réponse de contribuer à l’entraînement de l’un d’eux.
Dans les tests menés sur trois tailles de Qwen3, ULME a obtenu de meilleurs scores que la référence de représentation vectorielle reposant sur un modèle de base correspondant, pour la recherche générale comme pour celle axée sur le raisonnement.
Ses scores de génération de réponses étaient aussi généralement au moins aussi élevés que ceux d’une version entraînée uniquement pour la génération à partir des mêmes données de génération.
La recherche
Un même modèle de langage peut-il représenter une requête assez bien pour trouver des documents pertinents, tout en s’en servant pour générer une réponse ?
L’état final de la requête intervient dans les deux tâches, alors que la recherche demande une représentation comparable à l’échelle d’un corpus et que la génération de réponses demande des états capables de prédire les mots suivants.
ULME entraîne des lectures distinctes pour la recherche et la réponse dans un même modèle causal, en apprenant à un emplacement côté requête à correspondre à un résumé de la réponse.
Les auteurs ont testé si un même modèle pouvait trouver des documents pertinents et générer des réponses sans sacrifier l’une ou l’autre capacité.
ULME utilise un checkpoint unique pour les deux tâches, avec des emplacements appris qui séparent la représentation de la requête destinée à la recherche de la prédiction de la réponse, et un signal d’apprentissage qui relie la requête à la réponse.
Pour trois tailles de Qwen3, ULME a obtenu de meilleurs scores que LLM2VEC-GEN sur des tests généraux de représentation vectorielle du texte et de recherche axée sur le raisonnement.
Comparé à un ajustement fin consacré uniquement à la génération, issu du même modèle de base et utilisant les mêmes données de génération, il a généralement fait au moins aussi bien sur les tests de génération rapportés. Ces résultats étayent les performances conjointes dans le cadre de ces évaluations.
Ces résultats proviennent d’un seul checkpoint final par configuration ; les principaux tableaux présentent trois tailles de Qwen3.
Les auteurs signalent aussi une dégradation qualitative préliminaire de certains comportements en conversation, de mise en forme et de suivi des consignes, sans en mesurer l’ampleur.
Analyse
Détails
- Auteurs de l’article original
- Yang Liu, Zhiqiang Yuan, Shu-Dong Huang, Jiancheng Lv, Liwei Yang
- Soumis à arXiv
- Licence
- CC BY 4.0