English
Related papers

Related papers: KG-TREAT: Pre-training for Treatment Effect Estima…

200 papers

Designing a new clinical trial entails many decisions, such as defining a cohort and setting the study objectives to name a few, and therefore can benefit from recommendations based on exhaustive mining of past clinical trial records. Here,…

Artificial Intelligence · Computer Science 2023-09-29 Murthy V. Devarakonda , Smita Mohanty , Raja Rao Sunkishala , Nag Mallampalli , Xiong Liu

The effective analysis of high-dimensional Electronic Health Record (EHR) data, with substantial potential for healthcare research, presents notable methodological challenges. Employing predictive modeling guided by a knowledge graph (KG),…

Methodology · Statistics 2025-11-06 Zhiwei Xu , Ziming Gan , Doudou Zhou , Shuting Shen , Junwei Lu , Tianxi Cai

Knowledge graph embedding (KGE) has caught significant interest for its effectiveness in knowledge graph completion (KGC), specifically link prediction (LP), with recent KGE models cracking the LP benchmarks. Despite the rapidly growing…

Machine Learning · Computer Science 2024-05-16 Shurong Wang , Yufei Zhang , Xuliang Huang , Hongwei Wang

Given a dataset of individuals each described by a covariate vector, a treatment, and an observed outcome on the treatment, the goal of the individual treatment effect (ITE) estimation task is to predict outcome changes resulting from a…

Machine Learning · Computer Science 2024-06-07 Lokesh Nagalapatti , Pranava Singhal , Avishek Ghosh , Sunita Sarawagi

Knowledge Graph Embedding (KGE) has proven to be an effective approach to solving the Knowledge Graph Completion (KGC) task. Relational patterns which refer to relations with specific semantics exhibiting graph patterns are an important…

Artificial Intelligence · Computer Science 2023-08-16 Long Jin , Zhen Yao , Mingyang Chen , Huajun Chen , Wen Zhang

Knowledge graphs (KGs) represent world's facts in structured forms. KG completion exploits the existing facts in a KG to discover new ones. Translation-based embedding model (TransE) is a prominent formulation to do KG completion. Despite…

Artificial Intelligence · Computer Science 2019-10-11 Mojtaba Nayyeri , Chengjin Xu , Yadollah Yaghoobzadeh , Hamed Shariat Yazdi , Jens Lehmann

Estimating individual treatment effect (ITE) from observational graph data is crucial for decision-making in the fields such as commerce and medicine. This task is challenging due to interference, where individual outcomes can be influenced…

Machine Learning · Computer Science 2026-05-28 Xiaofeng Lin , Han Bao , Hisashi Kashima

Entity alignment aims to identify equivalent entity pairs between different knowledge graphs (KGs). Recently, the availability of temporal KGs (TKGs) that contain time information created the need for reasoning over time in such TKGs.…

Artificial Intelligence · Computer Science 2022-03-15 Chengjin Xu , Fenglong Su , Jens Lehmann

Data-to-text generation has recently attracted substantial interests due to its wide applications. Existing methods have shown impressive performance on an array of tasks. However, they rely on a significant amount of labeled data for each…

Computation and Language · Computer Science 2020-10-13 Wenhu Chen , Yu Su , Xifeng Yan , William Yang Wang

We propose a novel multi-task neural network approach for estimating distributional treatment effects (DTE) in randomized experiments. While DTE provides more granular insights into the experiment outcomes over conventional methods focusing…

Machine Learning · Computer Science 2025-07-11 Tomu Hirata , Undral Byambadalai , Tatsushi Oka , Shota Yasui , Shingo Uto

Population genetics seeks to quantify DNA variant associations with traits or diseases, as well as interactions among variants and with environmental factors. Computing millions of estimates in large cohorts in which small effect sizes are…

Knowledge graphs (KGs) are becoming essential resources for many downstream applications. However, their incompleteness may limit their potential. Thus, continuous curation is needed to mitigate this problem. One of the strategies to…

Artificial Intelligence · Computer Science 2023-08-29 Bayu Distiawan Trisedya , Flora D Salim , Jeffrey Chan , Damiano Spina , Falk Scholer , Mark Sanderson

Knowledge graph (KG) inference aims to address the natural incompleteness of KGs, including rule learning-based and KG embedding (KGE) models. However, the rule learning-based models suffer from low efficiency and generalization while KGE…

Artificial Intelligence · Computer Science 2022-08-23 Guanglin Niu , Bo Li , Yongfei Zhang , Shiliang Pu

Diabetes is a worldwide health issue affecting millions of people. Machine learning methods have shown promising results in improving diabetes prediction, particularly through the analysis of diverse data types, namely gene expression data.…

Machine Learning · Computer Science 2024-04-24 Rita T. Sousa , Heiko Paulheim

Outcome estimation of treatments for target individuals is an important foundation for decision making based on causal relations. Most existing outcome estimation methods deal with binary or multiple-choice treatments; however, in some…

Machine Learning · Computer Science 2021-09-14 Shonosuke Harada , Hisashi Kashima

Different from traditional knowledge graphs (KGs) where facts are represented as entity-relation-entity triplets, hyper-relational KGs (HKGs) allow triplets to be associated with additional relation-entity pairs (a.k.a qualifiers) to convey…

Machine Learning · Computer Science 2021-04-19 Donghan Yu , Yiming Yang

Causal inference methods for treatment effect estimation usually assume independent units. However, this assumption is often questionable because units may interact, resulting in spillover effects between them. We develop augmented inverse…

Methodology · Statistics 2025-04-08 Corinne Emmenegger , Meta-Lina Spohn , Timon Elmer , Peter Bühlmann

Knowledge graph (KG) embedding methods which map entities and relations to unique embeddings in the KG have shown promising results on many reasoning tasks. However, the same embedding dimension for both dense entities and sparse entities…

Computation and Language · Computer Science 2022-05-06 Linlin Chao , Xiexiong Lin , Taifeng Wang , Wei Chu

In this paper, we present the ``joint pre-training and local re-training'' framework for learning and applying multi-source knowledge graph (KG) embeddings. We are motivated by the fact that different KGs contain complementary information…

Computation and Language · Computer Science 2023-06-06 Zequn Sun , Jiacheng Huang , Jinghao Lin , Xiaozhou Xu , Qijin Chen , Wei Hu

Estimation of individualized treatment effects (ITE) from observational studies is a fundamental problem in causal inference and holds significant importance across domains, including healthcare. However, limited observational datasets pose…

Machine Learning · Computer Science 2024-02-14 Vinod Kumar Chauhan , Jiandong Zhou , Ghadeer Ghosheh , Soheila Molaei , David A. Clifton