返回
Directed Acyclic Graph Learning on Attributed Heterogeneous Network
DOI:10.1109/TKDE.2023.3266453.png)
摘要
En 中文
Learning the directed acyclic graph (DAG) among causal variables is a fundamental pre-task in causal discovery. Available DAG learning solutions canonically focus on homogeneous nodes with multiple variables and assume i.i.d. samples, how to learn DAG on typical attributed heterogeneous network (AHN) composed with different types of inter-dependent nodes and diverse attributes is a practical but more difficult task. In this paper, we propose HetDAG to identify DAG among nodes from heterogeneous network. HetDAG first embeds different types of node attributes and aggregates these embeddings as the node's raw representation. Then it uses contrastive learning with prior network structure to explore latent relationships between nodes and update the representation. Next, HetDAG introduces an attention-based DAG learning module that takes node representations as input to search DAG and orient edges between nodes. To the best of our knowledge, HetDAG is the first study to learn DAG on heterogeneous networks. Extensive experiments on both semi-synthetic and real data show that HetDAG can learn DAG in an efficacy way and outperforms the state-of-the-art approaches. The results on real biological networks confirm that HetDAG can find out the causal relations between lncRNAs and miRNAs.
Keyword:
Causality
contrastive learning
directed acyclic graph learning
gradient-based search
heterogeneous network embedding
期刊
IF:
10.4
论文数:
6.9K
被引数:
3.2W
机构
引用论文
Formylglycinamide Ribonucleotide Amidotransferase from Thermotoga maritima: Structural Insights into Complex Formation
Biochemistry
IF0
Stable learning establishes some common ground between causal inference and machine learning稳定学习在因果推理和机器学习之间建立了一些共同点
starBase v2.0: decoding miRNA-ceRNA, miRNA-ncRNA and protein-RNA interaction networks from large-scale CLIP-Seq datastarBase v2.0: 从大规模clip-seq数据中解码mirna-cerna、mirna-ncrna和蛋白质-RNA相互作用网络
NUCLEIC ACIDS RESEARCH
IF13.1

