GPT
R

rubert-tiny2

קוד פתוח

cointegratedmit2022-03-02

  • sentence-transformers
  • pytorch
  • safetensors
  • bert
  • pretraining

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

  • 29Mפרמטרים
  • 2,048טוקנים בהקשר
  • 228 MBמשקל הקבצים
  • 323,110הורדות בחודש

מה זה

מדובר בגרסה מעודכנת למודל rubert-tiny, שמבוססת על שלוש שכבות בלבד ומכוונת ישירות לשפה הרוסית. המודל מייצר ייצוגים וקטוריים שמקרבים אותו לביצועים של מודלים כבדים כמו LaBSE, אבל בגודל פיזי של 228 מגה-בייט בלבד. המשמעות היא חיסכון עצום בזמני עיבוד על מכונות פשוטות.

בהשוואה לגרסה הקודמת שלו, אוצר המילים הורחב ל־83,828 טוקנים וחלון ההקשר צמח ל־2,048 טוקנים. השיפור הזה מקל על ניתוח פסקאות ומסמכים ארוכים יותר ברוסית, והאימון על משימות NLI נותן לו הבנה טובה יותר של קשרים והקשרים בין מקטעי טקסט שונים.

מתאים ל

  • חיפוש סמנטי ברוסית

    התאמת שאילתות חיפוש למסמכים ברוסית בזמן תגובה אפסי ובשרתים רזים.

  • סיווג טקסטים קצרים

    שליפת וקטורים להזנה באלגוריתמי סיווג כמו KNN עבור פניות שירות.

  • בדיקת דמיון בין משפטים

    זיהוי כפילויות בתוכן רוסי בזכות אימון ייעודי על משימות NLI.

איפה זה נופל

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

שאלות
נפוצות

האם אפשר להשתמש בו לחיפוש בעברית או באנגלית?

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

האם חובה להחזיק GPU כדי לקבל ביצועים סבירים?

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

איך מריצים

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

יש גרסה מותאמת בשם WpythonW/rubert-tiny2-vllm אם רוצים לעבוד עם מנוע vLLM. להקים שרת עצמאי עבור מודל כזה לוקח דקות ספורות, ולכן חבל לשלם לספקי צד שלישי על API חיצוני.

from sentence_transformers import SentenceTransformer

m = SentenceTransformer("cointegrated/rubert-tiny2")
v = m.encode(["שלום עולם"])

הרישיון

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

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

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