Portrait of Zebang Shen
Portrait
Zebang Shen
Oberassistent
ETH Zürich
About Me

My current research interests lie at the intersection of stochastic processes and differential geometry. My current directions include (1) understanding and improving diffusion models under the data manifold hypothesis, and (2) understanding the implicit minima-selection phenomenon and improving active minima-selection strategies in nonconvex optimization.

In addition, I am interested in various machine learning topics, including the efficient training of large-scale machine learning models, neural network-based numerical PDE solvers, trustworthy machine learning, and statistical perspectives on machine learning.

Education
  • Zhejiang University
    Supervised by Prof. Hui Qian
    Ph.D.
    2014 - 2019
  • Zhejiang University
    Bachelor's degree
    2010 - 2014
Experience
Selected Publications (view all )
Landing with the score: Riemannian optimization through denoising

Andrey Kharitenko, Zebang Shen, Riccardo De Santi, Niao He, Florian Dorfler

International Conference on Learning Representations, vol. 2026, pp. 129064–129104. 2026

Landing with the score: Riemannian optimization through denoising

Andrey Kharitenko, Zebang Shen, Riccardo De Santi, Niao He, Florian Dorfler

International Conference on Learning Representations, vol. 2026, pp. 129064–129104. 2026

When scores learn geometry: Rate separations under the manifold hypothesis

Xiang Li, Zebang Shen, Ya-Ping Hsieh, Niao He

International Conference on Learning Representations, vol. 2026, pp. 40627–40660. 2026

When scores learn geometry: Rate separations under the manifold hypothesis

Xiang Li, Zebang Shen, Ya-Ping Hsieh, Niao He

International Conference on Learning Representations, vol. 2026, pp. 40627–40660. 2026

Entropy-dissipation Informed Neural Network for McKean-Vlasov Type PDEs

Zebang Shen, Zhenfu Wang

Thirty-seventh Conference on Neural Information Processing Systems 2023

Entropy-dissipation Informed Neural Network for McKean-Vlasov Type PDEs

Zebang Shen, Zhenfu Wang

Thirty-seventh Conference on Neural Information Processing Systems 2023

Self-Consistency of the Fokker Planck Equation

Zebang Shen, Zhenfu Wang, Satyen Kale, Alejandro Ribeiro, Amin Karbasi, Hamed Hassani

Proceedings of Thirty Fifth Conference on Learning Theory, vol. 178, pp. 817–841, PMLR. 2022

Self-Consistency of the Fokker Planck Equation

Zebang Shen, Zhenfu Wang, Satyen Kale, Alejandro Ribeiro, Amin Karbasi, Hamed Hassani

Proceedings of Thirty Fifth Conference on Learning Theory, vol. 178, pp. 817–841, PMLR. 2022

Sinkhorn barycenter via functional gradient descent

Zebang Shen, Zhenfu Wang, Alejandro Ribeiro, Hamed Hassani

Thirty-fifth Conference on Neural Information Processing Systems 2020

Sinkhorn barycenter via functional gradient descent

Zebang Shen, Zhenfu Wang, Alejandro Ribeiro, Hamed Hassani

Thirty-fifth Conference on Neural Information Processing Systems 2020

Stochastic conditional gradient++:(non) convex minimization and continuous submodular maximization

Hamed Hassani, Amin Karbasi, Aryan Mokhtari, Zebang Shen

SIAM Journal on Optimization, vol. 30, no. 4, pp. 3315–3344, Society for Industrial and Applied Mathematics. 2020

Stochastic conditional gradient++:(non) convex minimization and continuous submodular maximization

Hamed Hassani, Amin Karbasi, Aryan Mokhtari, Zebang Shen

SIAM Journal on Optimization, vol. 30, no. 4, pp. 3315–3344, Society for Industrial and Applied Mathematics. 2020

Sinkhorn natural gradient for generative models

Zebang Shen, Zhenfu Wang, Alejandro Ribeiro, Hamed Hassani

(Spotlight) Advances in Neural Information Processing Systems, vol. 33, pp. 1646–1656. 2020

Sinkhorn natural gradient for generative models

Zebang Shen, Zhenfu Wang, Alejandro Ribeiro, Hamed Hassani

(Spotlight) Advances in Neural Information Processing Systems, vol. 33, pp. 1646–1656. 2020

Hessian aided policy gradient

Zebang Shen, Hamed Hassani, Chao Mi, Hui Qian, Alejandro Ribeiro

International conference on machine learning, pp. 5729–5738, PMLR. 2019

Hessian aided policy gradient

Zebang Shen, Hamed Hassani, Chao Mi, Hui Qian, Alejandro Ribeiro

International conference on machine learning, pp. 5729–5738, PMLR. 2019

All publications