本論文は、電子健康記録(EHR)データを活用した薬物推奨システムBernGraphを提案している。EHRデータは多くの0値を含む疎なバイナリデータであり、従来の手法ではこの0値が学習に寄与しないため、性能が低下する問題があった。
BernGraphでは、EHRデータの0/1の値をベルヌーイ分布の平均値に変換することで、この問題を解決している。具体的には、各医療イベントの発生確率をベルヌーイ分布のパラメータとして表現し、ノード属性として用いる。また、2つの医療イベントの共起確率を条件付きベルヌーイ確率としてエッジ属性に用いる。これにより、グラフニューラルネットワークを用いて医療イベント間の複雑な相関関係を学習することができる。
実験の結果、BernGraphは既存手法と比較して、MIMIC-IIIデータセットでJaccard係数が4.6%、F1スコアが6.7%、PRAUCが6.3%向上するなど、優れた性能を示した。これは、EHRデータの0/1の値を確率的に表現することで、疎なデータに起因する問題を解決できたことを示している。BernGraphは、EHRデータのみを用いて優れた薬物推奨性能を実現しており、様々な医療予測タスクへの応用が期待される。
На другой язык
из исходного контента
arxiv.org
Дополнительные вопросы