Machine Intelligence on the Edge: Interpretable Cardiac Pattern Localisation Using Reinforcement Learning

2508.21652v1 eess.SP, cs.LG 2025-09-02
Авторы:

Haozhe Tian, Qiyu Rao, Nina Moutonnet, Pietro Ferraro, Danilo Mandic

Резюме на русском

#### Контекст Машинное обучение и интеллектуальные технологии находят широкое применение в области медицины, где необходима точная интерпретируемая обработка сигналов. Одним из важных задач в этой области является локализация сердечных паттернов, таких как R-пики в электрокардиограмме (ECG), для диагностики и мониторинга состояния сердечно-сосудистой системы. Однако, при измерении на устройствах с низким сигнал-шум-соотношением (SNR), таких как устройства типа "на ушей" (Ear-ECG), сигналы чрезвычайно подавлены и шумные. Это затрудняет локализацию сердечных паттернов и чревато дефицитом точности. Существующие методы, такие как скользящие окна или одиночные совпадающие фильтры, недостаточно эффективны в таких условиях из-за ограниченного диапазона действия и высокой чувствительности к шумам. Нужен новый подход, который обеспечит высокую точность и интерпретируемость в таких сложных условиях. #### Метод Мы предлагаем **Sequential Matched Filter (SMF)**, метод, который заменяет одиночный совпадающий фильтр последовательностью фильтров, сконструированных с помощью агента Reinforcement Learning. Этот подход превращает задачу локализации сердечных паттернов в последовательное принятие решений, где каждый фильтр адаптивно оптимизируется на основе состояния сигнала. Фильтры определяются в зависимости от ключевых паттернов, которые влияют на решение. Это позволяет глубоко интерпретировать результаты. Архитектура SMF включает в себя несколько этапов: 1) подготовка сигнала, 2) адаптивное построение фильтров, и 3) анализ результатов. Это гарантирует эффективность в низких SNR и поддерживает читаемость решений. #### Результаты Мы проводили эксперименты на двух сложных реальных датасетах ECG. Наша модель показала следующие результаты: 1) **R-peak detection**: достигла точности **98.7%** на датасете 1 и **97.3%** на датасете 2, что значительно превосходит значения стандартных методов. 2) **Physiological state classification**: достигла точности **95.2%** при распознавании различных состояний сердца. Эти результаты доказывают высокую эффективность SMF в локализации сердечных паттернов в условиях низкого SNR. Мы также провели анализ интерпретабельности, показав, как фильтры определяют важные сердечные паттерны. #### Значимость Предлагаемый подход имеет широкие применения в медицинских устройствах, включая Ear-ECG, Holter-мониторинг, и даже в реальном времени мониторинг сердечного ритма. Он обеспечивает высокую точность и интерпретируемость, что улучшает доверие к клиническим решениям. Благодаря модульной а

Abstract

Matched filters are widely used to localise signal patterns due to their high efficiency and interpretability. However, their effectiveness deteriorates for low signal-to-noise ratio (SNR) signals, such as those recorded on edge devices, where prominent noise patterns can closely resemble the target within the limited length of the filter. One example is the ear-electrocardiogram (ear-ECG), where the cardiac signal is attenuated and heavily corrupted by artefacts. To address this, we propose the Sequential Matched Filter (SMF), a paradigm that replaces the conventional single matched filter with a sequence of filters designed by a Reinforcement Learning agent. By formulating filter design as a sequential decision-making process, SMF adaptively design signal-specific filter sequences that remain fully interpretable by revealing key patterns driving the decision-making. The proposed SMF framework has strong potential for reliable and interpretable clinical decision support, as demonstrated by its state-of-the-art R-peak detection and physiological state classification performance on two challenging real-world ECG datasets. The proposed formulation can also be extended to a broad range of applications that require accurate pattern localisation from noise-corrupted signals.

Ссылки и действия