Разработка метода обработки сигнала электрокардиографии для эффективной оценки сердечного ритма пациента с использованием свёрточной нейронной сети
- Авторы: Гордиенко Д.В.1, Кравченко А.О.1
-
Учреждения:
- Московский государственный технический университет имени Н.Э. Баумана (национальный исследовательский университет)
- Выпуск: Том 5, № 1S (2024)
- Страницы: 109-111
- Раздел: МОЛОДЫЕ УЧЕНЫЕ: тезисы конференции НПКЦ
- Статья получена: 16.02.2024
- Статья одобрена: 22.03.2024
- Статья опубликована: 03.07.2024
- URL: https://jdigitaldiagnostics.com/DD/article/view/627084
- DOI: https://doi.org/10.17816/DD627084
- ID: 627084
Цитировать
Полный текст
Аннотация
Обоснование. Оценка ритма сердца пациента является первым этапом при аннотировании электрокардиограммы. В представленной работе был разработан метод, предназначенный для обработки сигнала электрокардиографии и оценки сердечного ритма. Метод основан на применении обученной свёрточной нейронной сети, что позволит в кратчайшие сроки после получения электрокардиограммы предоставить врачу предварительную информацию о возможной фибрилляции предсердий или наличии иных нарушений ритма. Кроме того, такие методы могут быть интегрированы в системы телемедицины, позволяя проводить мониторинг сердечного состояния дистанционно.
Цель — разработка метода обработки сигнала электрокардиографии для классификации сердечного ритма пациента по трём классам: синусовый ритм, фибрилляция предсердий и прочие аритмии.
Материалы и методы. Для тренировки и тестирования модели были выбраны электрокардиограммы пациентов, находящиеся в открытом доступе [1]. Программное обеспечение написано на языке программирования Python с использованием фреймворка TensorFlow. Тренировочная, валидационная и тестовая выборки формировались с соотношением 10:1:1, распределение по классам равномерное. Наборы данных для каждой записи подготавливались в трёх вариантах: совмещение графиков всех 12 отведений электрокардиограммы на одном изображении, получение спектрограмм II и V1 отведений с использованием вейвлета Гаусса, представление записи в виде векторной кардиограммы. За основу архитектуры свёрточной нейронной сети была взята архитектура ResNet18, которая была в дальнейшем модифицирована, и для каждого из представлений входных данных были произведены серии модификаций.
Результаты. Была получена сериализированная модель со следующими показателями точности: accuracy=43% для совмещения 12 электрокардиографических отведений на изображении; accuracy=43% для векторного представления электрокардиограммы; accuracy=69% для вейвлет-преобразования электрокардиограммы. В случае двухклассовой задачи, включающей в себя синусовый ритм и фибрилляцию предсердий, метрика accuracy для вейвлет-преобразования достигает 93% с метриками recall=93%, precision=94% и f1-score=93%.
Заключение. Полученные результаты демонстрируют возможности применения свёрточных нейронных сетей для оценки сердечного ритма пациентов. Дальнейшее развитие проекта предполагает подбор наилучшего алгоритма машинного обучения, апробация этого алгоритма для двухклассовой задачи и расширение решения для других классов нарушений ритма. Кроме того, возможно улучшение результатов классификации для трёхклассовой задачи при использовании лучшей модели и введении дополнительной кластеризации.
Полный текст
Обоснование. Оценка ритма сердца пациента является первым этапом при аннотировании электрокардиограммы. В представленной работе был разработан метод, предназначенный для обработки сигнала электрокардиографии и оценки сердечного ритма. Метод основан на применении обученной свёрточной нейронной сети, что позволит в кратчайшие сроки после получения электрокардиограммы предоставить врачу предварительную информацию о возможной фибрилляции предсердий или наличии иных нарушений ритма. Кроме того, такие методы могут быть интегрированы в системы телемедицины, позволяя проводить мониторинг сердечного состояния дистанционно.
Цель — разработка метода обработки сигнала электрокардиографии для классификации сердечного ритма пациента по трём классам: синусовый ритм, фибрилляция предсердий и прочие аритмии.
Материалы и методы. Для тренировки и тестирования модели были выбраны электрокардиограммы пациентов, находящиеся в открытом доступе [1]. Программное обеспечение написано на языке программирования Python с использованием фреймворка TensorFlow. Тренировочная, валидационная и тестовая выборки формировались с соотношением 10:1:1, распределение по классам равномерное. Наборы данных для каждой записи подготавливались в трёх вариантах: совмещение графиков всех 12 отведений электрокардиограммы на одном изображении, получение спектрограмм II и V1 отведений с использованием вейвлета Гаусса, представление записи в виде векторной кардиограммы. За основу архитектуры свёрточной нейронной сети была взята архитектура ResNet18, которая была в дальнейшем модифицирована, и для каждого из представлений входных данных были произведены серии модификаций.
Результаты. Была получена сериализированная модель со следующими показателями точности: accuracy=43% для совмещения 12 электрокардиографических отведений на изображении; accuracy=43% для векторного представления электрокардиограммы; accuracy=69% для вейвлет-преобразования электрокардиограммы. В случае двухклассовой задачи, включающей в себя синусовый ритм и фибрилляцию предсердий, метрика accuracy для вейвлет-преобразования достигает 93% с метриками recall=93%, precision=94% и f1-score=93%.
Заключение. Полученные результаты демонстрируют возможности применения свёрточных нейронных сетей для оценки сердечного ритма пациентов. Дальнейшее развитие проекта предполагает подбор наилучшего алгоритма машинного обучения, апробация этого алгоритма для двухклассовой задачи и расширение решения для других классов нарушений ритма. Кроме того, возможно улучшение результатов классификации для трёхклассовой задачи при использовании лучшей модели и введении дополнительной кластеризации.
Об авторах
Даниил Валерьевич Гордиенко
Московский государственный технический университет имени Н.Э. Баумана (национальный исследовательский университет)
Автор, ответственный за переписку.
Email: mrvanderk@mail.ru
ORCID iD: 0009-0007-9813-3475
Россия, Москва
Артём Олегович Кравченко
Московский государственный технический университет имени Н.Э. Баумана (национальный исследовательский университет)
Email: ikrav514@gmail.com
ORCID iD: 0009-0007-0507-4294
Россия, Москва
Список литературы
- PTB-XL — Atrial Fibrillation Detection [Internet]. Kaggle [дата обращения: 16.12.2023]. Доступ по ссылке: https://www.kaggle.com/datasets/arjunascagnetto/ptbxl-atrial-fibrillation-detection
- DOI: https://doi.org/10.17816/DD627084
Дополнительные файлы
![](/img/style/loading.gif)