GPT
M

mxbai-rerank-base-v2

קוד פתוח

mixedbread-aiapache-2.02025-03-03

  • transformers
  • safetensors
  • qwen2
  • text-generation
  • sentence-transformers

זהו מודל דירוג מחדש קטן ומהיר שמיועד לשפר תוצאות חיפוש בצינורות שליפה. הוא מאפשר לעבד מסמכים ארוכים מאוד בלי לחתוך טקסט ובלי לדרוש משאבי ענן כבדים.

  • 494Mפרמטרים
  • 32,768טוקנים בהקשר
  • 958 MBמשקל הקבצים
  • 26,389הורדות בחודש

מה זה

המודל מבצע שלב שני בחיפוש טקסטואלי, שבו לוקחים עשרות תוצאות ראשוניות ממנוע חיפוש ומסדרים אותן מחדש לפי התאמה מדויקת לשאילתה. עם כחצי מיליארד פרמטרים, הוא מתברג בדיוק בין מודלים זעירים שלא מבינים הקשר מורכב לבין ענקי שפה איטיים ויקרים מדי להרצה מקומית. הוא בנוי על ארכיטקטורת Qwen2 ורץ עם עשרים וארבע שכבות בפורמט bfloat16, מה שמבטיח הבנה סמנטית טובה למרות הגודל הקומפקטי שלו.

היתרון המרכזי כאן הוא חלון הקשר של 32,768 טוקנים, נתון חריג למודל דירוג בגודל כזה. רוב הכלים המתחרים בקטגוריה נחנקים אחרי חמש מאות או אלף טוקנים, מה שמאלץ מפתחים לפצל מסמכים לפסקאות קטנות ולשבור את ההקשר. כאן אתם יכולים לזרוק פנימה חוזים, מאמרים ומסמכים טכניים שלמים, ולקבל ציון רלוונטיות בלי הנדסה מסובכת מסביב.

מתאים ל

  • דירוג מחדש בחיפוש רב לשוני

    סידור תוצאות מדויק לפי שאילתה עבור מאגרי מידע בשפות כמו ערבית או רוסית שבהן המודל מאומן.

  • סינון קטעים למערכות RAG

    צמצום עשרות מסמכים ארוכים לפסקאות המדויקות ביותר לפני שליחתן למודל שפה גדול ויקר.

  • השוואת מסמכים משפטיים שלמים

    הערכת רלוונטיות של טקסטים ארוכים בלי לקטוע אותם, בזכות חלון הקשר של 32,768 טוקנים.

איפה זה נופל

החיסרון המרכזי לקורא הישראלי הוא היעדר עברית ברשימת השפות הנתמכות, שכוללת רק שפות כמו ערבית, בולגרית, בנגלה ואפריקאנס. אם תנסו לדרג טקסטים בעברית, המודל עשוי לקרוס בביצועים או לפרש מילים בצורה שגויה, ולכן הוא לא מתאים למאגרי מידע מקומיים ללא בדיקה יסודית מראש. בנוסף, חלון של שלושים ושניים אלף טוקנים הוא מלכודת ביצועים. אם תשלחו מסמכים באורך מלא לעיתים קרובות, זמן התגובה יזנק משמעותית ביחס לדירוג של פסקאות קצרות.

אותה משפחה

mxbai-rerank יצא ב־2 גרסאות שיש להן עמוד כאן. אלה האחרות, לפי כמה מורידים אותן.

שאלות
נפוצות

האם המודל מתאים למערכות שפועלות בעברית?

לא מומלץ לבנות עליו לפרויקטים בעברית. רשימת השפות הרשמית שלו כוללת שפות ספציפיות כמו ערבית ובולגרית, ועברית אינה מופיעה בה. כדי לעבוד איתו בארץ תצטרכו לבצע בדיקות השוואתיות יסודיות מול חלופות שתומכות בעברית באופן מוצהר.

האם חובה להשתמש במעבד גרפי חזק כדי להריץ אותו?

ממש לא. המודל שוקל פחות מג'יגה בייט ומונה פחות מחצי מיליארד פרמטרים, כך שכרטיס מסך בסיסי ביותר מריץ אותו ללא קושי. עם זאת, אם תנצלו את מלוא חלון ההקשר ותעבירו טקסטים של עשרות אלפי טוקנים, תרגישו האטה משמעותית בזמני התגובה.

איך מריצים

הקבצים שוקלים 958 מגה בייט בלבד, כך שכל כרטיס מסך בסיסי עם ארבעה עד שמונה גיגה זיכרון מריץ אותו בקלות דרך ספריית transformers. אין צורך בתשתית ייעודית כבדה או בשרתים מרובי מעבדים גרפיים, והוא יכול לפעול אפילו על מעבד רגיל אם זמן התגובה שלכם גמיש יותר.

אם אתם צריכים לדרג עשרות מסמכים בכל שאילתה בעומס גבוה מאוד, הזיכרון לא יהיה המגבלה אלא זמן החישוב, במיוחד אם תנצלו את כל חלון ההקשר. במצב כזה, שימוש בגרסה מרוחקת או בשירות מנוהל חוסך תחזוקה, אבל לרוב התרחישים קל וזול פי כמה להחזיק אותו בתוך השרת שלכם.

from transformers import pipeline

pipe = pipeline("text-ranking", model="mixedbread-ai/mxbai-rerank-base-v2")
print(pipe("שלום"))

הרישיון

המודל מופץ תחת רישיון apache-2.0, שמעניק חופש כמעט מוחלט. אתם רשאים להשתמש בו במוצרים מסחריים, לשנות אותו, לארח אותו עצמאית ולהפיץ אותו ללקוחות, כל עוד אתם משאירים את הצהרת זכויות היוצרים ומציינים את הרישיון המקורי.

  • שימוש מסחרי
  • שינוי הקוד
  • הפצה מחדש
  • שימוש בפטנטים

הרישיון המלא בעמוד המודל ↗