← Search

Wai Ming Tai

10 accepted papers

2026

Active Regression for Single-Index Models with Unknown Link Functions

ICML 2026poster

This paper studies active regression for single-index models under general $\ell_p$-loss with an unknown $1$-Lipschitz link function $f$, formulated as $\min_{f,x} \Vert f(Ax)-b\Vert_p^p$ with full access to $A$ but coordinate-query access to $b$. Prior work established upper bounds for known link f…

Cited by 0SourceScholar
2026

Mathesis: Towards Formal Theorem Proving from Natural Languages

ICLR 2026poster

Recent advances in large language models (LLMs) show strong promise for formal reasoning. However, most LLM-based theorem provers remain constrained by the need for expert-written formal statements as inputs, limiting their applicability to real-world problems expressed in natural language. We addre…

Cited by 0SourcecodeScholar
2026

ProofFlow: A Dependency Graph Approach to Faithful Proof Autoformalization

ICLR 2026poster

Proof autoformalization, the task of translating natural language theorems and proofs into machine-verifiable code, is a critical step for integrating large language models into rigorous mathematical workflows. Current approaches focus on producing executable code, but they frequently fail to preser…

Cited by 0SourcecodeScholar
2025

Dimension-Independent Rates for Structured Neural Density Estimation

ICML 2025poster

We show that deep neural networks can achieve dimension-independent rates of convergence for learning structured densities typical of image, audio, video, and text data. For example, in images, where each pixel becomes independent of the rest of the image when conditioned on pixels at most $t$ steps…

Cited by 2SourcePDFScholar
2024

Breaking the curse of dimensionality in structured density estimation

NeurIPS 2024poster

We consider the problem of estimating a structured multivariate density, subject to Markov conditions implied by an undirected graph. In the worst case, without Markovian assumptions, this problem suffers from the curse of dimensionality. Our main result shows how the curse of dimensionality can be…

Cited by 1SourcePDFScholar
2024

Inconsistency of Cross-Validation for Structure Learning in Gaussian Graphical Models

AISTATS 2024poster

Despite numerous years of research into the merits and trade-offs of various model selection criteria, obtaining robust results that elucidate the behavior of cross-validation remains a challenging endeavor. In this paper, we highlight the inherent limitations of cross-validation when employed to di…

Cited by 0SourcePDFScholar
2024

Optimal estimation of Gaussian (poly)trees

AISTATS 2024poster

We develop optimal algorithms for learning undirected Gaussian trees and directed Gaussian polytrees from data. We consider both problems of distribution learning (i.e. in KL distance) and structure learning (i.e. exact recovery). The first approach is based on the Chow-Liu algorithm, and learns an…