5 papers
Dense Associative Memory with Epanechnikov Energy
Benjamin Hoover, Zhaoyang Shi, Krishnakumar Balasubramanian +2
We propose a novel energy function for Dense Associative Memory (DenseAM) networks, the log-sum-ReLU (LSR), inspired by optimal kernel density estimation. Unlike the common log-sum…
Statistical-Computational Trade-offs for Recursive Adaptive Partitioning Estimators
Yan Shuo Tan, Jason M. Klusowski, Krishnakumar Balasubramanian
Models based on recursive adaptive partitioning such as decision trees and their ensembles are popular for high-dimensional regression as they can potentially avoid the curse of di…
Nonsmooth Nonparametric Regression via Fractional Laplacian Eigenmaps
Zhaoyang Shi, Krishnakumar Balasubramanian, Wolfgang Polonik
We develop nonparametric regression methods for the case when the true regression function is not necessarily smooth. More specifically, our approach is using the fractional Laplac…
Multivariate Gaussian Approximation for Random Forest via Region-based Stabilization
Zhaoyang Shi, Chinmoy Bhattacharjee, Krishnakumar Balasubramanian +1
We derive Gaussian approximation bounds for -Potential Nearest Neighbor (-PNN) based random forest predictions based on a set of training points given by a Poisson process un…
Gaussian and Bootstrap Approximation for Matching-based Average Treatment Effect Estimators
Zhaoyang Shi, Chinmoy Bhattacharjee, Krishnakumar Balasubramanian +1
We establish Gaussian approximation bounds for covariate and rank-matching-based Average Treatment Effect (ATE) estimators. By analyzing these estimators through the lens of stabil…