English
Related papers

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

200 papers

Treatment effect estimation, which helps understand the causality between treatment and outcome variable, is a central task in decision-making across various domains. While most studies focus on treatment effect estimation on individual…

Machine Learning · Computer Science 2025-03-11 Shonosuke Harada , Ryosuke Yoneda , Hisashi Kashima

Knowledge Graph Embedding (KGE) techniques are crucial in learning compact representations of entities and relations within a knowledge graph, facilitating efficient reasoning and knowledge discovery. While existing methods typically focus…

Computation and Language · Computer Science 2024-10-29 Pengcheng Jiang , Lang Cao , Cao Xiao , Parminder Bhatia , Jimeng Sun , Jiawei Han

Reasoning in a temporal knowledge graph (TKG) is a critical task for information retrieval and semantic search. It is particularly challenging when the TKG is updated frequently. The model has to adapt to changes in the TKG for efficient…

Artificial Intelligence · Computer Science 2021-05-11 Jiapeng Wu , Yishi Xu , Yingxue Zhang , Chen Ma , Mark Coates , Jackie Chi Kit Cheung

Precise estimation of treatment effects is crucial for evaluating intervention effectiveness. While deep learning models have exhibited promising performance in learning counterfactual representations for treatment effect estimation (TEE),…

Machine Learning · Computer Science 2024-01-24 Seungyeon Lee , Ruoqi Liu , Wenyu Song , Lang Li , Ping Zhang

Knowledge Graphs (KG) and associated Knowledge Graph Embedding (KGE) models have recently begun to be explored in the context of drug discovery and have the potential to assist in key challenges such as target identification. In the drug…

Biomolecules · Quantitative Biology 2022-06-01 Stephen Bonner , Ian P Barrett , Cheng Ye , Rowan Swiers , Ola Engkvist , Charles Tapley Hoyt , William L Hamilton

The rapid growth of electronic health record (EHR) datasets opens up promising opportunities to understand human diseases in a systematic way. However, effective extraction of clinical knowledge from the EHR data has been hindered by its…

Machine Learning · Computer Science 2022-06-06 Yuesong Zou , Ahmad Pesaranghader , Aman Verma , David Buckeridge , Yue Li

Knowledge Tracing (KT) aims to model a student's learning trajectory and predict performance on the next question. A key challenge is how to better represent the relationships among students, questions, and knowledge concepts (KCs).…

Artificial Intelligence · Computer Science 2026-01-26 Chi Yu , Hongyu Yuan , Zhiyi Duan

Treatment effect estimation can assist in effective decision-making in e-commerce, medicine, and education. One popular application of this estimation lies in the prediction of the impact of a treatment (e.g., a promotion) on an outcome…

Machine Learning · Computer Science 2023-09-26 Xiaofeng Lin , Guoxi Zhang , Xiaotian Lu , Han Bao , Koh Takeuchi , Hisashi Kashima

Knowledge Graph Embeddings (KGEs) have been intensively explored in recent years due to their promise for a wide range of applications. However, existing studies focus on improving the final model performance without acknowledging the…

Machine Learning · Computer Science 2022-01-25 Xutan Peng , Guanyi Chen , Chenghua Lin , Mark Stevenson

Adaptive experiments improve efficiency by adjusting treatment assignments based on past outcomes, but this adaptivity breaks the i.i.d.\ assumptions that underpin classical asymptotics. At the same time, many questions of interest are…

Machine Learning · Statistics 2026-05-05 Houssam Zenati , Bariscan Bozkurt , Arthur Gretton

Estimating individualised treatment effect (ITE) -- that is the causal effect of a set of variables (also called exposures, treatments, actions, policies, or interventions), referred to as \textit{composite treatments}, on a set of outcome…

Machine Learning · Computer Science 2025-12-19 Vinod Kumar Chauhan , Lei Clifton , Gaurav Nigam , David A. Clifton

Deep learning models have demonstrated promising results in estimating treatment effects (TEE). However, most of them overlook the variations in treatment outcomes among subgroups with distinct characteristics. This limitation hinders their…

Machine Learning · Computer Science 2024-01-31 Seungyeon Lee , Ruoqi Liu , Wenyu Song , Ping Zhang

Adoption of recently developed methods from machine learning has given rise to creation of drug-discovery knowledge graphs (KG) that utilize the interconnected nature of the domain. Graph-based modelling of the data, combined with KG…

Machine Learning · Computer Science 2022-07-27 Stephen Bonner , Ufuk Kirik , Ola Engkvist , Jian Tang , Ian P Barrett

Analyzing data from multiple sources offers valuable opportunities to improve the estimation efficiency of causal estimands. However, this analysis also poses many challenges due to population heterogeneity and data privacy constraints.…

Methodology · Statistics 2025-10-23 Rong Zhao , Jason Falvey , Xu Shi , Vernon M. Chinchilli , Chixiang Chen

Estimating the impact of trauma treatment protocols is complicated by the high dimensional yet finite sample nature of trauma data collected from observational studies. Viscoelastic assays are highly predictive measures of hemostasis.…

Applications · Statistics 2022-09-27 Linqing Wei , Lucy Z. Kornblith , Alan Hubbard

Knowledge Graph Embeddings (KGEs) support a wide range of downstream tasks over Knowledge Graphs (KGs). In practice, KGs evolve as new entities and facts are added, motivating Continual Knowledge Graph Embedding (CKGE) methods that update…

Machine Learning · Computer Science 2026-04-22 Gerard Pons , Carlos Escolano , Besim Bilalli , Anna Queralt

The autonomous driving (AD) industry is exploring the use of knowledge graphs (KGs) to manage the vast amount of heterogeneous data generated from vehicular sensors. The various types of equipped sensors include video, LIDAR and RADAR.…

Artificial Intelligence · Computer Science 2020-03-03 Ruwan Wickramarachchi , Cory Henson , Amit Sheth

Robust estimation of heterogeneous treatment effects is a fundamental challenge for optimal decision-making in domains ranging from personalized medicine to educational policy. In recent years, predictive machine learning has emerged as a…

Machine Learning · Statistics 2025-06-23 Maximilian Schuessler , Erik Sverdrup , Robert Tibshirani

Entity alignment (EA) which links equivalent entities across different knowledge graphs (KGs) plays a crucial role in knowledge fusion. In recent years, graph neural networks (GNNs) have been successfully applied in many embedding-based EA…

Computation and Language · Computer Science 2023-05-01 Feng Xie , Xiang Zeng , Bin Zhou , Yusong Tan

The burden of diseases is rising worldwide, with unequal treatment efficacy for patient populations that are underrepresented in clinical trials. Healthcare, however, is driven by the average population effect of medical treatments and,…

Machine Learning · Computer Science 2024-02-08 Ghadeer O. Ghosheh , Moritz Gögl , Tingting Zhu
‹ Prev 1 2 3 10 Next ›