$$\rightleftharpoonup{xx}$$
$$\longleftharp{xx}$$,
$$\longrightharp{xx}$$,
Предлагаемая методология
Предлагаемая модель Федеративного обучения (Adversarially Robust Federated Learning, AR-FL) направлена на безопасное и совместное прогнозирование риска смертности пациентов в различных медицинских учреждениях, одновременно защищая конфиденциальность данных и делая систему устойчивой к возражениям. Развертывание включает четыре основных элемента: (1) структуру Федеративного обучения, (2) соперническое обучение на основе оптимизации min-max, (3) механизм вниманительного контроля и (4) агрегирование параметров, сохраняющее конфиденциальность. Методология реализуется в систематическом потоке, предназначенном для состязательного устойчивого федеративного обучения с целью оценки риска смертности. Процесс, показанный на рисунке 2, начинается с сбора данных из многочисленных медицинских учреждений, за которым следует предварительная обработка данных для стандартизации признаков и обработки отсутствующих значений. Глобальная модель сначала создаётся на центральном сервере, а затем отправляется в каждое медицинское учреждение. Локальное состязательное обучение проводится внутри медицинских учреждений, повышая устойчивость модели к возмущениям и позволяя адаптировать её к данным, специфичным для области. После обучения параметры локальных моделей безопасно доставляются на сервер с помощью защищённой агрегации параметров. Сервер включает эти вклады локальной модели для обновления глобальной модели. Этот процесс повторяется несколько раз в итеративной оптимизации для облегчения совместной доработки модели. В результате модель тщательно оценивается с использованием клинических ориентиров, которые оценивают её прогностическую эффективность, устойчивость противника и обобщение между учреждениями.

Рисунок 2: Предлагаемая модель AR-FL. На данном рисунке представлена общая архитектура предлагаемой модели состязательно-устойчивого федеративного обучения, включая состязательное обучение, внимание с учётом области и безопасную агрегацию, сохраняющую конфиденциальность. Пожалуйста, нажмите здесь, чтобы увидеть увеличенную версию этой фигуры.
Сбор данных
Для этого исследования используются два общедоступных набора данных электронных медицинских карт (EHR):MIMIC-III 4 и база данных eICU Collaborative ResearchDatabase 6. MIMIC-III обеспечивает более 40 000 госпитализаций в отделения интенсивной терапии из одной крупной больницы, тогда как данные eICU охватывают более 200 000 пребывания в реанимации в различных больницах США. Соответствующие демографические характеристики пациентов, клинические измерения, результаты лабораторных анализов, зафиксированные события и исходы, зафиксированные в этих потоках данных, делают их очень привлекательными для задачи прогнозирования риска смертности. Наборы данных ЭМК разделяются на отдельные части, чтобы создать реалистичную многоинституциональную федеративную образовательную среду; каждая часть представляет собой разную больницу или учреждение. Метод отражает различия между учреждениями в характеристиках пациентов, клинических протоколах, стилях документации и доступности данных. Каждое смоделированное учреждение может получить доступ только к своему собственному поднабору данных и не обменивается сырыми записями с центральным сервером или другими участвующими клиентами. Такой подход гарантирует полное соблюдение принципов сохранения конфиденциальности, одновременно имитируя реальные ограничения межбольничного сотрудничества. Метки смертности извлекаются из полей исходов, указанных в наборах данных, например, из показателей смертности в больнице. Эти метки служат мишенями прогнозирования для федеративной модели обучения. Таким образом, каждое учреждение ведёт собственный локальный набор записей о характеристиках пациентов, связанных с исходами смертности, что позволяет проводить обучение по децентрализованной модели без раскрытия конфиденциальной информации о пациентах.
Предварительная обработка данных
Каждое учреждение самостоятельно обрабатывает свои локальные данные EHR перед началом обучения модели. Поскольку федеративное обучение запрещает обмен данными на уровне пациентов или даже сводной статистикой между участвующими учреждениями, все этапы предварительной обработки выполняются локально в каждом учреждении и основаны только на информации из набора данных этого учреждения. Процесс предварительной обработки состоит из нескольких ключевых этапов. Сначала нормируются численные клинические переменные, такие как жизненно важные показатели, результаты лабораторных тестов и физиологические измерения, чтобы устранить различия в диапазонах значений и стабилизировать сходимость моделей. Локальные данные каждого учреждения являются единственным источником для вычисления статистики, используемой для нормализации. Далее категориальные атрибуты, такие как пол, тип поступления или диагностические категории, преобразуются в числовые представления с помощью методов, таких как one-hot кодирование или специфические для учреждения отображения. Таким образом, федеративная модель может обрабатывать категориальные данные без использования общих кодировочных словарей или межинституциональных справочных таблиц. В-третьих, отсутствующие данные, которые часто встречаются в клинических наборах данных, обрабатываются с помощью локальных методов импутации. В зависимости от типа признака и клинического контекста учреждения могут прибегать к импутации средних средств, форвардному заполнению на основе предыдущих наблюдений или даже к клиническим заменам на основе правил. Поскольку каждое учреждение выполняет импутацию независимо, внешние метаданные или общие параметры импутации не требуются. Следуя этим процедурам, каждое учреждение может создать полный обработанный набор данных с соответствующими результатами смертности. Локальные наборы данных затем используются как входные данные для обучения федеративной модели. На протяжении всего процесса предварительной обработки за пределы учреждения не передаются исходные данные, производные статистические данные или промежуточные выходы, что гарантирует строгое соблюдение практик управления данными, сохраняющими конфиденциальность.
Инициализация модели
Пусть K {1, 2,..., K} — это собрание вовлечённых институтов (клиентов), где каждое учреждение обладает локальным набором данных.
{
где
представляет входные признаки ЭМК, а
— соответствующую метку смертности (бинарную или категорическую). Глобальная модель
обучается совместно на всех клиентах, где θ — общие параметры модели.
Процесс следует стандартному протоколу FedAvg:
Центральный сервер инициализирует глобальные
параметры и транслирует их. Каждый клиент k∈K выполняет локальное обучение, используя свои данные, и обновляет модель до
. Сервер объединяет обновления, беря взвешенное среднее:

, где n =
Процедура инициализации центрального сервера и этапы локального обучения проходят для раундов глобальной связи T.
Локальное состязательное обучение (оптимизация Min–Max)
Для повышения устойчивости каждый клиент проходит обучение по состязанию на своём местном объекте. Вместо того чтобы минимизировать эмпирический риск чистых данных, клиенты занимаются задачей min-max:
(x+δ),y)
L: Функция потерь
δ: Состязательное возмущение, ограниченное 
S: Пространство враждебной угрозы
(x+δ): Предсказание при возбужденном входе
В момент каждой локальной эпохи состязательные образцы создаются с использованием метода проецируемого градиентного спуска (PGD) или метода быстрого градиентного знака (FGSM). Затем обучение модели продолжается с этими образцами для повышения её устойчивости.
Механизм вниманительного отношения
Данные электронных медицинских карт (EHR) из разных учреждений могут сильно различаться по распределению, семантике признаков и клинической практике. Для решения этой проблемы модель AR-FL включает в свою модель модуль внимания, специфический для конкретной области.
Пусть
— входной вектор из учреждения k. Веса
внимания для каждой признака j∈ {1..., d} вычисляются следующим образом:
=
где — тренируемые веса внимания. Входные данные перевзвешиваются следующим образом:

Благодаря этому механизму модель может подчеркнуть характеристики, важные для клинической практики, в зависимости от характера используемых учреждениями данных, тем самым повышая её адаптивность и интерпретируемость.
Агрегация параметров, сохраняющая конфиденциальность,
AR-FL использует очень безопасную агрегацию вместе с дифференциальными методами конфиденциальности, если это необходимо для поддержания максимального уровня конфиденциальности во время общения. Безопасная агрегация: пользователи шифруют обновления своих моделей таким образом, что сервер может вычислить только сумму, но не отдельные взносы.
Дифференциальная конфиденциальность: добавляет шум в обновления моделей:

Где σ контролирует компромисс между конфиденциальностью и точностью.
Во время распределённых обновлений эти подходы гарантируют, что частные данные пациентов или информация учреждений не подвергаются риску утечки.
Глобальное обновление модели
После завершения локального этапа обучения в каждом учреждении центральный сервер собирает обновления модели, присылаемые клиентом. Эти обновления состоят исключительно из изменений параметров модели и не включают исходные данные пациентов или промежуточные представления признаков. Для объединения знаний, полученных различными учреждениями, сервер применяет метод федеративного усреднения (FedAvg). В этом процессе сервер вычисляет взвешенное среднее по локальным моделям на основе объёма доступных данных в каждом учреждении. Учреждения, предоставляющие более крупные наборы данных, оказывают пропорционально большее влияние на обновлённую глобальную модель. Поскольку обновления передавались с использованием защищённых методов агрегирования, сервер не может просматривать или изолировать параметры отдельного учреждения. Вместо этого сервер получает только зашифрованное или защищенное конфиденциальностью комбинированное представление. Это гарантирует конфиденциальность процесса обучения и уничтожение любых специфических для учреждения закономерностей или характеристик пациента, которые могли быть обнаружены. После вычисления агрегированного обновления сервер готовит новую глобальную модель, которая отражает общее количество знаний, полученных всеми сотрудничающими учреждениями в этом раунде коммуникации. Новая глобальная модель затем возвращается во все учреждения для начала следующего этапа местного сопернического обучения.
Итеративная оптимизация
Федеративное обучение — это многоступенчатый процесс, включающий несколько раундов коммуникации. Каждый раунд включает обучение локальной модели, безопасную отправку обновлений, глобальную агрегацию и последующее распространение улучшенной модели среди участников. Повторяющиеся раунды помогают модели постепенно двигаться к устойчивому и надёжному решению. Во время этих этапов каждая организация применяет свои знания о своей пациентской популяции, создавая модель, способную выявлять клинические закономерности в различных условиях. Следует отметить, что состязательная тренировка и механизм вниманительного внимания иногда реализуются через локальное обучение в каждом раунде. Таким образом, модель постоянно совершенствуется, её способность противостоять враждебным атакам растёт, а способность адаптироваться к различным распределениям функций между учреждениями растёт. Благодаря итеративной оптимизации модель не только улучшает свои результаты обобщения, но и становится всё более терпимой к дисбалансу данных, институциональным вариациям и условиям соперничества. Процедура выполняется до тех пор, пока не достигнет либо заданного количества раундов связи, либо глобальная модель не покажет стабильную сходимость.
Оценка
В конце всех раундов коммуникации итоговая глобальная модель тщательно оценивается с использованием отложенных тестовых наборов, полученных из наборов данных участников. Поскольку эти тестовые наборы не используются для обучения моделей, они обеспечивают справедливую и беспристрастную оценку производительности модели. Эффективность модели оценивается с помощью вычисления различных метрик с нескольких точек зрения. Чистая точность оценивает ситуацию, когда существуют нормальные условия. Он измеряет корректность модели при прогнозировании исходов смертности. В отличие от этого, состязательная точность оценивает степень, в которой модель способна сохранять свою производительность даже при изменении входных данных в противоположном ключе. Область под кривой операционной характеристики приемника (ROC) (AUC-ROC) даёт представление о навыках модели в различии живых и умерших пациентов. Показатель F1 служит индикатором связи точности и воспоминания, что особенно важно для несбалансированных клинических наборов данных. Пробел в обобщении выявляет разницу в эффективности между различными учреждениями, тем самым демонстрируя универсальность модели в адаптации к различным источникам данных. Риск конфиденциальности окончательно измеряется с помощью имитационных враждебных атак, которые определяют, действительно ли безопасная агрегация и опциональная дифференциальная конфиденциальность эффективны для защиты конфиденциальности пациентов. Все эти показатели оценки вместе дают всестороннюю картину предсказательной способности модели, устойчивости к враждебным атакам, стабильности между различными учреждениями и соблюдения требований по сохранению конфиденциальности. Алгоритм 1 (Дополнительный файл 1) ниже демонстрирует Adversarial Robust Federated Learning (AR-FL) для прогнозирования риска смертности.