4 papers
PCA Rerandomization
Hengtao Zhang, Guosheng Yin, Donald B. Rubin
Mahalanobis distance between treatment group and control group covariate means is often adopted as a balance criterion when implementing a rerandomization strategy. However, this c…
An Effective and Efficient Initialization Scheme for Training Multi-layer Feedforward Neural Networks
Zebin Yang, Hengtao Zhang, Agus Sudjianto +1
Network initialization is the first and critical step for training neural networks. In this paper, we propose a novel network initialization scheme based on the celebrated Stein's…
Balance-Subsampled Stable Prediction
Kun Kuang, Hengtao Zhang, Fei Wu +2
In machine learning, it is commonly assumed that training and test data share the same population distribution. However, this assumption is often violated in practice because the s…
Adaptive Iterative Hessian Sketch via A-Optimal Subsampling
Aijun Zhang, Hengtao Zhang, Guosheng Yin
Iterative Hessian sketch (IHS) is an effective sketching method for modeling large-scale data. It was originally proposed by Pilanci and Wainwright (2016; JMLR) based on randomized…