Academics

Publications

AuxiLight: A Fast, Lightweight Auxiliary Loss Balancing Algorithm with Application in Object 6D Pose Estimation
T. Zhang, Y. Hu, K. Yu, M. Salzmann
Under review

Understanding the Variance Collapse of SVGD in High Dimensions
J. Ba, M. Erdogdu, M. Ghassemi, S. Sun, T. Suzuki, D. Wu, T. Zhang (Alphabetical Order)
The International Conference on Learning Representations (ICLR 2021)

Generalization of Two-Layer Neural Networks: An Asymptotic View
J. Ba, M. Erdogdu, T. Suzuki, D. Wu, T. Zhang (Alphabetical Order)
The International Conference on Learning Representations (ICLR 2020), Spotlight

Towards Characterizing the High-dimensional Bias of Kernel-based Particle Inference Algorithms
J. Ba, M. Erdogdu, S.Sun, T. Suzuki, D. Wu, T. Zhang (Alphabetical Order)
Advances in Approximate Bayesian Inference 2019, Spotlight

Non-asymptotic Comparison of SVRG and SGD: Tradeoffs Between Compute and Speed
J. Ba, M. Erdogdu, Q. Zhang, T. Zhang (Alphabetical Order)
ArXiv Preprint

Service

I am teaching the following courses at EPFL:

• Graduate courses
  • CS-433, Machine Learning by Martin Jaggi and Nicolas Flammarion
  • CS-401, Applied Data Analysis by Bob West
• Undergraduate courses
  • CS-233, Introduction to Machine Learning by Mathieu Salzmann and Pascal Fua

I am also a reviewer for machine learning conferences including ICLR, ICML, NeurIPS, CVPR and ICCV.