arXiv reaDer
発作間欠期損傷連続体の脳波パターンを識別するための解釈可能な機械学習システム
An Interpretable Machine Learning System to Identify EEG Patterns on the Ictal-Interictal-Injury Continuum
多くの医療サブフィールドでは、臨床業務に使用される機械学習システムの解釈可能性を高めることが求められています。この論文では、ICU EEG モニタリングで一般的に遭遇する 6 種類の脳波パターン (発作、LPD、GPD、LRDA、GRDA、その他) の存在を予測するために、解釈可能な深層学習モデルを設計します。各予測には、専用のユーザー インターフェイスを使用して提供される質の高い説明が付随しています。この斬新なモデル アーキテクチャは、一連の典型的な例 (「プロトタイプ」) を学習し、新しい EEG セグメントをこれらのプロトタイプと比較して決定を下します。これらのプロトタイプは、単一クラス (1 つのクラスのみに関連付けられている) またはデュアル クラス (2 つのクラスに関連付けられている) のいずれかです。モデルを解釈する 3 つの主な方法を提示します。1) グローバル構造保存方法を使用して、1275 次元の cEEG 潜在機能を 2D 空間にマッピングして、発作時 - 発作間欠時 - 損傷の連続体を視覚化し、その高次元構造への洞察を得ます。 . 2) 予測はケースベースの推論を使用して行われ、本質的に「この EEG はその EEG のように見える」という形式の説明を提供します。モデルによって行われた予測の分布と比較します。私たちのモデルは、識別性能メトリクス AUROC (受信者動作特性曲線の下の領域) と AUPRC (精度-再現率曲線の下の領域)、およびタスク固有の p<0.01 で、対応する解釈不可能な (ブラック ボックス) モデルよりも優れたパフォーマンスを発揮します。解釈可能性の指標。 2D 埋め込み空間を探索するユーザー インターフェイスのビデオを提供し、発作間欠期損傷連続脳波パターンの構造の最初のグローバルな概要を提供します。私たちの解釈可能なモデルと専用のユーザー インターフェイスは、cEEG パターンを扱う開業医の参考として機能します。
In many medical subfields, there is a call for greater interpretability in the machine learning systems used for clinical work. In this paper, we design an interpretable deep learning model to predict the presence of 6 types of brainwave patterns (Seizure, LPD, GPD, LRDA, GRDA, other) commonly encountered in ICU EEG monitoring. Each prediction is accompanied by a high-quality explanation delivered with the assistance of a specialized user interface. This novel model architecture learns a set of prototypical examples (``prototypes'') and makes decisions by comparing a new EEG segment to these prototypes. These prototypes are either single-class (affiliated with only one class) or dual-class (affiliated with two classes). We present three main ways of interpreting the model: 1) Using global-structure preserving methods, we map the 1275-dimensional cEEG latent features to a 2D space to visualize the ictal-interictal-injury continuum and gain insight into its high-dimensional structure. 2) Predictions are made using case-based reasoning, inherently providing explanations of the form ``this EEG looks like that EEG.'' 3) We map the model decisions to a 2D space, allowing a user to see how the current sample prediction compares to the distribution of predictions made by the model. Our model performs better than the corresponding uninterpretable (black box) model with p<0.01 for discriminatory performance metrics AUROC (area under the receiver operating characteristic curve) and AUPRC (area under the precision-recall curve), as well as for task-specific interpretability metrics. We provide videos of the user interface exploring the 2D embedded space, providing the first global overview of the structure of ictal-interictal-injury continuum brainwave patterns. Our interpretable model and specialized user interface can act as a reference for practitioners who work with cEEG patterns.
updated: Thu Mar 16 2023 18:32:52 GMT+0000 (UTC)
published: Wed Nov 09 2022 21:33:40 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト