We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

cs

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Missing Data Imputation Based on Dynamically Adaptable Structural Equation Modeling with Self-Attention

Authors: Ou Deng, Qun Jin
Abstract: Addressing missing data in complex datasets including electronic health records (EHR) is critical for ensuring accurate analysis and decision-making in healthcare. This paper proposes dynamically adaptable structural equation modeling (SEM) using a self-attention method (SESA), an approach to data imputation in EHR. SESA innovates beyond traditional SEM-based methods by incorporating self-attention mechanisms, thereby enhancing model adaptability and accuracy across diverse EHR datasets. Such enhancement allows SESA to dynamically adjust and optimize imputation and overcome the limitations of static SEM frameworks. Our experimental analyses demonstrate the achievement of robust predictive SESA performance for effectively handling missing data in EHR. Moreover, the SESA architecture not only rectifies potential mis-specifications in SEM but also synergizes with causal discovery algorithms to refine its imputation logic based on underlying data structures. Such features highlight its capabilities and broadening applicational potential in EHR data analysis and beyond, marking a reasonable leap forward in the field of data imputation.
Subjects: Machine Learning (cs.LG)
Cite as: arXiv:2308.12388 [cs.LG]
  (or arXiv:2308.12388v4 [cs.LG] for this version)

Submission history

From: Ou Deng [view email]
[v1] Wed, 23 Aug 2023 19:01:17 GMT (3218kb,D)
[v2] Wed, 27 Sep 2023 08:23:02 GMT (4505kb,D)
[v3] Thu, 4 Apr 2024 03:38:32 GMT (256kb,D)
[v4] Thu, 25 Apr 2024 06:34:19 GMT (174kb,D)

Link back to: arXiv, form interface, contact.