用张量分解学习时间因果表示,提升医疗数据可解释性
Toward Temporal Causal Representation Learning with Tensor Decomposition
- 将时序因果表示与不规则张量分解联合建模,挖掘潜在结构
- 在MIMIC-III等真实医疗数据上,优于现有方法并增强可解释性
- 适用于医疗表型分析与网络恢复,适合需要因果推理的场景
时间因果表示学习是揭示观测研究中复杂模式的强大工具,常以低维时间序列形式呈现。但在许多实际应用中,数据具有高维度、输入长度不一,天然表现为不规则张量。为分析此类数据,不规则张量分解对提取有意义的聚类至关重要。本文提出一种新的潜变量因果公式,并构建CaRTeD联合学习框架,融合时序因果表示学习与不规则张量分解。该框架为下游任务(如潜结构建模、因果信息提取)提供张量因子蓝图,并支持更灵活的正则化设计。理论上,算法收敛至稳定点,填补了当前不规则张量分解在收敛性理论上的空白。在合成数据和真实世界电子健康记录(EHR)数据集MIMIC-III上的实验表明,所提方法在表型预测与网络恢复任务中均优于先进方法,显著提升因果表示的可解释性。
原文摘要 · Abstract (English)
Temporal causal representation learning is a powerful tool for uncovering complex patterns in observational studies, which are often represented as low-dimensional time series. However, in many real-world applications, data are high-dimensional with varying input lengths and naturally take the form of irregular tensors. To analyze such data, irregular tensor decomposition is critical for extracting meaningful clusters that capture essential information. In this paper, we focus on modeling causal representation learning based on the transformed information. First, we present a novel causal formulation for a set of latent clusters. We then propose CaRTeD, a joint learning framework that integrates temporal causal representation learning with irregular tensor decomposition. Notably, our framework provides a blueprint for downstream tasks using the learned tensor factors, such as modeling latent structures and extracting causal information, and offers a more flexible regularization design to enhance tensor decomposition. Theoretically, we show that our algorithm converges to a stationary point. More importantly, our results fill the gap in theoretical guarantees for the convergence of state-of-the-art irregular tensor decomposition. Experimental results on synthetic and real-world electronic health record (EHR) datasets (MIMIC-III), with extensive benchmarks from both phenotyping and network recovery perspectives, demonstrate that our proposed method outperforms state-of-the-art techniques and enhances the explainability of causal representations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。