$$\rightleftharpoonup{xx}$$
$$\longleftharp{xx}$$,
$$\longrightharp{xx}$$,
המתודולוגיה המוצעת
מודל הלמידה הפדרטיבית החזקה (AR-FL) המוצע נועד לחזות סיכון לתמותת מטופלים בצורה מאובטחת ושיתופית בין מוסדות בריאות שונים, תוך שמירה על פרטיות הנתונים והפיכת המערכת לעמידה בפני הפרעות עוינות. הפריסה כוללת ארבעה אלמנטים עיקריים: (1) מבנה הלמידה המאוחדת, (2) אימון יריבים מבוסס אופטימיזציה מינימום-מקסימום, (3) מנגנון הקשב המודע לתחום, ו-(4) איגוד פרמטרים לשמירה על פרטיות. המתודולוגיה מבוצעת בזרימה שיטתית המיועדת ללמידה פדרלית עוינת וחזקה להערכת סיכון התמותה. התהליך, המתואר באיור 2, מתחיל באיסוף נתונים ממוסדות בריאות רבים, ואחריו עיבוד מוקדם של נתונים לתקנון תכונות ולטיפול בערכים חסרים. מודל גלובלי נוצר תחילה על ידי שרת מרכזי ואז נשלח לכל מוסד רפואי. הכשרה יריבה מקומית מתקיימת בתוך מוסדות הבריאות, מה שמגביר את עמידות המודל בפני הפרעות ומאפשר התאמתו לנתונים ספציפיים לתחום. לאחר האימון, הפרמטרים של המודלים המקומיים מועברים לשרת בצורה מאובטחת באמצעות אגרגציה מאובטחת של פרמטרים. השרת משלב את תרומות המודל המקומי הללו כדי לעדכן את המודל הגלובלי. תהליך זה חוזר מספר פעמים באופטימיזציה איטרטיבית כדי לאפשר שיפור שיתופי של המודל. כתוצאה מכך, המודל מוערך בקפדנות באמצעות מדדים קליניים שמעריכים את ביצועיו החיזויים, החוסן האברסרי וההכללה בין מוסדות.

איור 2: מודל AR-FL המוצע. איור זה מציג את הארכיטקטורה הכוללת של מודל הלמידה המאוחדת העוינת-חזקה המוצע, הכולל הכשרה יריבית, תשומת לב מודעת לתחום, ואגרגציה מאובטחת לשמירה על פרטיות. אנא לחצו כאן כדי לצפות בגרסה מוגדלת של הדמות הזו.
איסוף נתונים
למחקר זה נעשה שימוש בשני מערכי נתונים נגישים לציבור של רשומות בריאות אלקטרוניות (EHR): MIMIC-III4 ומאגר המחקר השיתופי של eICU6. MIMIC-III מספק מעל 40,000 אשפוזים בטיפול נמרץ מבית חולים מרכזי אחד, בעוד שנתוני ה-eICU כוללים מעל 200,000 אשפוזים בטיפול נמרץ מבתי חולים שונים בארה"ב. הדמוגרפיה של המטופלים, המדידות הקליניות, תוצאות בדיקות המעבדה, האירועים המתועדים והתוצאות שנרשמו בזרמי הנתונים הללו הופכים אותם לאטרקטיביים מאוד למשימת חיזוי סיכון התמותה. מערכי הנתונים של ה-EHR מחולקים לחלקים שונים ליצירת סביבת למידה ריאליסטית ורב-מוסדית; כל חלק מייצג בית חולים או מוסד שונה. השיטה מתעדת הבדלים בין מוסדות במאפייני המטופלים, בפרוטוקולים קליניים, בסגנונות התיעוד ובזמינות הנתונים. כל מוסד מדומה יכול לגשת רק לתת-מערך הנתונים שלו ואינו משתף רשומות גולמיות עם השרת המרכזי או עם לקוחות משתתפים אחרים. גישה זו מבטיחה עמידה מלאה בעקרונות שמירת הפרטיות, תוך סימום מגבלות החיים האמיתיים של שיתוף פעולה בין בתי חולים. תוויות תמותה מופקות משדות התוצאה המצוינים במערכי הנתונים, לדוגמה, מדדי תמותה בבית חולים. תוויות אלו משמשות כמטרות חיזוי למודל הלמידה המאוחדת. לכן, כל מוסד שומר על סט מקומי משלו של רשומות תכונות מטופלים הקשורות לתוצאות התמותה, ומאפשר אימון מודלים מבוזר מבלי לחשוף מידע רגיש של מטופלים.
קדם-עיבוד נתונים
כל מוסד מעבד את נתוני ה-EHR המקומיים שלו באופן עצמאי לפני תחילת ההכשרה במודל. מכיוון שלמידה פדרטיבית אינה מאפשרת שיתוף נתונים ברמת המטופל או אפילו סטטיסטיקות סיכום בין המוסדות המשתתפים, כל שלבי העיבוד המוקדם מתבצעים מקומית בכל מוסד ומבוססים אך ורק על המידע במאגר הנתונים של אותו מוסד. זרימת הקדם-עיבוד מורכבת ממספר שלבים קריטיים. בתחילה, משתנים קליניים נומריים כגון סימנים חיוניים, תוצאות בדיקות מעבדה ומדידות פיזיולוגיות מנורמלים כדי לבטל הבדלים בטווחי ערכים ולייצב את התכנסות המודלים. הנתונים המקומיים של כל מוסד הם המקור היחיד לחישוב הסטטיסטיקות המשמשות לנרמל. לאחר מכן, מאפיינים קטגוריים כמו מגדר, סוג קבלה או קטגוריות אבחנתיות מומרת לייצוגים מספריים באמצעות שיטות כמו קידוד חד-חם או מיפויים ייעודיים למוסדות. לכן, המודל המאוחד יכול לטפל בנתונים קטגוריים ללא צורך במילוני קידוד משותפים או טבלאות ייחוס חוצות מוסדות. שלישית, נתונים חסרים, הנפוצים במאגרי נתונים קליניים, מטופלים באמצעות טכניקות איפוטציה מקומיות. בהתאם לסוג התכונה ולהקשר הקליני, מוסדות עשויים לפנות למינוי מיין, מילוי קדימה בהתבסס על תצפיות קודמות, או אפילו החלפות קליניות מבוססות כללים. מכיוון שכל מוסד מבצע אימפוטציה באופן עצמאי, אין צורך במטא-דאטה חיצוניים או פרמטרים משותפים של אימפוטציה. בהתאם להליכים אלו, כל מוסד יכול ליצור מאגר נתונים מעובד מלא יחד עם תוצאות התמותה הנלוות. מערכי הנתונים המקומיים משמשים אז כקלט לאימון המודל המאוחד. לאורך כל צינור העיבוד המוקדם, לא מועברים נתונים גולמיים, סטטיסטיקות נגזרות או פלטים ביניים מחוץ למוסד, מה שמבטיח שמירה קפדנית על נהלי ניהול נתונים השומרי פרטיות.
אתחול מודל
יהי K {1, 2,..., K} אוסף המוסדות (הלקוחות) המעורבים, כאשר לכל מוסד יש מאגר נתונים מקומי.
{
כאשר
מייצג תכונות EHR קלט ו
- היא תווית התמותה המשויכת (בינארית או קטגורית). מודל
גלובלי מאומן בשיתוף פעולה בין כל הלקוחות, כאשר θ הם פרמטרי המודל המשותפים.
התהליך פועל לפי פרוטוקול הממוצע הסטנדרטי FedReated Averating (FedAvg):
השרת המרכזי מאתחל פרמטרים גלובליים
ומשדר אותם. כל לקוח k∈K מבצע אימון מקומי באמצעות הנתונים שלו ומעדכן את המודל ל-
. השרת משלב את העדכונים על ידי לקיחת הממוצע המשוקלל:

, כאשר n =
הליך אתחול השרת המרכזי ושלבי ההדרכה המקומיים מתקיימים בסבבי תקשורת גלובלית T (T global communication).
אימון יריבים מקומי (אופטימיזציה של מינימום–מקס)
כדי לשפר את החוסן, כל לקוח מקיים הכשרה יריבה באתר המקומי שלו. במקום למזער סיכון אמפירי על נתונים נקיים, הלקוחות מתמודדים עם בעיית מינימום-מקסימום:
(x+δ),y)
L: פונקציית אובדן
δ: הפרעה עוינת המוגבלת על ידי 
S: מרחב איום עוין
(x+δ): חיזוי תחת קלט מופרע
בזמן כל תקופה מקומית, נוצרים דגימות יריבה באמצעות ירידת גרדיאנט מוקרנת (PGD) או שיטת סימן גרדיאנט מהיר (FGSM). לאחר מכן, ההכשרה של המודל נמשכת עם דגימות אלו במטרה להגדיל את עמידותה.
מנגנון קשב מודע לתחום
נתוני רשומות בריאות אלקטרוניות (EHR) ממוסדות שונים יכולים להשתנות מאוד מבחינת הפצת הנתונים, סמנטיקת התכונות והפרקטיקות הקליניות. כדי להתמודד עם בעיה זו, מודל AR-FL משלב מודול קשב ייעודי לתחום במודל שלו.
יהי
וקטור הקלט מהמוסד k. משקלי
הקשב לכל תכונה j∈ {1..., d} מחושבים כך:
=
כאשר הם משקלי קשב שניתן לאמן. הקלט משוקלל מחדש כך:

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

כאשר σ שולט בפשרה בין פרטיות לדיוק.
במהלך עדכונים מבוזרים, גישות אלו מבטיחות שמידע פרטי של מטופלים או מוסדות אינם בסיכון לדליפה.
עדכון מודל גלובלי
לאחר סיום שלב ההכשרה המקומי בכל מוסד, השרת המרכזי אוסף את עדכוני המודל שהלקוח הוגש. עדכונים אלו כוללים רק שינויים בפרמטרי המודל ואינם כוללים נתוני מטופלים גולמיים או ייצוגים של תכונות ביניים. כדי למזג את הידע שנרכש מהמוסדות השונים, השרת מיישם את שיטת הממוצע הפדרלי (FedAvg). בתהליך זה, השרת מחשב ממוצע משוקלל של המודלים המקומיים בהתבסס על כמות הנתונים הזמינה בכל מוסד. מוסדות התורמים מאגרי נתונים גדולים יותר משפיעים באופן יחסי גבוה יותר על המודל הגלובלי המעודכן. מכיוון שהעדכונים שודרו באמצעות טכניקות אגרגציה מאובטחות, השרת אינו יכול לראות או לבודד פרמטרים של מוסד מסוים. במקום זאת, השרת מקבל ייצוג משולב מוצפן או מוגן על ידי פרטיות בלבד. דבר זה מבטיח את סודיות תהליך ההכשרה ואת השמדת כל דפוס או מאפיין רפואי ספציפי למוסד שייתכן שזוהו. לאחר חישוב העדכון המצטבר, השרת מכין מודל גלובלי חדש שמייצג את הידע הכולל שנרכש מכל המוסדות המשתפים פעולה באותו סבב תקשורת. המודל הגלובלי החדש נשלח חזרה לכל מוסד כדי להתחיל בסבב הבא של הכשרה יריבה מקומית.
אופטימיזציה איטרטיבית
למידה מאוחדת היא תהליך רב-שלבי הכולל מספר סבבי תקשורת. כל סבב כולל הכשרה של המודל המקומי, שליחת עדכונים מאובטחת, אגרגציה גלובלית והפצה של המודל המשופר למשתתפים. הסבבים החוזרים עוזרים למודל להתקדם בהדרגה לעבר פתרון יציב וחזק. במהלך סבבים אלו, כל ארגון מביא את הידע שלו על אוכלוסיית המטופלים שלו, ובכך יוצר מודל שיכול לזהות דפוסים קליניים בסביבות שונות. יש לציין כי אימון יריבים ומנגנון תשומת הלב המודעת לתחום מיושמים לעיתים דרך אימון מקומי בכל סבב. לכן, המודל משתפר כל הזמן, וכוחו להתמודד עם התקפות עוינות גדל, בעוד יכולתו להסתגל להתפלגויות תכונות שונות בין מוסדות משתפרת. באמצעות אופטימיזציה איטרטיבית, המודל לא רק משפר את ביצועי ההכללה שלו, אלא גם הופך לסובלני יותר ויותר כלפי חוסר איזון נתונים, שונות מוסדית וסביבות יריבות. ההליך מתבצע עד שמושג מספר קבוע מראש של סבבי תקשורת או עד שהמודל הגלובלי מצביע על התכנסות יציבה.
הערכה
בסיום כל סבבי התקשורת, המודל הגלובלי הסופי נבדק לעומק באמצעות מערכי בדיקה שמוחזקים שמקורם במאגרי הנתונים של המשתתפים. מכיוון שמערכות בדיקה אלו אינן משמשות לאימון מודלים, הן מספקות מדד הוגן ובלתי משוחד לביצועי המודל. יעילות המודל מוערכת באמצעות חישוב מדדים שונים מזוויות שונות. דיוק נקי מעריך את המצב שבו קיימים תנאים רגילים. הוא מודד את נכונות המודל בחיזוי תוצאות התמותה. לעומת זאת, דיוק אדרסרי מעריך את מידת יכולתו של המודל לשמור על ביצועיו גם כאשר נתוני הקלט משתנים באופן יריב. השטח מתחת לעקומת מאפיין הפעולה של המקלט (ROC) (AUC-ROC) מספק תובנות לגבי מיומנות המודל בהבחנה בין מטופלים חיים למתים. ציון F1 משמש כאינדיקטור לקשר דיוק-זיכרון, שהוא משמעותי במיוחד עבור מאגרי נתונים קליניים לא מאוזנים. פער ההכללה חושף את הפער בביצועים בין מוסדות שונים, ובכך מדגים את הרב-גוניות של המודל בהתאמה למקורות נתונים מגוונים. סיכון הפרטיות נמדד לבסוף באמצעות שימוש במתקפות עוינות מדומות, שקובעות האם אגרגציה מאובטחת ופרטיות דיפרנציאלית אופציונלית אכן יעילים בהגנה על סודיות המטופל. כל מדדי ההערכה הללו יחד מספקים תמונה מקיפה של כוח החיזוי של המודל, ההתנגדות להתקפות עוינות, יציבות בין מוסדות שונים והציות לדרישות שמירת הפרטיות. האלגוריתם 1 (קובץ משלים 1) למטה מציג את הלמידה הפדרטיבית החזקה (AR-FL) לחיזוי סיכון לתמותה.