English
Related papers

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

200 papers

Heterogeneous treatment effects (HTEs) are commonly identified during randomized controlled trials (RCTs). Identifying subgroups of patients with similar treatment effects is of high interest in clinical research to advance precision…

Machine Learning · Computer Science 2022-12-06 Peniel N. Argaw , Elizabeth Healey , Isaac S. Kohane

We develop a Gaussian-process mixture model for heterogeneous treatment effect estimation that leverages the use of transformed outcomes. The approach we will present attempts to improve point estimation and uncertainty quantification…

Methodology · Statistics 2018-12-19 Abbas Zaidi , Sayan Mukherjee

Heterogeneous treatment effects, which vary according to individual covariates, are crucial in fields such as personalized medicine and tailored treatment strategies. In many applications, rather than considering the heterogeneity induced…

Methodology · Statistics 2025-08-26 Peng Wu , Pengtao Zeng , Zhaoqing Tian , Shaojie Wei

Knowledge graph embedding (KGE) models represent each entity and relation of a knowledge graph (KG) with low-dimensional embedding vectors. These methods have recently been applied to KG link prediction and question answering over…

Computation and Language · Computer Science 2022-03-22 Apoorv Saxena , Adrian Kochsiek , Rainer Gemulla

Knowledge graphs (KGs) play a vital role in enhancing search results and recommendation systems. With the rapid increase in the size of the KGs, they are becoming inaccuracy and incomplete. This problem can be solved by the knowledge graph…

Machine Learning · Computer Science 2024-08-06 Wanxu Wei , Yitong Song , Bin Yao

Knowledge graph (KG) embedding has been used to benefit the diagnosis of animal diseases by analyzing electronic medical records (EMRs), such as notes and veterinary records. However, learning representations to capture entities and…

Artificial Intelligence · Computer Science 2023-09-08 Van Thuy Hoang , Sang Thanh Nguyen , Sangmyeong Lee , Jooho Lee , Luong Vuong Nguyen , O-Joun Lee

Kernel matching is a widely used technique for estimating treatment effects, particularly valuable in observational studies where randomized controlled trials are not feasible. While kernel-matching approaches have demonstrated practical…

Methodology · Statistics 2025-12-11 Chong Ding , Zheng Li , Hon Keung Tony Ng , Wei Gao

Knowledge tracing (KT) which aims at predicting learner's knowledge mastery plays an important role in the computer-aided educational system. In recent years, many deep learning models have been applied to tackle the KT task, which have…

Computers and Society · Computer Science 2022-08-30 Hanshuang Tong , Zhen Wang , Yun Zhou , Shiwei Tong , Wenyuan Han , Qi Liu

Temporal Knowledge Graph (TKG) completion models traditionally assume access to the entire graph during training. This overlooks challenges stemming from the evolving nature of TKGs, such as: (i) the model's requirement to generalize and…

Artificial Intelligence · Computer Science 2025-07-28 Mehrnoosh Mirtaheri , Ryan A. Rossi , Sungchul Kim , Kanak Mahadik , Tong Yu , Xiang Chen , Mohammad Rostami

Developing new drugs for target diseases is a time-consuming and expensive task, drug repurposing has become a popular topic in the drug development field. As much health claim data become available, many studies have been conducted on the…

Machine Learning · Computer Science 2023-02-22 Yaobin Ling , Pulakesh Upadhyaya , Luyao Chen , Xiaoqian Jiang , Yejin Kim

Consider the problem of improving the estimation of conditional average treatment effects (CATE) for a target domain of interest by leveraging related information from a source domain with a different feature space. This heterogeneous…

Machine Learning · Computer Science 2022-10-13 Ioana Bica , Mihaela van der Schaar

Knowledge graph embedding (KGE) constitutes a foundational task, directed towards learning representations for entities and relations within knowledge graphs (KGs), with the objective of crafting representations comprehensive enough to…

Artificial Intelligence · Computer Science 2024-09-10 Yang Liu , Chuan Zhou , Peng Zhang , Yanan Cao , Yongchao Liu , Zhao Li , Hongyang Chen

The Average Treatment Effect (ATE) is a global measure of the effectiveness of an experimental treatment intervention. Classical methods of its estimation either ignore relevant covariates or do not fully exploit them. Moreover, past work…

Methodology · Statistics 2013-11-05 Emil Pitkin , Richard Berk , Lawrence Brown , Andreas Buja , Ed George , Kai Zhang , Linda Zhao

Heterogeneous treatment effects (HTE) based on patients' genetic or clinical factors are of significant interest to precision medicine. Simultaneously modeling HTE and corresponding main effects for randomized clinical trials with…

Machine Learning · Statistics 2023-02-06 Heng Chen , Michael L. LeBlanc , James Y. Dai

Developing tools for estimating heterogeneous treatment effects (HTE) and individualized treatment effects has been an area of active research in recent years. While these tools have proven to be useful in many contexts, a concern when…

Methodology · Statistics 2025-03-07 Mahsa Ashouri , Nicholas C. Henderson

In causal inference, a variety of causal effect estimands have been studied, including the sample, uncensored, target, conditional, optimal subpopulation, and optimal weighted average treatment effects. Ad-hoc methods have been developed…

Methodology · Statistics 2019-10-18 Nathan Kallus , Michele Santacatterina

Predicated on the increasing abundance of electronic health records, we investi- gate the problem of inferring individualized treatment effects using observational data. Stemming from the potential outcomes model, we propose a novel multi-…

Machine Learning · Computer Science 2017-05-30 Ahmed M. Alaa , Mihaela van der Schaar

The Generalized Estimating Equations (GEE) approach is a widely used statistical method for analyzing longitudinal data and clustered data in clinic studies. In dentistry, due to multiple outcomes obtained from one patient, the outcomes…

Methodology · Statistics 2025-02-25 Hoi-Jeong Lim , Soo-Hyeon Park

Learning transferable representation of knowledge graphs (KGs) is challenging due to the heterogeneous, multi-relational nature of graph structures. Inspired by Transformer-based pretrained language models' success on learning transferable…

Computation and Language · Computer Science 2023-03-29 Sanxing Chen , Hao Cheng , Xiaodong Liu , Jian Jiao , Yangfeng Ji , Jianfeng Gao

Knowledge graph (KG) embedding encodes the entities and relations from a KG into low-dimensional vector spaces to support various applications such as KG completion, question answering, and recommender systems. In real world, knowledge…

Databases · Computer Science 2022-06-02 Tianxing Wu , Arijit Khan , Melvin Yong , Guilin Qi , Meng Wang