$$\rightleftharpoonup{xx}$$
$$\longleftharp{xx}$$,
$$\longrightharp{xx}$$,
מחקר זה השתמש במאגר נתונים אנונימי זמין לציבור; לכן, לא נדרשה הסכמה מדעת. הגישה המוצעת כוללת שישה רכיבים: עיבוד מוקדם, מודול קונבולוציה, מודול קשב, אינטגרציה של מודולי קונבולוציה ותשומת לב, ושכבות מחוברות לחלוטין (FC). איור 1 ממחיש את כל תהליך העבודה של השיטה שלנו.
רכישת מערכי נתונים
תמונות MRI מוח מסומנות מתקבלות ממאגר נתונים זמין לציבור של Kaggle, הכולל ארבע קטגוריות גידול ורקמה תקינה.
קדם-עיבוד נתונים
התמונות משתנות לרזולוציה אחידה ועוצמה שמנורמלות. הרחבה אופציונלית (סיבוב, הפיכה, זום) מיושמת לשיפור ההכללה. מערך הנתונים מחולק לקבוצות אימון (70%) ומערכות בדיקה (30%).
ארכיטקטורת מודלים
שלושה CNN עמוד שדרה שהוכשרו מראש ב-ImageNet — MobileNetV2, EfficientNetV1 ו-Inception-ResNet-V2 — שימשו לסיווג גידולי מוח לארבע קטגוריות באמצעות תמונות MRI. MobileNetV2 ו-EfficientNetV1 הוגדרו עם ממד קלט של 224 × 224 פיקסלים, בעוד ש-Inception-ResNet-V2 דרש גודל קלט של 299 × 299 פיקסלים. בתהליך למידת ההעברה, השכבות הקונבולוציונליות הראשונות הוקפאו בהתחלה כדי לשמור על תכונות תמונה כלליות, בעוד שהשכבות העליונות כוונו בעדינות כדי להתאים את המודל לסיווג גידולי מוח. כל עמוד שדרה הוצמד עם ראש סיווג קל משקל שכלל Global Average Pooling, שכבות צפופות, dropout לרגולריזציה, ופונקציית הפעלה של Softmax לחיזוי ארבע מחלקות. Dropout יושם כדי לצמצם התאמת יתר ולשפר הכללה. הארכיטקטורה המתוכננת שילבה למעשה חילוץ תכונות יעיל עם דרישות חישוביות נמוכות יותר, והקלה על סיווג מדויק תוך שמירה על התאמה לפריסה בתרחישי מוגבל משאבים והוכחת מושג.
חילוץ תכונות מבוסס קשב
תמונות קלט מעובדות דרך שכבת קונבולוציה ראשונית ואחריה מודול תשומת לב מותאם אישית. איגום מקסימום מקבילי וממוצע לוכדים תכונות בולטות והקשריות, אשר מחוברים ומעודנים באמצעות קונבולוציה כדי להדגיש אזורים רלוונטיים לגידולים.
עמוד שדרה קונבולוציוני ומיזוג
תכונות משוקללות לפי קשב מועברות לעמודי שדרה מאומנים מראש (MobileNetV2, EfficientNetV1, או Inception-ResNet-V2). שכבות קונבולוציה ברמה גבוהה מחוברות באופן לוגי להפחתת גישה לזיכרון ולשפר את היעילות החישובית.
סיווג
תכונות שטוחות עוברות דרך שכבות מחוברות במלואן עם הפעלה וניתוק ReLU6, ויוצרות ייצוג קומפקטי לסיווג סופי של ארבע מחלקות softmax.
הערכה
ביצועי המודל מוערכים על סט הבדיקה באמצעות מטריצות דיוק, דיוק, שחזור, ציון F1 ובלבול.
פיתוח מסגרת
הפתרון המוצע מיושם בפייתון ומשתמש ב-Keras להרצה. טבלה 1 ממחישה את ההיפרפרמטרים. מערך הנתונים מחולק למערך אימון ומערך בדיקות לפי יחס של 70:30. חמישה חלוקות שרירותיות שונות של רכבות/בדיקות משמשות להצגת הביצועים המאוזנים הסופיים לכל מערך נתונים.
קדם-עיבוד
יש לעדכן את תמונות הקלט כי מסגרת MobileNetV2 משמשת כבסיס לפתרון המוצע. כדי להפיק תכונות ברזולוציה גבוהה מתמונות, נעשה שימוש במודל MobileNetV2 המאומן מראש עם ממד הקלט שלו. כל תמונה מוקלטת מותאמת ל-224×224 פיקסלים בטווח [-1, 1].
מודל הקונבולוציה
MobileNetV2 משפרת את זיהוי התמונה באמצעות ארכיטקטורת רשת עצבית קונבולוציונית (CNN) יעילה, שנועדה למזער דרישות חישוביות. בעוד ש-CNN מסורתיים משיגים דיוק גבוה, הם לעיתים דורשים זיכרון וכוח עיבוד משמעותיים. MobileNetV1 הציג את מושג הקונבולוציה הניתנת להפרדה בעומק, שמחלק את הקונבולוציה הסטנדרטית לשתי פעולות מובחנות: קונבולוציה עומקית לסינון קלטים וקונבולוציה נקודתית לאינטגרציה של תכונות. MobileNetV2 בונה על רעיון זה על ידי שילוב בלוקים שאריתיים של צוואר בקבוק המורכבים משכבות התרחבות, קונבולוציה עומקית, שכבות הקרנה וחיבורים שאריתיים. שכבת ההקרנה דוחסת ערוצי תכונה, בעוד שחיבורים שאריים מאפשרים זרימת מידע טובה יותר ברחבי הרשת. כדי לשפר את היציבות ולשפר את ביצועי הלמידה, מופעלים ReLU6 ונירמול אצווה לאחר פעולות קונבולוציונליות.
במחקר זה, MobileNetV2, שהוכשר מראש ב-ImageNet, שימש כמודל עמוד שדרה להפקת תכונות, תוך מתן ייצוגים סמנטיים ברמה גבוהה ויעילים תוך שמירה על עלויות חישוביות נמוכות לסיווג גידולי מוח. לאחר כל שכבת קונבולוציה קיימת שכבת הפעלה ReLU6, שהיא התאמה של פונקציית ההפעלה של ReLU, שבה הגודל ממוקסם בשש, ושכבת נרמול אצווה. שכבות קונבולוציה 1x1 נפוצות, איגוד ממוצע וסיווג מוחלות על 17 מהבלוקים הנותרים של צוואר בקבוק, ויוצרות את הארכיטקטורת המלאה של MobileNetV2. השתמשנו ב-MobileNetV2 כרשת עמוד השדרה שהוכשרה מראש ב-ImageNet. במקרה כזה, ההעתקה הקונבולוציונית מתקבלת מהבלוק השארי האחרון לאחר שעבר דרך שכבת הקונבולוציה של MobileNetV2. בלוקים שאריתיים ברמה נמוכה יותר יוצרים מפות תכונות קטנות יותר. חסימות כאלה אינן תורמות למחקר שלנו כי הן אינן אוספות נתונים ברמה גבוהה לזיהוי תמונה כולל. במהלך בניית המודלים והאימון, השכבות שלפני השכבות הקונבולוציוניות שמייצרות מפת תכונות תלת-ממדיות של 7 × 7 נשמרות קבועות. במונחים מתמטיים, זה מבוטא להלן ב-Eq (1):
F1(I) = Conv(I) (1)
כאן,F1(I) מייצג את התכונה הקונבולוציונית המופקת מתמונת הקלט, I.
מודול המיקוד
מנגנון הקשב של המוח האנושי, שמרמז שאנשים מתמקדים באופן טבעי יותר בקלטים חזותיים משמעותיים מאשר בכל פרט בתמונה, משמש כבסיס לעיצוב רשת הטרנספורמציה הזו. מודלים מבוססי קשב רבים נוצרו במחקר למידה עמוקה בעקבות רעיון זה. הקורלציות המרחביות הנראות בייצוגי גידולי מוח מודגשות על ידי מנגנון הקשב המוצע. בעוד שמודול הקשב בערוץ לוכד את המידע החשוב ביותר הקשור לגידול בין ערוצי התכונה, יחידת הקשב האזורית מזהה את האזורים המידעיים ביותר בתמונת ה-MRI. באמצעות שיטת תשומת לב מובנית, יחידת המיקוד המוצעת מתמקדת באזורים המכילים מידע מבחין על גידול.
לשיפור ייצוג התכונות, משתמשים בטכניקות כמו Top Activation Pooling וחילוץ מאוזן של תכונות לפני שלב גילוי המאפיינים המרחבי. אסטרטגיה זו משפרת את יכולת המודל להדגיש אזורי גידול בעלי חשיבות קלינית תוך מזעור פרטים רקע לא רלוונטיים, ובכך משפרת את דיוק הסיווג ואת יעילות למידת התכונות. השני הוא השרשור של הטנזורים המאגדים המקסימלי והממוצע. לבסוף, פונקציית ההפעלה של סיגמא, בשילוב עם פילטר קונבולוציוני 7x7, תשמש להפעלת גירויים ויזואליים בתוך הוויזואליות. השלבים השונים של מודול המיקוד מסומנים במשוואות (2) ו-(3). כאשר Pool(F) מאוזן ∈ RHxWx1 מייצג את פעולת האיגום הממוצעת ו-MaxPool(F) ∈ RHxWx1 מתאר את פעולת האיגום המקסימלי.
F2(I) = [AvgPool(F); MaxPool(F)] (2)
ניתן לשלב את שני מפות האלמנטים הללו בתלת-ממד ליצירת מפת אלמנטים
F2(I) = ∈ RHxwx1. F3(I) = σ(f7x7(F2(I))) (3)
כאשר σ היא פונקציית ההפעלה שמציינת סיגמא. מאפייני תשומת הלב, המיוצגים על ידי טנזור מקודד גובה HxWxC (V), רוחב (W) ועובי (C), מיוצגים על ידי F3(I), בעוד ש-f7x7 מייצג את פעולת הקונבולוציה של גודל המסנן (7x7).
מיזוג תשומת לב ומודל קונבולוציה
בניגוד לרשת שתלויה במקודד יחיד, יחידת מיזוג רב-מודלית יכולה להפיק מידע יעיל ומקיף בהרבה עבור רוב המודואליות. בלוק ההיתוך מיועד להנחות את התכונות המקסימליות החיוניות של חוצה מודליות; לכן, הוא יאתר אזורי תשומת לב להוצאת אזור הגידול. זהו חיבור פשוט של ייצוגים סמויים בודדים, אך חלק מהמידע יאבד. לכן נוצרה מפה משולבת של אובייקטים על ידי שילוב מפות התכונות של מודולי הקונבולוציה והתצפית באמצעות שיטת הקישור הבסיסית של סיטאולה ואחרים. שתי מפות התכונות הללו תפסו הבדלים בתכונות תמונה שונות, ולכן החלטנו להשתמש בשיטת חיבור פשוטה במקום אסטרטגיות מבוססות מקסימום, מינימום או סכום. דבר זה גם הופך אותו לקל יותר חישובית מהשיטות האלטרנטיביות המשתמשות בשיטות ביליניאריות כצורת חישוב מכפלה טנזורית. כאשר שתי תכונות כאלה משולבות, התוצאה נקראת טנזור תכונות. הביטוי המתמטי של פלט הטנזור המצטבר של תווים T ניתן על ידי משוואה (4).
[F1(I), F3(I)] = T(I) (4)
כאשר F1(I) ∈ RHxWx1 ו-F2(I) ∈ RHxWx1280. אלו טנזורים תלת-ממדיים עם אותו רוחב (W) ואותו גובה (H). יתרה מזאת, ניתן לקשר אותם ביצירת k תלת-ממדית של טנזורים תלת-ממדיים, כאשר T(I) ∈ LxWx1281.
שכבות מחוברות לחלוטין (FC)
השתמשנו בסדרה של שכבות FC לאחר מיזוג תכונות כדי להמיר טנזור וולומטרי שעוצב מחדש לווקטור תכונה ליניארי. ראש הסיווג מורכב משכבת Global Average Pooling, שכבות צפופות מחוברות במלואן, רגולריזציה של Dropout, ושכבת פלט Softmax. פונקציית ההפעלה ReLU6 משמשת, עם 128 יחידות והסתברות ליציאה של 50%.