GPT
N

NuminaMath-7B-TIR

קוד פתוח

AI-MOapache-2.02024-07-04

  • transformers
  • safetensors
  • llama
  • text-generation
  • alignment-handbook

מודל ייעודי לפתרון בעיות מתמטיקה שנעזר בהרצת קוד פייתון כדי לא לטעות בחישובים. הוא זכה בפרס הראשון של תחרות ה־AI Math Olympiad עם תוצאה של 29 מתוך 50.

  • 6.9Bפרמטרים
  • 4,096טוקנים בהקשר
  • 12.9 GBמשקל הקבצים
  • 460הורדות בחודש

מה זה

הבסיס כאן הוא המודל deepseek-math-7b-base, שעבר שני שלבי אימון ייעודיים. בשלב הראשון אימנו אותו על שרשראות חשיבה מילוליות, ובשלב השני לימדו אותו לפרק שאלות מורכבות, לכתוב קוד פייתון שמבצע את החישוב בפועל, ולהשתמש בפלט של הקוד כדי להמשיך את ההסקה.

התוצאה היא פתרון בעיות ברמה תחרותית. במבחן MATH הוא מגיע ל־68.1 אחוז, ובתחרות AMC 2023 הוא פותר 20 מתוך 40 שאלות ללא דגימות מרובות. אלה מספרים שמשאירים מאחור מודלים כלליים באותו סדר גודל כמו Llama 3 8B או Qwen2 7B, אבל חשוב לזכור שהיתרון הזה נשען ישירות על היכולת שלו להריץ פייתון ולא רק לנחש את המספר הבא.

מתאים ל

  • פתרון בעיות אולימפיאדה

    פותר שאלות מתמטיות קשות בזכות שילוב חישובי ביניים בפייתון שמונעים טעויות אריתמטיות.

  • עוזר מחקר בחישובים

    כותב ומריץ קוד סימבולי ומספרי כדי לאמת פתרונות ואלגוריתמים מתמטיים מורכבים.

  • בדיקת תרגילים במדעים

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

איפה זה נופל

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

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

שאלות
נפוצות

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

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

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

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

איך מריצים

כדי להריץ אותו מקומית בפורמט bfloat16 תצטרכו כרטיס מסך עם לפחות 16 גיגה זיכרון וידאו, למשל RTX 4090 או כרטיס שרת מקביל, כי המשקלים תופסים 12.9 גיגה. הטעינה נעשית דרך ספריית transformers הרגילה עם ארכיטקטורת LlamaForCausalLM.

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

from transformers import pipeline

pipe = pipeline("text-generation", model="AI-MO/NuminaMath-7B-TIR")
print(pipe("שלום"))

הרישיון

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

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

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