27 citations · 59 across the 7 of their papers we have counts for
11 papers
MIRACLE: Causally-Aware Imputation via Learning Missing Data Mechanisms
Trent Kyono, Yao Zhang, Alexis Bellot +1
Missing data is an important problem in machine learning practice. Starting from the premise that imputation methods should preserve the causal structure of the data, we develop a…
Learning outside the Black-Box: The pursuit of interpretable models
Jonathan Crabbé, Yao Zhang, William Zame +1
Machine Learning has proved its ability to produce accurate models but the deployment of these models outside the machine learning community has been hindered by the difficulties o…
CASTLE: Regularization via Auxiliary Causal Graph Discovery
Trent Kyono, Yao Zhang, Mihaela van der Schaar
Regularization improves generalization of supervised models to out-of-sample data. Prior works have shown that prediction in the causal direction (effect from cause) results in low…
Woodpecker-DL: Accelerating Deep Neural Networks via Hardware-Aware Multifaceted Optimizations
Yongchao Liu, Yue Jin, Yong Chen +4
Accelerating deep model training and inference is crucial in practice. Existing deep learning frameworks usually concentrate on optimizing training speed and pay fewer attentions t…
AutoCP: Automated Pipelines for Accurate Prediction Intervals
Yao Zhang, William Zame, Mihaela van der Schaar
Successful application of machine learning models to real-world prediction problems, e.g. financial forecasting and personalized medicine, has proved to be challenging, because suc…
Robust Recursive Partitioning for Heterogeneous Treatment Effects with Uncertainty Quantification
Hyun-Suk Lee, Yao Zhang, William Zame +3
Subgroup analysis of treatment effects plays an important role in applications from medicine to public policy to recommender systems. It allows physicians (for example) to identify…