arrow
返回

Directed Acyclic Graph Learning on Attributed Heterogeneous Network

delete2023-10-01
delete5
PRE
AI
J
Jiaxuan Liang
J
Jun Wang *
G
Guoxian Yu
W
Wei Guo
C
Carlotta Domeniconi
M
Maozu Guo
DOI:10.1109/TKDE.2023.3266453delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

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

期刊

IEEE Transactions on Knowledge and Data Engineering 封面图
IEEE Transactions on Knowledge and Data Engineering
IF:
10.4
论文数:
6.9K
被引数:
3.2W

机构

G
George Mason University
学者数:
7.7K
论文数: 7.9K
被引数: 1.0W
B
beijing university of civil engineering & architecture
学者数:
3.5K
论文数: 2.7K
被引数: 2
S
shandong university
学者数:
9.5W
论文数: 6.4W
被引数: 94
学者 查看更多机构
引用论文

引用论文

err分享
err收藏
Rotational spectra of indole in the lowest vibrational states
err2019-09-01
err0
PREAI
errK. Vávra; K. Luková; P. Kania; J. Koucký; Š. Urban
err分享
err收藏
The heritability of insomnia: A meta‐analysis of twin studies
err2020-12-03
err0
errOAAI
errNicola L. Barclay; Desi Kocevska; Wichor M. Bramer; Eus J. W. Van Someren; Philip Gehrman
err分享
err收藏
Causal gene identification using combinatorial V-structure search
err2013-07-01
err38
PREAI
errCai, Ruichu; Zhang, Zhenjie; Hao, Zhifeng
err分享
err收藏
学者 查看更多内容