papers

Publications (87)

cs.LG2019

Convergent Policy Optimization for Safe Reinforcement Learning

Ming Yu, Zhuoran Yang, Mladen Kolar +1

stat.ML2026

Provable Accelerated Bayesian Optimization with Knowledge Transfer

Haitao Lin, Boxin Zhao, Mladen Kolar +1

stat.ML2019

High-dimensional Varying Index Coefficient Models via Stein's Identity

Sen Na, Zhuoran Yang, Zhaoran Wang +1

math.ST2023

On the Lasso for Graphical Continuous Lyapunov Models

Philipp Dettling, Mathias Drton, Mladen Kolar

math.OC2026

Muon with Nesterov Momentum: Heavy-Tailed Noise and (Randomized) Inexact Polar Decomposition

Sayantan Choudhury, Xiaoran Cheng, Martin Takáč +2

stat.ME2023

Latent Multimodal Functional Graphical Model Estimation

Katherine Tsai, Boxin Zhao, Sanmi Koyejo +1

stat.ME2019

Constrained High Dimensional Statistical Inference

Ming Yu, Varun Gupta, Mladen Kolar

cs.LG2026

SMART: A Spectral Transfer Approach to Multi-Task Learning

Boxin Zhao, Mladen Kolar, Jinchi Lv

math.ST2017

ROCKET: Robust Confidence Intervals via Kendall's Tau for Transelliptical Graphical Models

Rina Foygel Barber, Mladen Kolar

stat.ML2010

Ultra-high Dimensional Multiple Output Learning With Simultaneous Orthogonal Matching Pursuit: A Sure Screening Approach

Mladen Kolar, Eric P. Xing

stat.ME2021

Post-selection inference on high-dimensional varying-coefficient quantile regression model

Ran Dai, Mladen Kolar

stat.ML2021

A Nonconvex Framework for Structured Dynamic Covariance Recovery

Katherine Tsai, Mladen Kolar, Oluwasanmi Koyejo

stat.ML2020

Tensor Canonical Correlation Analysis with Convergence and Statistical Guarantees

You-Lin Chen, Mladen Kolar, Ruey S. Tsay

cs.LG2023

Addressing Budget Allocation and Revenue Allocation in Data Market Environments Using an Adaptive Sampling Algorithm

Boxin Zhao, Boxiang Lyu, Raul Castro Fernandez +1

stat.ME2020

Simultaneous Inference for Pairwise Graphical Models with Generalized Score Matching

Ming Yu, Varun Gupta, Mladen Kolar

stat.ML2022

FuDGE: A Method to Estimate a Functional Differential Graph in a High-Dimensional Setting

Boxin Zhao, Y. Samuel Wang, Mladen Kolar

cs.LG2024

Pessimism Meets Risk: Risk-Sensitive Offline Reinforcement Learning

Dake Zhang, Boxiang Lyu, Shuang Qiu +2

stat.ML2016

Efficient Distributed Learning with Sparsity

Jialei Wang, Mladen Kolar, Nathan Srebro +1

math.OC2022

Gradient-Variation Bound for Online Convex Optimization with Constraints

Shuang Qiu, Xiaohan Wei, Mladen Kolar

stat.ML2026

Trans-Glasso: A Transfer Learning Approach to Precision Matrix Estimation

Boxin Zhao, Cong Ma, Mladen Kolar

math.OC2022

An Adaptive Stochastic Sequential Quadratic Programming with Differentiable Exact Augmented Lagrangians

Sen Na, Mihai Anitescu, Mladen Kolar

stat.ME2022

Joint Gaussian Graphical Model Estimation: A Survey

Katherine Tsai, Oluwasanmi Koyejo, Mladen Kolar

cs.LG2016

Distributed Multi-Task Learning with Shared Representation

Jialei Wang, Mladen Kolar, Nathan Srebro

cs.LG2023

L-SVRG and L-Katyusha with Adaptive Sampling

Boxin Zhao, Boxiang Lyu, Mladen Kolar

cs.LG2023

One Policy is Enough: Parallel Exploration with a Single Policy is Near-Optimal for Reward-Free Reinforcement Learning

Pedro Cisneros-Velarde, Boxiang Lyu, Sanmi Koyejo +1

stat.ME2021

Robust Inference for High-Dimensional Linear Models via Residual Randomization

Y. Samuel Wang, Si Kai Lee, Panos Toulis +1

stat.ML2018

Kernel Meets Sieve: Post-Regularization Confidence Bands for Sparse Additive Model

Junwei Lu, Mladen Kolar, Han Liu

stat.ME2024

High-Dimensional Markov-switching Ordinary Differential Processes

Katherine Tsai, Mladen Kolar, Sanmi Koyejo

stat.ME2024

Confidence Sets for Causal Orderings

Y. Samuel Wang, Mladen Kolar, Mathias Drton

cs.LG2026

Gradient Clipping Beyond Vector Norms: A Spectral Approach for Matrix-Valued Parameters

Alexander Yukhimchuk, Mladen Kolar, Martin Takáč +1

stat.ML2020

Provably Efficient Neural Estimation of Structural Equation Model: An Adversarial Approach

Luofeng Liao, You-Lin Chen, Zhuoran Yang +3

stat.ML2020

Estimation of a Low-rank Topic-Based Model for Information Cascades

Ming Yu, Varun Gupta, Mladen Kolar

stat.ML2020

Semiparametric Nonlinear Bipartite Graph Representation Learning with Provable Guarantees

Sen Na, Yuwei Luo, Zhuoran Yang +2

stat.ML2018

Provable Gaussian Embedding with One Observation

Ming Yu, Zhuoran Yang, Tuo Zhao +2

stat.ML2013

Graph Estimation From Multi-attribute Data

Mladen Kolar, Han Liu, Eric P. Xing

stat.ML2010

Estimating time-varying networks

Mladen Kolar, Le Song, Amr Ahmed +1

cs.LG2022

Local AdaGrad-Type Algorithm for Stochastic Convex-Concave Optimization

Luofeng Liao, Li Shen, Jia Duan +2

math.ST2020

Estimating Differential Latent Variable Graphical Models with Applications to Brain Connectivity

Sen Na, Mladen Kolar, Oluwasanmi Koyejo

stat.ML2014

Inference for Sparse Conditional Precision Matrices

Jialei Wang, Mladen Kolar

math.OC2024

Fully Stochastic Trust-Region Sequential Quadratic Programming for Equality-Constrained Optimization Problems

Yuchen Fang, Sen Na, Michael W. Mahoney +1

stat.ML2010

Union Support Recovery in Multi-task Learning

Mladen Kolar, John Lafferty, Larry Wasserman

stat.ML2015

A General Framework for Robust Testing and Confidence Regions in High-Dimensional Quantile Regression

Tianqi Zhao, Mladen Kolar, Han Liu

stat.ML2019

Direct Estimation of Differential Functional Graphical Models

Boxin Zhao, Y. Samuel Wang, Mladen Kolar

math.ST2013

Estimating Undirected Graphs Under Weak Assumptions

Larry Wasserman, Mladen Kolar, Alessandro Rinaldo

math.OC2023

Inequality Constrained Stochastic Nonlinear Optimization via Active-Set Sequential Quadratic Programming

Sen Na, Mihai Anitescu, Mladen Kolar

stat.ML2022

Provably Training Overparameterized Neural Network Classifiers with Non-convex Constraints

You-Lin Chen, Zhaoran Wang, Mladen Kolar

stat.ME2021

Two-sample inference for high-dimensional Markov networks

Byol Kim, Song Liu, Mladen Kolar

stat.ML2014

Optimal variable selection in multi-group sparse discriminant analysis

Irina Gaynanova, Mladen Kolar

stat.ML2019

Recovery of simultaneous low rank and two-way sparse coefficient matrices, a nonconvex approach

Ming Yu, Varun Gupta, Mladen Kolar

cs.LG2026

Privacy from Symmetry: Orthogonally Equivariant Transformers for LLM Inference

Alexander Yukhimchuk, Andrey Shulga, Mladen Kolar +1

math.OC2023

A Fast Temporal Decomposition Procedure for Long-horizon Nonlinear Dynamic Programming

Sen Na, Mihai Anitescu, Mladen Kolar

cs.LG2022

Personalized Federated Learning with Multiple Known Clusters

Boxiang Lyu, Filip Hanzely, Mladen Kolar

cs.LG2025

Adaptive Batch Size Schedules for Distributed Training of Language Models with Data and Model Parallelism

Tim Tsz-Kit Lau, Weijian Li, Chenwei Xu +2

math.OC2024

Trust-Region Sequential Quadratic Programming for Stochastic Optimization with Random Models

Yuchen Fang, Sen Na, Michael W. Mahoney +1

cs.LG2019

Partially Linear Additive Gaussian Graphical Models

Sinong Geng, Minhao Yan, Mladen Kolar +1

cs.LG2016

Sketching Meets Random Projection in the Dual: A Provable Recovery Algorithm for Big and High-dimensional Data

Jialei Wang, Jason D. Lee, Mehrdad Mahdavi +2

stat.ML2012

Variance function estimation in high-dimensions

Mladen Kolar, James Sharpnack

cs.LG2025

Adaptive Client Sampling in Federated Learning via Online Learning with Bandit Feedback

Boxin Zhao, Lingxiao Wang, Ziqi Liu +4

math.ST2026

Simultaneous Inference for Covariance and Precision Matrices of Long-Range Dependent Time Series

Percy S. Zhai, Mladen Kolar, Wei Biao Wu

stat.ML2019

Joint Nonparametric Precision Matrix Estimation with Confounding

Sinong Geng, Mladen Kolar, Oluwasanmi Koyejo

stat.ML2024

Communication-Efficient Adaptive Batch Size Strategies for Distributed Local Gradient Methods

Tim Tsz-Kit Lau, Weijian Li, Chenwei Xu +2

cs.LG2024

AdAdaGrad: Adaptive Batch Size Schemes for Adaptive Gradient Methods

Tim Tsz-Kit Lau, Han Liu, Mladen Kolar

stat.ML2018

Distributed Stochastic Multi-Task Learning with Graph Regularization

Weiran Wang, Jialei Wang, Mladen Kolar +1

stat.ML2025

High-Dimensional Differential Parameter Inference in Exponential Family using Time Score Matching

Daniel J. Williams, Leyang Wang, Qizhen Ying +2

cs.LG2022

Pessimism meets VCG: Learning Dynamic Mechanism Design via Offline Reinforcement Learning

Boxiang Lyu, Zhaoran Wang, Mladen Kolar +1

stat.ML2018

Scalable Peaceman-Rachford Splitting Method with Proximal Terms

Sen Na, Mingyuan Ma, Mladen Kolar

stat.ML2017

Uniform Inference for High-dimensional Quantile Regression: Linear Functionals and Regression Rank Scores

Jelena Bradic, Mladen Kolar

stat.ML2013

Recovering Block-structured Activations Using Compressive Measurements

Sivaraman Balakrishnan, Mladen Kolar, Alessandro Rinaldo +1

math.OC2023

Convergence Analysis of Accelerated Stochastic Gradient Descent under the Growth Condition

You-Lin Chen, Sen Na, Mladen Kolar

stat.ML2010

Estimating Networks With Jumps

Mladen Kolar, Eric P. Xing

math.ST2014

Mean and variance estimation in high-dimensional heteroscedastic models with non-convex penalties

James Sharpnack, Mladen Kolar

stat.ML2013

Sparsistent Estimation of Time-Varying Discrete Markov Random Fields

Mladen Kolar, Eric P. Xing

cs.LG2024

Personalized Binomial DAGs Learning with Network Structured Covariates

Boxin Zhao, Weishi Wang, Dingyuan Zhu +5

stat.ML2013

Optimal Feature Selection in High-Dimensional Discriminant Analysis

Mladen Kolar, Han Liu

stat.ML2020

Posterior Ratio Estimation of Latent Variables

Song Liu, Yulong Zhang, Mingxuan Yi +1

math.ST2023

Inconsistency of cross-validation for structure learning in Gaussian graphical models

Zhao Lyu, Wai Ming Tai, Mladen Kolar +1

math.OC2023

Constrained Optimization via Exact Augmented Lagrangian and Randomized Iterative Sketching

Ilgee Hong, Sen Na, Michael W. Mahoney +1

stat.ML2019

Learning Influence-Receptivity Network Structure with Guarantee

Ming Yu, Varun Gupta, Mladen Kolar

cs.LG2023

Personalized Federated Learning: A Unified Framework and Universal Optimization Techniques

Filip Hanzely, Boxin Zhao, Mladen Kolar

cs.LG2021

Natural Actor-Critic Converges Globally for Hierarchical Linear Quadratic Regulator

Yuwei Luo, Zhuoran Yang, Zhaoran Wang +1

math.ST2020

High-dimensional Index Volatility Models via Stein's Identity

Sen Na, Mladen Kolar

stat.ML2024

Instrumental Variable Value Iteration for Causal Offline Reinforcement Learning

Luofeng Liao, Zuyue Fu, Zhuoran Yang +3

stat.ML2018

Post-Regularization Inference for Time-Varying Nonparanormal Graphical Models

Junwei Lu, Mladen Kolar, Han Liu

stat.ML2015

Distributed Multitask Learning

Jialei Wang, Mladen Kolar, Nathan Srebro

stat.ML2020

Statistical Inference for Networks of High-Dimensional Point Processes

Xu Wang, Mladen Kolar, Ali Shojaie

stat.ML2024

High-dimensional Functional Graphical Model Structure Learning via Neighborhood Selection Approach

Boxin Zhao, Percy S. Zhai, Y. Samuel Wang +1

cs.LG2022

Dynamic Regret Minimization for Control of Non-stationary Linear Dynamical Systems

Yuwei Luo, Varun Gupta, Mladen Kolar