Optimization as Estimation with Gaussian Processes in Bandit Settings. - Zi Wang 004; Bolei Zhou; Stefanie Jegelka (AISTATS, 2016)

Max-value Entropy Search for Efficient Bayesian Optimization. - Zi Wang 004; Stefanie Jegelka (ICML, 2017)

Batched High-dimensional Bayesian Optimization via Structural Kernel Learning. - Zi Wang 004; Chengtao Li; Stefanie Jegelka; Pushmeet Kohli (ICML, 2017)

Focused model-learning and planning for non-Gaussian continuous state-action systems. - Zi Wang 004; Stefanie Jegelka; Leslie Pack Kaelbling; Tomás Lozano-Pérez (ICRA, 2017)

Batched Large-scale Bayesian Optimization in High-dimensional Spaces. - Zi Wang 004; Clement Gehring; Pushmeet Kohli; Stefanie Jegelka (AISTATS, 2018)

Active Model Learning and Diverse Action Sampling for Task and Motion Planning. - Zi Wang 004; Caelan Reed Garrett; Leslie Pack Kaelbling; Tomás Lozano-Pérez (IROS, 2018)

Regret bounds for meta Bayesian optimization with an unknown Gaussian process prior. - Zi Wang 004; Beomjoon Kim; Leslie Pack Kaelbling (NeurIPS, 2018)

Learning sparse relational transition models. - Victoria Xia; Zi Wang 004; Leslie Pack Kaelbling (CoRR, 2018)

Learning sparse relational transition models. - Victoria Xia; Zi Wang 004; Kelsey R. Allen; Tom Silver; Leslie Pack Kaelbling (ICLR (Poster), 2019)

Learning to guide task and motion planning using score-space representation. - Beomjoon Kim; Zi Wang 004; Leslie Pack Kaelbling; Tomás Lozano-Pérez (Int. J. Robotics Res., 2019)

Learning compositional models of robot skills for task and motion planning. - Zi Wang 004; Caelan Reed Garrett; Leslie Pack Kaelbling; Tomás Lozano-Pérez (Int. J. Robotics Res., 2021)